From 5263bba2ca0afc23f0f192b9aee9e020650774bb Mon Sep 17 00:00:00 2001
From: Jericho Tolentino <68654047+jericht@users.noreply.github.com>
Date: Tue, 5 Dec 2023 17:01:06 -0600
Subject: [PATCH] fix: increase timeout for fleet creation (#43)

Signed-off-by: Jericho Tolentino <68654047+jericht@users.noreply.github.com>
---
 src/deadline_test_fixtures/deadline/resources.py | 2 ++
 test/unit/deadline/test_resources.py             | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/deadline_test_fixtures/deadline/resources.py b/src/deadline_test_fixtures/deadline/resources.py
index 0e7f572a..8cad0739 100644
--- a/src/deadline_test_fixtures/deadline/resources.py
+++ b/src/deadline_test_fixtures/deadline/resources.py
@@ -143,6 +143,8 @@ def create(
             client=client,
             desired_status="ACTIVE",
             allowed_statuses=set(["CREATE_IN_PROGRESS"]),
+            interval_s=10,
+            max_retries=6 * 5,  # 5 minutes to allow CMF fleet creation to complete
         )
 
         return fleet
diff --git a/test/unit/deadline/test_resources.py b/test/unit/deadline/test_resources.py
index 8559ab68..958b523a 100644
--- a/test/unit/deadline/test_resources.py
+++ b/test/unit/deadline/test_resources.py
@@ -165,6 +165,8 @@ def test_create(self, farm: Farm) -> None:
             client=mock_client,
             desired_status="ACTIVE",
             allowed_statuses=set(["CREATE_IN_PROGRESS"]),
+            interval_s=10,
+            max_retries=6 * 5,  # 5 minutes for CMF fleet creation
         )
 
     def test_delete(self, fleet: Fleet) -> None: