Adds test

Author: AltamashShaikh

API.php

@@ -51,13 +51,13 @@ public function getTreemapData(
         $availableHeight = false,
         $show_evolution_values = false
     ) {
-        if (!Request::isCurrentApiRequestTheRootApiRequest()) {
+        if (!Request::isCurrentApiRequestTheRootApiRequest() && !defined('PIWIK_TEST_MODE')) {
             return [];
         }
         list($module, $method) = explode('.', $apiMethod);
         $disAllowedApiActions = ['getBulkRequest'];
         // Block if API action does not start with get
-        if (!$method || in_array(strtolower($method), $disAllowedApiActions) || stripos($method, 'get') !== 0) {
+        if (!$method || in_array($method, $disAllowedApiActions) || stripos($method, 'get') !== 0) {
             throw new BadRequestException(Piwik::translate('TreemapVisualization_InvalidApiMethodException'));
         }
 

tests/Integration/ApiTest.php

@@ -0,0 +1,78 @@
+<?php
+
+/**
+ * Matomo - free/libre analytics platform
+ *
+ * @link    https://matomo.org
+ * @license https://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
+ */
+
+namespace Piwik\Plugins\TreemapVisualization\tests\Integration;
+
+use Piwik\Http\BadRequestException;
+use Piwik\Plugins\TreemapVisualization\API;
+use Piwik\Tests\Framework\Fixture;
+use Piwik\Tests\Framework\TestCase\IntegrationTestCase;
+
+/**
+ * @group TreemapVisualization
+ * @group ApiTest
+ * @group Plugins
+ */
+class ApiTest extends IntegrationTestCase
+{
+    private $api;
+
+    public function setUp(): void
+    {
+        parent::setUp();
+
+        Fixture::createSuperUser();
+        $this->api = API::getInstance();
+    }
+
+    /**
+     * @dataProvider validMethod
+     */
+    public function testGetTreemapDataAllowsValidApiMethod($method): void
+    {
+        $result = $this->api->getTreemapData(
+            $method,
+            'nb_visits',
+            'day',
+            '2025-02-03'
+        );
+
+        $this->assertIsArray($result);
+    }
+
+    /**
+     * @dataProvider inValidMethod
+     */
+    public function testGetTreemapDataRejectsInvalidApiMethod($method): void
+    {
+        $this->expectException(BadRequestException::class);
+        $this->api->getTreemapData(
+            $method,
+            'nb_visits',
+            'day',
+            '2025-02-03'
+        );
+    }
+
+    public function validMethod()
+    {
+        return [
+            ['API.getMatomoVersion'],
+            ['API.getPhpVersion'],
+        ];
+    }
+
+    public function inValidMethod()
+    {
+        return [
+            ['API.getBulkRequest'],
+            ['UsersManager.deleteUser'],
+        ];
+    }
+}