diff --git a/tests/app/aws/test_s3.py b/tests/app/aws/test_s3.py index 7ff3eac01..d625f6b06 100644 --- a/tests/app/aws/test_s3.py +++ b/tests/app/aws/test_s3.py @@ -87,20 +87,22 @@ def test_get_phone_number_from_s3( phone_number = get_phone_number_from_s3("service_id", job_id, job_row_number) assert phone_number == expected_phone_number + def mock_s3_get_object_slowdown(*args, **kwargs): error_response = { - 'Error': { - 'Code': 'SlowDown', - 'Message': 'Reduce your request rate', + "Error": { + "Code": "SlowDown", + "Message": "Reduce your request rate", } } - raise ClientError(error_response, 'GetObject') + raise ClientError(error_response, "GetObject") + def test_get_job_from_s3_exponential_backoff(mocker): - get_s3_object = mocker.patch("app.aws.s3.get_s3_object", side_effect=mock_s3_get_object_slowdown) + mocker.patch("app.aws.s3.get_s3_object", side_effect=mock_s3_get_object_slowdown) with pytest.raises(Exception) as exc_info: - job = get_job_from_s3("service_id", "job_id") - assert 'Failed to get object after 5 attempts' in str(exc_info) + get_job_from_s3("service_id", "job_id") + assert "Failed to get object after 5 attempts" in str(exc_info) @pytest.mark.parametrize(