From 65838d2695df2aa9253537099caada677d5cccfe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guilhem=20Barth=C3=A9s?= Date: Mon, 30 Sep 2024 16:09:16 +0200 Subject: [PATCH] chore: improve `test_function_cancelled_cp` MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Guilhem Barthés --- tests/test_docker_image_build.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_docker_image_build.py b/tests/test_docker_image_build.py index 96bd57d5..ddb11f0d 100644 --- a/tests/test_docker_image_build.py +++ b/tests/test_docker_image_build.py @@ -120,6 +120,8 @@ def test_function_cancelled_cp(factory, cfg, client, worker, default_dataset): cp = client.wait_compute_plan(cp.key, raise_on_failure=False, timeout=cfg.options.future_timeout) assert cp.status == "PLAN_STATUS_CANCELED" - client.wait_function(function_wait.key, timeout=cfg.options.future_timeout) + # Check that `function_wait` (built first) reaches status DONE or CANCELED + function_wait = client.wait_function(function_wait.key, raise_on_failure=False, timeout=cfg.options.future_timeout) + assert function_wait.status != "FUNCTION_STATUS_FAILED" function_test = client.wait_function(function_test.key, raise_on_failure=False, timeout=cfg.options.future_timeout) assert function_test.status == "FUNCTION_STATUS_CANCELED"