Update mock_gce to set _engine_ready = True

globus · Nov 13, 2024 · 15e3773 · 15e3773
1 parent 250be46
commit 15e3773
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/compute_endpoint/tests/integration/endpoint/executors/test_gcengine_retries.py b/compute_endpoint/tests/integration/endpoint/executors/test_gcengine_retries.py
@@ -49,14 +49,14 @@ def mock_gce(tmp_path):
             working_dir=scripts_dir,
             provider=LocalProvider(min_blocks=0, max_blocks=0, init_blocks=0),
         )
+        engine._engine_ready = True
         engine.results_passthrough = queue.Queue()
         yield engine
 
 
 def test_success_after_1_fail(mock_gce, serde, ez_pack_task):
     engine = mock_gce
     engine.max_retries_on_system_failure = 2
-    engine._engine_ready = True
     q = engine.results_passthrough
     task_id = uuid.uuid1()
     num = random.randint(1, 10000)
@@ -78,7 +78,6 @@ def test_repeated_fail(mock_gce, ez_pack_task):
     fail_count = 2
     engine = mock_gce
     engine.max_retries_on_system_failure = fail_count
-    engine._engine_ready = True
     q = engine.results_passthrough
     task_id = uuid.uuid1()