From f7dce28546eb046b99539b1ab4a98b0d7a8037b3 Mon Sep 17 00:00:00 2001 From: Ryan Ahearn Date: Mon, 31 Oct 2022 16:32:20 -0400 Subject: [PATCH] Use safe env getter for test verification --- tests/app/celery/test_ftp_update_tasks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/app/celery/test_ftp_update_tasks.py b/tests/app/celery/test_ftp_update_tasks.py index e0931d31f..4cc6c4baa 100644 --- a/tests/app/celery/test_ftp_update_tasks.py +++ b/tests/app/celery/test_ftp_update_tasks.py @@ -92,9 +92,9 @@ def test_update_letter_notifications_statuses_calls_with_correct_bucket_location s3_mock.assert_called_with('{}-ftp'.format( current_app.config['NOTIFY_EMAIL_DOMAIN']), 'NOTIFY-20170823160812-RSP.TXT', - os.environ['AWS_ACCESS_KEY_ID'], - os.environ['AWS_SECRET_ACCESS_KEY'], - os.environ['AWS_REGION'], + os.environ.get('AWS_ACCESS_KEY_ID'), + os.environ.get('AWS_SECRET_ACCESS_KEY'), + os.environ.get('AWS_REGION'), )