Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/test_provenance_db.py
Show All 35 Lines | for status in TEST_OBJECTS["origin_visit_status"]: | ||||
assert isinstance(status, OriginVisitStatus) | assert isinstance(status, OriginVisitStatus) | ||||
if status.snapshot is not None: | if status.snapshot is not None: | ||||
entry = OriginEntry(url=status.origin, snapshot=status.snapshot) | entry = OriginEntry(url=status.origin, snapshot=status.snapshot) | ||||
origin_add(provenance, archive, [entry]) | origin_add(provenance, archive, [entry]) | ||||
# TODO: check some facts here | # TODO: check some facts here | ||||
def test_provenance_flavor(provenance: ProvenanceInterface) -> None: | def test_provenance_flavor(provenance: ProvenanceInterface) -> None: | ||||
assert isinstance(provenance.storage, ProvenanceDBBase) | if isinstance(provenance.storage, ProvenanceDBBase): | ||||
assert provenance.storage.flavor in ("with-path", "without-path") | assert provenance.storage.flavor in ("with-path", "without-path") | ||||
backend_class: Type[ProvenanceStorageInterface] | backend_class: Type[ProvenanceStorageInterface] | ||||
if provenance.storage.flavor == "with-path": | if provenance.storage.flavor == "with-path": | ||||
backend_class = ProvenanceWithPathDB | backend_class = ProvenanceWithPathDB | ||||
else: | else: | ||||
backend_class = ProvenanceWithoutPathDB | backend_class = ProvenanceWithoutPathDB | ||||
assert isinstance(provenance.storage, backend_class) | assert isinstance(provenance.storage, backend_class) |