Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/algos/test_origin.py
| Show First 20 Lines • Show All 87 Lines • ▼ Show 20 Lines | def test_origin_get_latest_visit_status_none(swh_storage): | ||||
| """ | """ | ||||
| # unknown origin so no result | # unknown origin so no result | ||||
| assert origin_get_latest_visit_status(swh_storage, "unknown-origin") is None | assert origin_get_latest_visit_status(swh_storage, "unknown-origin") is None | ||||
| # unknown type so no result | # unknown type so no result | ||||
| origin = Origin.from_dict(data.origin) | origin = Origin.from_dict(data.origin) | ||||
| swh_storage.origin_add_one(origin) | swh_storage.origin_add_one(origin) | ||||
| swh_storage.origin_visit_add( | swh_storage.origin_visit_add( | ||||
| [ | [OriginVisit(origin=origin.url, date=data.date_visit1, type="git",),] | ||||
| OriginVisit( | |||||
| origin=origin.url, | |||||
| date=data.date_visit1, | |||||
| type="git", | |||||
| status="ongoing", | |||||
| snapshot=None, | |||||
| ), | |||||
| ] | |||||
| )[0] | )[0] | ||||
| actual_origin_visit = origin_get_latest_visit_status( | actual_origin_visit = origin_get_latest_visit_status( | ||||
| swh_storage, origin.url, type="unknown" | swh_storage, origin.url, type="unknown" | ||||
| ) | ) | ||||
| assert actual_origin_visit is None | assert actual_origin_visit is None | ||||
| actual_origin_visit = origin_get_latest_visit_status( | actual_origin_visit = origin_get_latest_visit_status( | ||||
| swh_storage, origin.url, require_snapshot=True | swh_storage, origin.url, require_snapshot=True | ||||
| Show All 12 Lines | def init_storage_with_origin_visits(swh_storage): | ||||
| """ | """ | ||||
| origin1 = Origin.from_dict(data.origin) | origin1 = Origin.from_dict(data.origin) | ||||
| origin2 = Origin.from_dict(data.origin2) | origin2 = Origin.from_dict(data.origin2) | ||||
| swh_storage.origin_add([origin1, origin2]) | swh_storage.origin_add([origin1, origin2]) | ||||
| ov1, ov2 = swh_storage.origin_visit_add( | ov1, ov2 = swh_storage.origin_visit_add( | ||||
| [ | [ | ||||
| OriginVisit( | OriginVisit( | ||||
| origin=origin1.url, | origin=origin1.url, date=data.date_visit1, type=data.type_visit1, | ||||
| date=data.date_visit1, | |||||
| type=data.type_visit1, | |||||
| status="ongoing", | |||||
| snapshot=None, | |||||
| ), | ), | ||||
| OriginVisit( | OriginVisit( | ||||
| origin=origin2.url, | origin=origin2.url, date=data.date_visit2, type=data.type_visit2, | ||||
| date=data.date_visit2, | |||||
| type=data.type_visit2, | |||||
| status="ongoing", | |||||
| snapshot=None, | |||||
| ), | ), | ||||
| ] | ] | ||||
| ) | ) | ||||
| snapshot = Snapshot.from_dict(data.complete_snapshot) | snapshot = Snapshot.from_dict(data.complete_snapshot) | ||||
| swh_storage.snapshot_add([snapshot]) | swh_storage.snapshot_add([snapshot]) | ||||
| date_now = now() | date_now = now() | ||||
| ▲ Show 20 Lines • Show All 156 Lines • ▼ Show 20 Lines | def test_origin_get_latest_visit_status_filter_snapshot(swh_storage): | ||||
| assert actual_ov2.visit == ov2.visit | assert actual_ov2.visit == ov2.visit | ||||
| assert actual_ov2.type == ov2.type | assert actual_ov2.type == ov2.type | ||||
| assert actual_ovs22 == ovs22 | assert actual_ovs22 == ovs22 | ||||
| date_now = now() | date_now = now() | ||||
| # Add another visit | # Add another visit | ||||
| swh_storage.origin_visit_add( | swh_storage.origin_visit_add( | ||||
| [ | [OriginVisit(origin=origin2.url, date=date_now, type=data.type_visit2,),] | ||||
| OriginVisit( | |||||
| origin=origin2.url, | |||||
| date=date_now, | |||||
| type=data.type_visit2, | |||||
| status="ongoing", | |||||
| snapshot=None, | |||||
| ), | |||||
| ] | |||||
| ) | ) | ||||
| # Requiring the latest visit with a snapshot, we still find the previous visit | # Requiring the latest visit with a snapshot, we still find the previous visit | ||||
| ov2, ovs22 = origin_get_latest_visit_status( | ov2, ovs22 = origin_get_latest_visit_status( | ||||
| swh_storage, origin2.url, require_snapshot=True | swh_storage, origin2.url, require_snapshot=True | ||||
| ) | ) | ||||
| assert actual_ov2.origin == ov2.origin | assert actual_ov2.origin == ov2.origin | ||||
| assert actual_ov2.visit == ov2.visit | assert actual_ov2.visit == ov2.visit | ||||
| assert actual_ov2.type == ov2.type | assert actual_ov2.type == ov2.type | ||||
| assert actual_ovs22 == ovs22 | assert actual_ovs22 == ovs22 | ||||