Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
| Show First 20 Lines • Show All 1,527 Lines • ▼ Show 20 Lines | def test_origin_visit_upsert_new(self, swh_storage, use_url): | ||||
| assert origin_id is not None | assert origin_id is not None | ||||
| # when | # when | ||||
| swh_storage.origin_visit_upsert([ | swh_storage.origin_visit_upsert([ | ||||
| { | { | ||||
| 'origin': data.origin2, | 'origin': data.origin2, | ||||
| 'date': data.date_visit2, | 'date': data.date_visit2, | ||||
| 'visit': 123, | 'visit': 123, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| }, | }, | ||||
| { | { | ||||
| 'origin': data.origin2, | 'origin': data.origin2, | ||||
| 'date': '2018-01-01 23:00:00+00', | 'date': '2018-01-01 23:00:00+00', | ||||
| 'visit': 1234, | 'visit': 1234, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| }, | }, | ||||
| ]) | ]) | ||||
| # then | # then | ||||
| actual_origin_visits = list(swh_storage.origin_visit_get( | actual_origin_visits = list(swh_storage.origin_visit_get( | ||||
| origin_url)) | origin_url)) | ||||
| assert actual_origin_visits == [ | assert actual_origin_visits == [ | ||||
| { | { | ||||
| 'origin': origin_id, | 'origin': origin_id, | ||||
| 'date': data.date_visit2, | 'date': data.date_visit2, | ||||
| 'visit': 123, | 'visit': 123, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| }, | }, | ||||
| { | { | ||||
| 'origin': origin_id, | 'origin': origin_id, | ||||
| 'date': data.date_visit3, | 'date': data.date_visit3, | ||||
| 'visit': 1234, | 'visit': 1234, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| }, | }, | ||||
| ] | ] | ||||
| expected_origin = data.origin2 | expected_origin = data.origin2 | ||||
| data1 = { | data1 = { | ||||
| 'origin': expected_origin, | 'origin': expected_origin, | ||||
| 'date': data.date_visit2, | 'date': data.date_visit2, | ||||
| 'visit': 123, | 'visit': 123, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| } | } | ||||
| data2 = { | data2 = { | ||||
| 'origin': expected_origin, | 'origin': expected_origin, | ||||
| 'date': data.date_visit3, | 'date': data.date_visit3, | ||||
| 'visit': 1234, | 'visit': 1234, | ||||
| 'type': data.origin2['type'], | 'type': data.type_visit2, | ||||
| 'status': 'full', | 'status': 'full', | ||||
| 'metadata': None, | 'metadata': None, | ||||
| 'snapshot': None, | 'snapshot': None, | ||||
| } | } | ||||
| assert list(swh_storage.journal_writer.objects) == [ | assert list(swh_storage.journal_writer.objects) == [ | ||||
| ('origin', expected_origin), | ('origin', expected_origin), | ||||
| ('origin_visit', data1), | ('origin_visit', data1), | ||||
| ('origin_visit', data2)] | ('origin_visit', data2)] | ||||
| ▲ Show 20 Lines • Show All 1,918 Lines • Show Last 20 Lines | |||||