Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/package/tests/test_loader.py
Show First 20 Lines • Show All 74 Lines • ▼ Show 20 Lines | def get_package_info(self, version): | ||||
yield (f"branch-{version}", p_info) | yield (f"branch-{version}", p_info) | ||||
def _load_release(self, p_info, origin): | def _load_release(self, p_info, origin): | ||||
return None | return None | ||||
def test_loader_origin_visit_failure(swh_storage): | def test_loader_origin_visit_failure(swh_storage): | ||||
"""Failure to add origin or origin visit should failed immediately""" | """Failure to add origin or origin visit should failed immediately""" | ||||
loader = PackageLoader(swh_storage, "some-url") | loader = StubPackageLoader(swh_storage, "some-url") | ||||
loader.storage = FakeStorage() | loader.storage = FakeStorage() | ||||
actual_load_status = loader.load() | actual_load_status = loader.load() | ||||
assert actual_load_status == {"status": "failed"} | assert actual_load_status == {"status": "failed"} | ||||
loader.storage = FakeStorage2() | loader.storage = FakeStorage2() | ||||
actual_load_status2 = loader.load() | actual_load_status2 = loader.load() | ||||
Show All 14 Lines | rel2 = Release( | ||||
name=b"bbbb", | name=b"bbbb", | ||||
message=b"bbbb", | message=b"bbbb", | ||||
target=target, | target=target, | ||||
target_type=ModelObjectType.DIRECTORY, | target_type=ModelObjectType.DIRECTORY, | ||||
synthetic=False, | synthetic=False, | ||||
) | ) | ||||
storage.release_add([rel1, rel2]) | storage.release_add([rel1, rel2]) | ||||
loader = PackageLoader(storage, "http://example.org/") # type: ignore | loader = StubPackageLoader(storage, "http://example.org/") | ||||
p_info = Mock(wraps=BasePackageInfo(None, None, None)) # type: ignore | p_info = Mock(wraps=BasePackageInfo(None, None, None)) # type: ignore | ||||
# The PackageInfo does not support extids | # The PackageInfo does not support extids | ||||
p_info.extid.return_value = None | p_info.extid.return_value = None | ||||
known_extids = {("extid-type", 0, b"extid-of-aaaa"): [rel1.swhid()]} | known_extids = {("extid-type", 0, b"extid-of-aaaa"): [rel1.swhid()]} | ||||
whitelist = {b"unused"} | whitelist = {b"unused"} | ||||
assert loader.resolve_object_from_extids(known_extids, p_info, whitelist) is None | assert loader.resolve_object_from_extids(known_extids, p_info, whitelist) is None | ||||
Show All 32 Lines | def test_resolve_object_from_extids_missing_target() -> None: | ||||
rel = Release( | rel = Release( | ||||
name=b"aaaa", | name=b"aaaa", | ||||
message=b"aaaa", | message=b"aaaa", | ||||
target=target, | target=target, | ||||
target_type=ModelObjectType.DIRECTORY, | target_type=ModelObjectType.DIRECTORY, | ||||
synthetic=False, | synthetic=False, | ||||
) | ) | ||||
loader = PackageLoader(storage, "http://example.org/") # type: ignore | loader = StubPackageLoader(storage, "http://example.org/") | ||||
p_info = Mock(wraps=BasePackageInfo(None, None, None)) # type: ignore | p_info = Mock(wraps=BasePackageInfo(None, None, None)) # type: ignore | ||||
known_extids = {("extid-type", 0, b"extid-of-aaaa"): [rel.swhid()]} | known_extids = {("extid-type", 0, b"extid-of-aaaa"): [rel.swhid()]} | ||||
p_info.extid.return_value = ("extid-type", 0, b"extid-of-aaaa") | p_info.extid.return_value = ("extid-type", 0, b"extid-of-aaaa") | ||||
whitelist = {rel.id} | whitelist = {rel.id} | ||||
# Targeted release is missing from the storage | # Targeted release is missing from the storage | ||||
▲ Show 20 Lines • Show All 349 Lines • Show Last 20 Lines |