Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/test_provenance_db.py
| Show All 20 Lines | def ts2dt(ts: dict) -> datetime.datetime: | ||||
| return timestamp.replace(microsecond=ts["timestamp"]["microseconds"]) | return timestamp.replace(microsecond=ts["timestamp"]["microseconds"]) | ||||
| def test_provenance_origin_add(provenance, swh_storage_with_objects): | def test_provenance_origin_add(provenance, swh_storage_with_objects): | ||||
| """Test the origin_add function""" | """Test the origin_add function""" | ||||
| archive = ArchiveStorage(swh_storage_with_objects) | archive = ArchiveStorage(swh_storage_with_objects) | ||||
| for status in TEST_OBJECTS["origin_visit_status"]: | for status in TEST_OBJECTS["origin_visit_status"]: | ||||
| if status.snapshot is not None: | if status.snapshot is not None: | ||||
| entry = OriginEntry( | entry = OriginEntry(url=status.origin, snapshot=status.snapshot) | ||||
| url=status.origin, date=status.date, 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): | def test_provenance_flavor(provenance): | ||||
| assert provenance.storage.flavor in ("with-path", "without-path") | assert provenance.storage.flavor in ("with-path", "without-path") | ||||
| 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) | ||||