diff --git a/tests/app/aws/test_s3.py b/tests/app/aws/test_s3.py index 4e443e6f9..b5fe7348d 100644 --- a/tests/app/aws/test_s3.py +++ b/tests/app/aws/test_s3.py @@ -279,12 +279,12 @@ def test_get_job_from_s3_exponential_backoff_on_throttling(mocker): def test_get_job_from_s3_exponential_backoff_on_no_such_key(mocker): # We try multiple times to retrieve the job, and if we can't we return None mock_get_object = mocker.patch( - "app.aws.s3.get_s3_object", side_effect=mock_s3_get_object_slowdown + "app.aws.s3.get_s3_object", side_effect=mock_s3_get_object_no_such_key ) mocker.patch("app.aws.s3.file_exists", return_value=True) job = get_job_from_s3("service_id", "job_id") assert job is None - assert mock_get_object.call_count == 8 + assert mock_get_object.call_count == 2 def test_get_job_from_s3_exponential_backoff_on_random_exception(mocker):