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 | |||||