mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 19:03:30 -05:00
24 lines
786 B
Python
24 lines
786 B
Python
from unittest.mock import Mock
|
|
|
|
from app.s3_client.s3_csv_client import set_metadata_on_csv_upload
|
|
|
|
|
|
def test_sets_metadata(client_request, mocker):
|
|
mocked_s3_object = Mock(
|
|
bucket_name="test-csv-upload", key="service-1234-notify/5678.csv"
|
|
)
|
|
mocked_get_s3_object = mocker.patch(
|
|
"app.s3_client.s3_csv_client.get_csv_upload",
|
|
return_value=mocked_s3_object,
|
|
)
|
|
|
|
set_metadata_on_csv_upload("1234", "5678", foo="bar", baz=True)
|
|
|
|
mocked_get_s3_object.assert_called_once_with("1234", "5678")
|
|
mocked_s3_object.copy_from.assert_called_once_with(
|
|
CopySource="test-csv-upload/service-1234-notify/5678.csv",
|
|
Metadata={"baz": "True", "foo": "bar"},
|
|
MetadataDirective="REPLACE",
|
|
ServerSideEncryption="AES256",
|
|
)
|