diff --git a/tests/middleware/test_retries.py b/tests/middleware/test_retries.py index b2c1be44..1e6a32ca 100644 --- a/tests/middleware/test_retries.py +++ b/tests/middleware/test_retries.py @@ -381,6 +381,7 @@ def do_work(): # And the exhausted handler should have been called. assert len(called_at) == 1 + def test_on_retry_exhausted_is_not_sent_for_success(stub_broker, stub_worker): attempted_at = [] called_at = [] @@ -405,6 +406,7 @@ def do_work(): # And the exhausted callback should have never been called assert len(called_at) == 0 + def test_on_retry_exhausted_is_not_sent_for_eventual_success(stub_broker, stub_worker): attempted_at = [] called_at = []