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 |