def test_build_dict_contains_origin_and_sha():
mc = meta_data_creator.MetaDataCreator('filename', 'sha', 'origin')
expected_sha = 'sha'
expected_origin = 'origin'
dic = mc.build_and_return_dict()
actual_sha = dic['sha']
actual_origin = dic['origin']
assert expected_origin == actual_origin
assert expected_sha == actual_sha
def test_json_file_write_to_gcp_called():
gcs_client_mock = MagicMock(speç=GCStorage)
bucket_blob_mock = MagicMock(spec=storage.bucket.Bucket)
gcs_client_mock.get_bucket.return_value = bucket_blob_mock
blob_mock = MagicMock(spec=storage.blob.Blob)
bucket_blob_mock.blob.return_value = blob_mock
mc = meta_data_creator.MetaDataCreator('filename', 'sha', 'origin')
mc.write_file_to_gcp(gcs_client_mock)
gcs_client_mock.store_metadata.assert_called_once()
def test_invalid_filename_throw_value_error():
with pytest.raises(ValueError):
meta_data_creator.MetaDataCreator(1, 'sha', 'origin')
def test_output_filename_is_original_filename_with_json_ext():
mc = meta_data_creator.MetaDataCreator('filename', 'sha', 'origin')
assert mc.output_filename == 'filename.json'
def test_invalid_commit_throw_value_error():
with pytest.raises(ValueError):
meta_data_creator.MetaDataCreator('filename', {}, 'origin')