def test_download_hashes_exception(cbapi_mock, hashes):
"""Unit test _download_hashes function when an Exception is thrown."""
with pytest.raises(Exception):
cbapi_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
Exception)
dl = _download_hashes(cbapi_mock.api, hashes, 60)
assert dl is None
def test_download_hashes(cbcloud_api_mock, hashes):
"""Unit test _download_hashes function."""
cbcloud_api_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
FILE_DOWNLOAD_RESP)
hash_dl = _download_hashes(cbcloud_api_mock.api, hashes, 60)
assert len(hash_dl.found) == 1
assert len(hash_dl.not_found) == 1
assert len(hash_dl.error) == 0
def test_validate_download(cbcloud_api_mock, hashes):
"""Unit test _validate_download function."""
cbcloud_api_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
FILE_DOWNLOAD_RESP)
hash_dl = _download_hashes(cbcloud_api_mock.api, hashes, 60)
found_hashes, redownload_found_hashes = _validate_download(
cbcloud_api_mock.api, hash_dl, 60)
assert len(found_hashes) == 1
assert redownload_found_hashes is None
def test_validate_download_not_found(cbapi_mock, hashes):
"""Unit test _validate_download function with hashes not in UBS."""
with pytest.raises(Exception):
cbapi_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
Exception)
hash_dl = _download_hashes(cbapi_mock.api, hashes, 60)
check_hash_dl, retry = _validate_download(cbapi_mock.api, hash_dl, 60)
assert check_hash_dl is None
assert retry is None
def test_download_binary_metadata_not_found(cbapi_mock, hashes):
"""Unit test _download_binary_metadata function with hashes not in UBS."""
with pytest.raises(Exception):
cbapi_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
Exception) # put in the Exception
hash_dl = _download_hashes(cbapi_mock.api, hashes, 60)
assert hash_dl is None
check_hash_dl, retry = _validate_download(cbapi_mock.api, hash_dl, 60)
with pytest.raises(ValueError):
try_meta = _download_binary_metadata(cbapi_mock.api, check_hash_dl)
assert try_meta is None
assert check_hash_dl is None
assert retry is None
def test_download_binary_metadata(cbcloud_api_mock, hashes):
"""Unit test _download_binary_metadata function."""
cbcloud_api_mock.mock_request("POST", "/ubs/v1/orgs/test/file/_download",
FILE_DOWNLOAD_RESP)
hash_dl = _download_hashes(cbcloud_api_mock.api, hashes, 60)
found_hashes, redownload_found_hashes = _validate_download(
cbcloud_api_mock.api, hash_dl, 60)
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 = _download_binary_metadata(cbcloud_api_mock.api, found_hashes[0])
assert type(metadata) == dict
for key in METADATA_VALID.keys():
assert key in metadata
for key in metadata.keys():
assert key in METADATA_VALID