From bb77e3bc8dfd345164a14eef65266e5a2c1518f2 Mon Sep 17 00:00:00 2001 From: Simon Kelly Date: Thu, 9 Jul 2026 15:34:46 +0200 Subject: [PATCH] fix: de-flake celery retry test task_success fires async; wait for the update count before asserting. Co-Authored-By: Claude Opus 4.8 (1M context) --- tests/test_celery.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_celery.py b/tests/test_celery.py index da50a45..ac53bb2 100644 --- a/tests/test_celery.py +++ b/tests/test_celery.py @@ -316,6 +316,7 @@ def add_retry(self, a, b, is_retry=False): get_task.return_value = task_for_test() result = add_retry.delay(2, 2) assert result.get(timeout=10, propagate=True) == 4 + _wait_for_call_count(update, 4) assert create.call_count == 2 assert update.call_args_list == [ @@ -422,6 +423,7 @@ def task_signature(self, a): result = chain.delay() assert result.get(timeout=10, propagate=True) == 16 + _wait_for_call_count(update, 6) assert create.call_count == 3 assert get_task.call_count == 3