Changeset View
Changeset View
Standalone View
Standalone View
swh/journal/tests/test_replay.py
Show First 20 Lines • Show All 80 Lines • ▼ Show 20 Lines | assert OBJECT_TYPE_KEYS['revision'][1] == \ | ||||
[rev['id'] for rev in OBJECT_TYPE_KEYS['revision'][1]])) | [rev['id'] for rev in OBJECT_TYPE_KEYS['revision'][1]])) | ||||
assert OBJECT_TYPE_KEYS['release'][1] == \ | assert OBJECT_TYPE_KEYS['release'][1] == \ | ||||
list(storage.release_get( | list(storage.release_get( | ||||
[rel['id'] for rel in OBJECT_TYPE_KEYS['release'][1]])) | [rel['id'] for rel in OBJECT_TYPE_KEYS['release'][1]])) | ||||
origins = list(storage.origin_get( | origins = list(storage.origin_get( | ||||
[orig for orig in OBJECT_TYPE_KEYS['origin'][1]])) | [orig for orig in OBJECT_TYPE_KEYS['origin'][1]])) | ||||
assert OBJECT_TYPE_KEYS['origin'][1] == \ | assert OBJECT_TYPE_KEYS['origin'][1] == \ | ||||
[{'url': orig['url'], 'type': orig['type']} for orig in origins] | [{'url': orig['url']} for orig in origins] | ||||
for origin in origins: | for origin in origins: | ||||
origin_id_or_url = \ | origin_id_or_url = \ | ||||
origin['id'] if ENABLE_ORIGIN_IDS else origin['url'] | origin['id'] if ENABLE_ORIGIN_IDS else origin['url'] | ||||
expected_visits = [ | expected_visits = [ | ||||
{ | { | ||||
**visit, | **visit, | ||||
'origin': origin_id_or_url, | 'origin': origin_id_or_url, | ||||
'date': dateutil.parser.parse(visit['date']), | 'date': dateutil.parser.parse(visit['date']), | ||||
} | } | ||||
for visit in OBJECT_TYPE_KEYS['origin_visit'][1] | for visit in OBJECT_TYPE_KEYS['origin_visit'][1] | ||||
if visit['origin']['url'] == origin['url'] | if visit['origin']['url'] == origin['url'] | ||||
and visit['origin']['type'] == origin['type'] | |||||
] | ] | ||||
actual_visits = list(storage.origin_visit_get( | actual_visits = list(storage.origin_visit_get( | ||||
origin_id_or_url)) | origin_id_or_url)) | ||||
for visit in actual_visits: | for visit in actual_visits: | ||||
del visit['visit'] # opaque identifier | del visit['visit'] # opaque identifier | ||||
assert expected_visits == actual_visits | assert expected_visits == actual_visits | ||||
contents = list(storage.content_get_metadata( | contents = list(storage.content_get_metadata( | ||||
▲ Show 20 Lines • Show All 98 Lines • Show Last 20 Lines |