From cdfc6e640a0727ac593e90cdcb876edd14df2fd4 Mon Sep 17 00:00:00 2001
From: Bradley Schofield <ionicisere@gmail.com>
Date: Mon, 2 Sep 2024 13:18:19 +0900
Subject: [PATCH] Add Tests

---
 tests/e2e/General/UsageTest.php | 109 +++++++++++++++++++++++++++++++-
 1 file changed, 108 insertions(+), 1 deletion(-)

diff --git a/tests/e2e/General/UsageTest.php b/tests/e2e/General/UsageTest.php
index f4e45c293a..75b9e58bf3 100644
--- a/tests/e2e/General/UsageTest.php
+++ b/tests/e2e/General/UsageTest.php
@@ -13,6 +13,7 @@ use Tests\E2E\Scopes\SideServer;
 use Tests\E2E\Services\Functions\FunctionsBase;
 use Utopia\Database\Helpers\Permission;
 use Utopia\Database\Helpers\Role;
+use Utopia\Database\Query;
 use Utopia\Database\Validator\Datetime as DatetimeValidator;
 
 class UsageTest extends Scope
@@ -761,7 +762,7 @@ class UsageTest extends Scope
 
     /** @depends testPrepareFunctionsStats */
     #[Retry(count: 1)]
-    public function testFunctionsStats(array $data): void
+    public function testFunctionsStats(array $data): array
     {
         $functionId = $data['functionId'];
         $executionTime = $data['executionTime'];
@@ -818,6 +819,112 @@ class UsageTest extends Scope
         $this->validateDates($response['body']['executionsTime']);
         $this->assertGreaterThan(0, $response['body']['buildsTime'][array_key_last($response['body']['buildsTime'])]['value']);
         $this->validateDates($response['body']['buildsTime']);
+
+        return $data;
+    }
+
+    /** @depends testFunctionsStats */
+    public function testCustomDomainsFunctionStats(array $data): void
+    {
+        $functionId = $data['functionId'];
+
+        $response = $this->client->call(Client::METHOD_PUT, '/functions/' . $functionId, array_merge([
+            'content-type' => 'application/json',
+            'x-appwrite-project' => $this->getProject()['$id']
+        ], $this->getHeaders()), [
+            'name' => 'Test',
+            'execute' => ['any']
+        ]);
+
+        $this->assertEquals(200, $response['headers']['status-code']);
+
+        $rules = $this->client->call(Client::METHOD_GET, '/proxy/rules', array_merge([
+            'content-type' => 'application/json',
+            'x-appwrite-project' => $this->getProject()['$id'],
+        ], $this->getHeaders()), [
+            'queries' => [
+                Query::equal('resourceId', [$functionId])->toString(),
+                Query::equal('resourceType', ['function'])->toString(),
+            ],
+        ]);
+
+        $this->assertEquals(200, $rules['headers']['status-code']);
+        $this->assertEquals(1, $rules['body']['total']);
+        $this->assertCount(1, $rules['body']['rules']);
+        $this->assertNotEmpty($rules['body']['rules'][0]['domain']);
+
+        $domain = $rules['body']['rules'][0]['domain'];
+
+        $response = $this->client->call(
+            Client::METHOD_GET,
+            '/functions/' . $functionId . '/usage?range=30d',
+            $this->getConsoleHeaders()
+        );
+        
+        $this->assertEquals(200, $response['headers']['status-code']);
+        $this->assertEquals(19, count($response['body']));
+        $this->assertEquals('30d', $response['body']['range']);
+
+        $functionsMetrics = $response['body'];
+
+        $response = $this->client->call(
+            Client::METHOD_GET,
+            '/project/usage',
+            $this->getConsoleHeaders(),
+            [
+                'period' => '1h',
+                'startDate' => self::getToday(),
+                'endDate' => self::getTomorrow(),
+            ]
+        );
+
+        $this->assertEquals(200, $response['headers']['status-code']);
+
+        $projectMetrics = $response['body'];        
+
+        // Create custom domain execution
+        $proxyClient = new Client();
+        $proxyClient->setEndpoint('http://' . $domain);
+
+        $response = $proxyClient->call(Client::METHOD_GET, '/', array_merge([
+            'content-type' => 'application/json',
+            'x-appwrite-project' => $this->getProject()['$id']
+        ]));
+
+        $this->assertEquals(200, $response['headers']['status-code']);
+
+        sleep(self::WAIT);
+
+        // Compare new values with old values
+        $response = $this->client->call(
+            Client::METHOD_GET,
+            '/functions/' . $functionId . '/usage?range=30d',
+            $this->getConsoleHeaders()
+        );
+        
+        $this->assertEquals(200, $response['headers']['status-code']);
+        $this->assertEquals(19, count($response['body']));
+        $this->assertEquals('30d', $response['body']['range']);
+
+        // Check if the new values are greater than the old values
+        $this->assertEquals($functionsMetrics['executionsTotal'] + 1, $response['body']['executionsTotal']);
+        $this->assertGreaterThan($functionsMetrics['executionsTimeTotal'], $response['body']['executionsTimeTotal']);
+        $this->assertGreaterThan($functionsMetrics['executionsMbSecondsTotal'], $response['body']['executionsMbSecondsTotal']);
+
+        $response = $this->client->call(
+            Client::METHOD_GET,
+            '/project/usage',
+            $this->getConsoleHeaders(),
+            [
+                'period' => '1h',
+                'startDate' => self::getToday(),
+                'endDate' => self::getTomorrow(),
+            ]
+        );
+
+        $this->assertEquals(200, $response['headers']['status-code']);
+        $this->assertEquals($projectMetrics['executionsTotal'] + 1, $response['body']['executionsTotal']);
+        $this->assertGreaterThan($projectMetrics['executionsMbSecondsTotal'], $response['body']['executionsMbSecondsTotal']);
     }
 
     public function tearDown(): void