mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-19 05:59:44 -04:00
Use credentials output by terraform/development
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
import uuid
|
||||
from io import BytesIO
|
||||
from os import getenv
|
||||
from unittest.mock import ANY
|
||||
|
||||
import pytest
|
||||
@@ -178,6 +177,7 @@ def test_upload_contact_list_page(client_request):
|
||||
def test_upload_csv_file_shows_error_banner(
|
||||
client_request,
|
||||
mocker,
|
||||
notify_admin,
|
||||
mock_s3_upload,
|
||||
mock_get_job_doesnt_exist,
|
||||
mock_get_users_by_service,
|
||||
@@ -205,13 +205,14 @@ def test_upload_csv_file_shows_error_banner(
|
||||
_data={'file': (BytesIO(''.encode('utf-8')), 'invalid.csv')},
|
||||
_follow_redirects=True,
|
||||
)
|
||||
bucket_creds = notify_admin.config['CONTACT_LIST_BUCKET']
|
||||
mock_upload.assert_called_once_with(
|
||||
filedata='',
|
||||
region='us-west-2',
|
||||
bucket_name='test-contact-list',
|
||||
region=bucket_creds['region'],
|
||||
bucket_name=bucket_creds['bucket'],
|
||||
file_location=f"service-{SERVICE_ONE_ID}-notify/{fake_uuid}.csv",
|
||||
access_key=getenv('AWS_ACCESS_KEY_ID'),
|
||||
secret_key=getenv('AWS_SECRET_ACCESS_KEY'),
|
||||
access_key=bucket_creds['access_key_id'],
|
||||
secret_key=bucket_creds['secret_access_key'],
|
||||
)
|
||||
mock_set_metadata.assert_called_once_with(
|
||||
ANY,
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
from collections import namedtuple
|
||||
from os import getenv
|
||||
from unittest.mock import call
|
||||
|
||||
import pytest
|
||||
@@ -14,21 +13,10 @@ from app.s3_client.s3_logo_client import (
|
||||
upload_email_logo,
|
||||
)
|
||||
|
||||
default_access_key = getenv('AWS_ACCESS_KEY_ID')
|
||||
default_secret_key = getenv('AWS_SECRET_ACCESS_KEY')
|
||||
default_region = getenv('AWS_REGION')
|
||||
bucket = 'test_bucket'
|
||||
bucket_credentials = {
|
||||
'bucket': bucket,
|
||||
'access_key_id': default_access_key,
|
||||
'secret_access_key': default_secret_key,
|
||||
'region': default_region
|
||||
}
|
||||
data = {'data': 'some_data'}
|
||||
filename = 'test.png'
|
||||
svg_filename = 'test.svg'
|
||||
upload_id = 'test_uuid'
|
||||
region = 'us-west-2'
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
@@ -37,26 +25,29 @@ def upload_filename(fake_uuid):
|
||||
temp=TEMP_TAG.format(user_id=fake_uuid), unique_id=upload_id, filename=filename)
|
||||
|
||||
|
||||
def test_upload_email_logo_calls_correct_args(client_request, mocker, fake_uuid, upload_filename):
|
||||
@pytest.fixture
|
||||
def bucket_credentials(notify_admin):
|
||||
return notify_admin.config['LOGO_UPLOAD_BUCKET']
|
||||
|
||||
|
||||
def test_upload_email_logo_calls_correct_args(client_request, mocker, bucket_credentials, fake_uuid, upload_filename):
|
||||
mocker.patch('uuid.uuid4', return_value=upload_id)
|
||||
mocker.patch.dict('flask.current_app.config', {'LOGO_UPLOAD_BUCKET': bucket_credentials})
|
||||
mocked_s3_upload = mocker.patch('app.s3_client.s3_logo_client.utils_s3upload')
|
||||
|
||||
upload_email_logo(filename=filename, user_id=fake_uuid, filedata=data)
|
||||
|
||||
mocked_s3_upload.assert_called_once_with(
|
||||
filedata=data,
|
||||
region=region,
|
||||
region=bucket_credentials['region'],
|
||||
file_location=upload_filename,
|
||||
bucket_name=bucket,
|
||||
bucket_name=bucket_credentials['bucket'],
|
||||
content_type='image/png',
|
||||
access_key=default_access_key,
|
||||
secret_key=default_secret_key,
|
||||
access_key=bucket_credentials['access_key_id'],
|
||||
secret_key=bucket_credentials['secret_access_key'],
|
||||
)
|
||||
|
||||
|
||||
def test_persist_logo(client_request, mocker, fake_uuid, upload_filename):
|
||||
mocker.patch.dict('flask.current_app.config', {'LOGO_UPLOAD_BUCKET': bucket_credentials})
|
||||
def test_persist_logo(client_request, bucket_credentials, mocker, fake_uuid, upload_filename):
|
||||
mocked_get_s3_object = mocker.patch('app.s3_client.s3_logo_client.get_s3_object')
|
||||
mocked_delete_s3_object = mocker.patch('app.s3_client.s3_logo_client.delete_s3_object')
|
||||
|
||||
@@ -65,7 +56,11 @@ def test_persist_logo(client_request, mocker, fake_uuid, upload_filename):
|
||||
persist_logo(upload_filename, new_filename)
|
||||
|
||||
mocked_get_s3_object.assert_called_once_with(
|
||||
bucket, new_filename, default_access_key, default_secret_key, default_region)
|
||||
bucket_credentials['bucket'],
|
||||
new_filename,
|
||||
bucket_credentials['access_key_id'],
|
||||
bucket_credentials['secret_access_key'],
|
||||
bucket_credentials['region'])
|
||||
mocked_delete_s3_object.assert_called_once_with(upload_filename)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user