darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 18:01:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix flake 8

Browse Source
This commit is contained in:
Kenneth Kehl
2024-08-13 15:45:39 -07:00
parent db9197e7a6
commit 7258105c91
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/app/aws/test_s3.py
View File

@@ -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) phone_number = get_phone_number_from_s3("service_id", job_id, job_row_number)
assert phone_number == expected_phone_number assert phone_number == expected_phone_number
def mock_s3_get_object_slowdown(*args, **kwargs): def mock_s3_get_object_slowdown(*args, **kwargs):
error_response = { error_response = {
'Error': { "Error": {
'Code': 'SlowDown', "Code": "SlowDown",
'Message': 'Reduce your request rate', "Message": "Reduce your request rate",
} }
} }
raise ClientError(error_response, 'GetObject') raise ClientError(error_response, "GetObject")
def test_get_job_from_s3_exponential_backoff(mocker): 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: with pytest.raises(Exception) as exc_info:
job = get_job_from_s3("service_id", "job_id") get_job_from_s3("service_id", "job_id")
assert 'Failed to get object after 5 attempts' in str(exc_info) assert "Failed to get object after 5 attempts" in str(exc_info)
@pytest.mark.parametrize( @pytest.mark.parametrize(