From 7258105c916f434463b9be7d9de249b608d3177c Mon Sep 17 00:00:00 2001 From: Kenneth Kehl <@kkehl@flexion.us> Date: Tue, 13 Aug 2024 15:45:39 -0700 Subject: [PATCH] fix flake 8 --- tests/app/aws/test_s3.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) 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(