From 15e3773c0a3e60700d870de6ff13e33b9e79e9bc Mon Sep 17 00:00:00 2001 From: Yadu Babuji Date: Tue, 12 Nov 2024 21:53:37 -0600 Subject: [PATCH] Update mock_gce to set _engine_ready = True --- .../integration/endpoint/executors/test_gcengine_retries.py | 3 +-- 1 file changed, 1 insertion(+), 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 index e8ea441e8..905b3c58b 100644 --- a/compute_endpoint/tests/integration/endpoint/executors/test_gcengine_retries.py +++ b/compute_endpoint/tests/integration/endpoint/executors/test_gcengine_retries.py @@ -49,6 +49,7 @@ 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 @@ -56,7 +57,6 @@ def mock_gce(tmp_path): 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()