From 2384113a046b98febfd44d671794a984e80155bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matej=20Ba=C4=8Do?= <matejbaco2000@gmail.com>
Date: Mon, 3 Mar 2025 12:44:28 +0100
Subject: [PATCH] Fix flaky tests

---
 tests/e2e/Services/Functions/FunctionsCustomServerTest.php | 2 ++
 tests/e2e/Services/Sites/SitesCustomServerTest.php         | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/tests/e2e/Services/Functions/FunctionsCustomServerTest.php b/tests/e2e/Services/Functions/FunctionsCustomServerTest.php
index 2daab87ffa..e85cfbafc5 100644
--- a/tests/e2e/Services/Functions/FunctionsCustomServerTest.php
+++ b/tests/e2e/Services/Functions/FunctionsCustomServerTest.php
@@ -2,6 +2,7 @@
 
 namespace Tests\E2E\Services\Functions;
 
+use Appwrite\Tests\Retry;
 use Appwrite\Functions\Specification;
 use Tests\E2E\Client;
 use Tests\E2E\Scopes\ProjectCustom;
@@ -511,6 +512,7 @@ class FunctionsCustomServerTest extends Scope
     /**
      * @depends testUpdateFunction
      */
+    #[Retry(count: 3)]
     public function testCancelDeploymentBuild($data): void
     {
         $functionId = $data['functionId'];
diff --git a/tests/e2e/Services/Sites/SitesCustomServerTest.php b/tests/e2e/Services/Sites/SitesCustomServerTest.php
index ff624862b4..dc8af05ae5 100644
--- a/tests/e2e/Services/Sites/SitesCustomServerTest.php
+++ b/tests/e2e/Services/Sites/SitesCustomServerTest.php
@@ -2,6 +2,7 @@
 
 namespace Tests\E2E\Services\Sites;
 
+use Appwrite\Tests\Retry;
 use Appwrite\Sites\Specification;
 use Tests\E2E\Client;
 use Tests\E2E\Scopes\ProjectCustom;
@@ -674,6 +675,7 @@ class SitesCustomServerTest extends Scope
         $this->cleanupSite($siteId);
     }
 
+    #[Retry(count: 3)]
     public function testCancelDeploymentBuild(): void
     {
         $siteId = $this->setupSite([