def test_get_metadata(cbcloud_api_mock, hashes):
"""Unit test get_metadata function."""
cbcloud_api_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
FILE_DOWNLOAD_RESP)
found_hashes = download_hashes(cbcloud_api_mock.api, hashes)
cbcloud_api_mock.mock_request(
"GET",
f"/ubs/v1/orgs/test/sha256/{found_hashes[0]['sha256']}/metadata",
METADATA_DOWNLOAD_RESP[found_hashes[0]['sha256']])
metadata = get_metadata(cbcloud_api_mock.api, found_hashes[0])
assert isinstance(metadata, dict)
for key in METADATA_VALID.keys():
assert key in metadata
for key in metadata.keys():
assert key in METADATA_VALID
def test_download_hashes_invalid(cbcloud_api_mock, input):
"""Unit test _download_hashes function with empty input."""
cbcloud_api_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
None)
found_hashes = download_hashes(cbcloud_api_mock.api, input)
assert found_hashes == []