Skip to content

Commit 016ac76

Browse files
committed
add mock response for S3 version id in bulk upload tests
1 parent 138d014 commit 016ac76

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

lambdas/tests/unit/services/test_bulk_upload_service.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -848,6 +848,9 @@ def test_create_lg_records_and_copy_files(set_env, mocker, mock_uuid, repo_under
848848
repo_under_test.convert_to_document_reference = mocker.MagicMock(
849849
return_value=test_document_reference
850850
)
851+
repo_under_test.bulk_upload_s3_repository.copy_to_lg_bucket = mocker.MagicMock(
852+
return_value=MOCK_COPY_OBJECT_RESPONSE
853+
)
851854
TEST_STAGING_METADATA.retries = 0
852855
repo_under_test.resolve_source_file_path(TEST_STAGING_METADATA)
853856

@@ -865,6 +868,10 @@ def test_create_lg_records_and_copy_files(set_env, mocker, mock_uuid, repo_under
865868
dest_file_key=expected_dest_file_key,
866869
)
867870
assert test_document_reference.uploaded.__eq__(True)
871+
assert (
872+
test_document_reference.s3_version_id
873+
== MOCK_COPY_OBJECT_RESPONSE["VersionId"]
874+
)
868875
assert repo_under_test.bulk_upload_s3_repository.copy_to_lg_bucket.call_count == 3
869876
repo_under_test.dynamo_repository.create_record_in_lg_dynamo_table.assert_any_call(
870877
test_document_reference

0 commit comments

Comments
 (0)