From 24bf5d258469b2fbfd6f3c44d325cceb4f580bc1 Mon Sep 17 00:00:00 2001
From: loks0n <22452787+loks0n@users.noreply.github.com>
Date: Tue, 20 Feb 2024 15:27:01 +0000
Subject: [PATCH] fix: usage tests

---
 tests/e2e/General/UsageTest.php | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/tests/e2e/General/UsageTest.php b/tests/e2e/General/UsageTest.php
index a466d9aa69..b6bc5e4220 100644
--- a/tests/e2e/General/UsageTest.php
+++ b/tests/e2e/General/UsageTest.php
@@ -214,6 +214,7 @@ class UsageTest extends Scope
                     ], $this->getHeaders()),
                 );
 
+                $this->assertEquals(204, $response['headers']['status-code']);
                 $this->assertEmpty($response['body']);
 
                 $requestsTotal += 1;
@@ -257,13 +258,14 @@ class UsageTest extends Scope
                 ]
             );
 
+            $this->assertEquals(201, $response['headers']['status-code']);
             $this->assertNotEmpty($response['body']['$id']);
 
             $fileSize = $response['body']['sizeOriginal'];
 
             $storageTotal += $fileSize;
-            $filesTotal = $filesTotal + 1;
-            $requestsTotal = $requestsTotal + 1;
+            $filesTotal += 1;
+            $requestsTotal += 1;
 
             $fileId = $response['body']['$id'];
 
@@ -276,6 +278,7 @@ class UsageTest extends Scope
                     ], $this->getHeaders()),
                 );
 
+                $this->assertEquals(204, $response['headers']['status-code']);
                 $this->assertEmpty($response['body']);
 
                 $requestsTotal += 1;
@@ -310,9 +313,7 @@ class UsageTest extends Scope
         $response = $this->client->call(
             Client::METHOD_GET,
             '/project/usage',
-            array_merge([
-                'x-appwrite-project' => $this->getProject()['$id']
-            ], $this->getHeaders()),
+            $this->getConsoleHeaders(),
             [
                 'period' => '1d',
                 'startDate' => self::getToday(),
@@ -329,9 +330,7 @@ class UsageTest extends Scope
         $response = $this->client->call(
             Client::METHOD_GET,
             '/storage/usage?range=30d',
-            array_merge([
-                'x-appwrite-project' => $this->getProject()['$id']
-            ], $this->getHeaders()),
+            $this->getConsoleHeaders()
         );
 
         $this->assertEquals($storageTotal, $response['body']['storage'][array_key_last($response['body']['storage'])]['value']);
@@ -344,9 +343,7 @@ class UsageTest extends Scope
         $response = $this->client->call(
             Client::METHOD_GET,
             '/storage/' . $bucketId . '/usage?range=30d',
-            array_merge([
-                'x-appwrite-project' => $this->getProject()['$id']
-            ], $this->getHeaders()),
+            $this->getConsoleHeaders()
         );
 
         $this->assertEquals($storageTotal, $response['body']['storage'][array_key_last($response['body']['storage'])]['value']);