Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
Show First 20 Lines • Show All 349 Lines • ▼ Show 20 Lines | def setUp(self): | ||||
self.origin = { | self.origin = { | ||||
'url': 'file:///dev/null', | 'url': 'file:///dev/null', | ||||
'type': 'git', | 'type': 'git', | ||||
} | } | ||||
self.origin2 = { | self.origin2 = { | ||||
'url': 'file:///dev/zero', | 'url': 'file:///dev/zero', | ||||
'type': 'git', | 'type': 'hg', | ||||
} | } | ||||
self.provider = { | self.provider = { | ||||
'name': 'hal', | 'name': 'hal', | ||||
'type': 'deposit-client', | 'type': 'deposit-client', | ||||
'url': 'http:///hal/inria', | 'url': 'http:///hal/inria', | ||||
'metadata': { | 'metadata': { | ||||
'location': 'France' | 'location': 'France' | ||||
▲ Show 20 Lines • Show All 1,089 Lines • ▼ Show 20 Lines | def test_origin_visit_add(self): | ||||
self.assertIsNone(self.storage.origin_get([self.origin2])[0]) | self.assertIsNone(self.storage.origin_get([self.origin2])[0]) | ||||
origin_id = self.storage.origin_add_one(self.origin2) | origin_id = self.storage.origin_add_one(self.origin2) | ||||
self.assertIsNotNone(origin_id) | self.assertIsNotNone(origin_id) | ||||
# when | # when | ||||
origin_visit1 = self.storage.origin_visit_add( | origin_visit1 = self.storage.origin_visit_add( | ||||
origin_id, | origin_id, | ||||
type='git', | |||||
date=self.date_visit2) | |||||
actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | |||||
self.assertEqual(actual_origin_visits, | |||||
[{ | |||||
'origin': origin_id, | |||||
'date': self.date_visit2, | |||||
'visit': origin_visit1['visit'], | |||||
'type': 'git', | |||||
'status': 'ongoing', | |||||
'metadata': None, | |||||
'snapshot': None, | |||||
}]) | |||||
expected_origin = self.origin2.copy() | |||||
data = { | |||||
'origin': expected_origin, | |||||
'date': self.date_visit2, | |||||
'visit': origin_visit1['visit'], | |||||
'type': 'git', | |||||
'status': 'ongoing', | |||||
'metadata': None, | |||||
'snapshot': None, | |||||
} | |||||
self.assertEqual(list(self.journal_writer.objects), | |||||
[('origin', expected_origin), | |||||
('origin_visit', data)]) | |||||
def test_origin_visit_add_default_type(self): | |||||
# given | |||||
self.assertIsNone(self.storage.origin_get([self.origin2])[0]) | |||||
origin_id = self.storage.origin_add_one(self.origin2) | |||||
self.assertIsNotNone(origin_id) | |||||
# when | |||||
origin_visit1 = self.storage.origin_visit_add( | |||||
origin_id, | |||||
date=self.date_visit2) | date=self.date_visit2) | ||||
# then | # then | ||||
self.assertEqual(origin_visit1['origin'], origin_id) | self.assertEqual(origin_visit1['origin'], origin_id) | ||||
self.assertIsNotNone(origin_visit1['visit']) | self.assertIsNotNone(origin_visit1['visit']) | ||||
actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | ||||
self.assertEqual(actual_origin_visits, | self.assertEqual(actual_origin_visits, | ||||
[{ | [{ | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': 'hg', | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
expected_origin = self.origin2.copy() | expected_origin = self.origin2.copy() | ||||
data = { | data = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': 'hg', | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data)]) | ('origin_visit', data)]) | ||||
def test_origin_visit_update(self): | def test_origin_visit_update(self): | ||||
# given | # given | ||||
origin_id = self.storage.origin_add_one(self.origin2) | origin_id = self.storage.origin_add_one(self.origin) | ||||
origin_id2 = self.storage.origin_add_one(self.origin) | origin_id2 = self.storage.origin_add_one(self.origin2) | ||||
origin_visit1 = self.storage.origin_visit_add( | origin_visit1 = self.storage.origin_visit_add( | ||||
origin_id, | origin_id, | ||||
date=self.date_visit2) | date=self.date_visit2) | ||||
origin_visit2 = self.storage.origin_visit_add( | origin_visit2 = self.storage.origin_visit_add( | ||||
origin_id, | origin_id, | ||||
date=self.date_visit3) | date=self.date_visit3) | ||||
Show All 14 Lines | def test_origin_visit_update(self): | ||||
status='partial') | status='partial') | ||||
# then | # then | ||||
actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | ||||
self.assertEqual(actual_origin_visits, [{ | self.assertEqual(actual_origin_visits, [{ | ||||
'origin': origin_visit2['origin'], | 'origin': origin_visit2['origin'], | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': visit1_metadata, | 'metadata': visit1_metadata, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}, { | }, { | ||||
'origin': origin_visit2['origin'], | 'origin': origin_visit2['origin'], | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
actual_origin_visits_bis = list(self.storage.origin_visit_get( | actual_origin_visits_bis = list(self.storage.origin_visit_get( | ||||
origin_id, limit=1)) | origin_id, limit=1)) | ||||
self.assertEqual(actual_origin_visits_bis, | self.assertEqual(actual_origin_visits_bis, | ||||
[{ | [{ | ||||
'origin': origin_visit2['origin'], | 'origin': origin_visit2['origin'], | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': visit1_metadata, | 'metadata': visit1_metadata, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
actual_origin_visits_ter = list(self.storage.origin_visit_get( | actual_origin_visits_ter = list(self.storage.origin_visit_get( | ||||
origin_id, last_visit=origin_visit1['visit'])) | origin_id, last_visit=origin_visit1['visit'])) | ||||
self.assertEqual(actual_origin_visits_ter, | self.assertEqual(actual_origin_visits_ter, | ||||
[{ | [{ | ||||
'origin': origin_visit2['origin'], | 'origin': origin_visit2['origin'], | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
actual_origin_visits2 = list(self.storage.origin_visit_get(origin_id2)) | actual_origin_visits2 = list(self.storage.origin_visit_get(origin_id2)) | ||||
self.assertEqual(actual_origin_visits2, | self.assertEqual(actual_origin_visits2, | ||||
[{ | [{ | ||||
'origin': origin_visit3['origin'], | 'origin': origin_visit3['origin'], | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit3['visit'], | 'visit': origin_visit3['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'partial', | 'status': 'partial', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
expected_origin = self.origin2.copy() | expected_origin = self.origin.copy() | ||||
expected_origin2 = self.origin.copy() | expected_origin2 = self.origin2.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data3 = { | data3 = { | ||||
'origin': expected_origin2, | 'origin': expected_origin2, | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit3['visit'], | 'visit': origin_visit3['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data4 = { | data4 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'metadata': visit1_metadata, | 'metadata': visit1_metadata, | ||||
'status': 'full', | 'status': 'full', | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data5 = { | data5 = { | ||||
'origin': expected_origin2, | 'origin': expected_origin2, | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': origin_visit3['visit'], | 'visit': origin_visit3['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'partial', | 'status': 'partial', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin', expected_origin2), | ('origin', expected_origin2), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
Show All 20 Lines | def test_origin_visit_update_missing_snapshot(self): | ||||
origin_visit['origin'], origin_visit['visit']) | origin_visit['origin'], origin_visit['visit']) | ||||
self.assertEqual(actual_origin_visit['snapshot'], self.snapshot['id']) | self.assertEqual(actual_origin_visit['snapshot'], self.snapshot['id']) | ||||
# when | # when | ||||
self.storage.snapshot_add([self.snapshot]) | self.storage.snapshot_add([self.snapshot]) | ||||
self.assertEqual(actual_origin_visit['snapshot'], self.snapshot['id']) | self.assertEqual(actual_origin_visit['snapshot'], self.snapshot['id']) | ||||
def test_origin_visit_get_by(self): | def test_origin_visit_get_by(self): | ||||
origin_id = self.storage.origin_add_one(self.origin2) | origin_id = self.storage.origin_add_one(self.origin) | ||||
origin_id2 = self.storage.origin_add_one(self.origin) | origin_id2 = self.storage.origin_add_one(self.origin2) | ||||
origin_visit1 = self.storage.origin_visit_add( | origin_visit1 = self.storage.origin_visit_add( | ||||
origin_id, | origin_id, | ||||
date=self.date_visit2) | date=self.date_visit2) | ||||
self.storage.snapshot_add(origin_id, origin_visit1['visit'], | self.storage.snapshot_add(origin_id, origin_visit1['visit'], | ||||
self.snapshot) | self.snapshot) | ||||
Show All 11 Lines | def test_origin_visit_get_by(self): | ||||
origin_id, origin_visit1['visit'], status='full', | origin_id, origin_visit1['visit'], status='full', | ||||
metadata=visit1_metadata) | metadata=visit1_metadata) | ||||
expected_origin_visit = origin_visit1.copy() | expected_origin_visit = origin_visit1.copy() | ||||
expected_origin_visit.update({ | expected_origin_visit.update({ | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'type': self.origin['type'], | |||||
'metadata': visit1_metadata, | 'metadata': visit1_metadata, | ||||
'status': 'full', | 'status': 'full', | ||||
'snapshot': self.snapshot['id'], | 'snapshot': self.snapshot['id'], | ||||
}) | }) | ||||
# when | # when | ||||
actual_origin_visit1 = self.storage.origin_visit_get_by( | actual_origin_visit1 = self.storage.origin_visit_get_by( | ||||
origin_visit1['origin'], origin_visit1['visit']) | origin_visit1['origin'], origin_visit1['visit']) | ||||
Show All 9 Lines | def test_origin_visit_upsert_new(self): | ||||
self.assertIsNotNone(origin_id) | self.assertIsNotNone(origin_id) | ||||
# when | # when | ||||
self.storage.origin_visit_upsert([ | self.storage.origin_visit_upsert([ | ||||
{ | { | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': 123, | 'visit': 123, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}, | }, | ||||
{ | { | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': '2018-01-01 23:00:00+00', | 'date': '2018-01-01 23:00:00+00', | ||||
'visit': 1234, | 'visit': 1234, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}, | }, | ||||
]) | ]) | ||||
# then | # then | ||||
actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | ||||
self.assertEqual(actual_origin_visits, [ | self.assertEqual(actual_origin_visits, [ | ||||
{ | { | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': 123, | 'visit': 123, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}, | }, | ||||
{ | { | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': 1234, | 'visit': 1234, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}, | }, | ||||
]) | ]) | ||||
expected_origin = self.origin2.copy() | expected_origin = self.origin2.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': 123, | 'visit': 123, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit3, | 'date': self.date_visit3, | ||||
'visit': 1234, | 'visit': 1234, | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
('origin_visit', data2)]) | ('origin_visit', data2)]) | ||||
def test_origin_visit_upsert_existing(self): | def test_origin_visit_upsert_existing(self): | ||||
# given | # given | ||||
self.assertIsNone(self.storage.origin_get([self.origin2])[0]) | self.assertIsNone(self.storage.origin_get([self.origin2])[0]) | ||||
origin_id = self.storage.origin_add_one(self.origin2) | origin_id = self.storage.origin_add_one(self.origin2) | ||||
self.assertIsNotNone(origin_id) | self.assertIsNotNone(origin_id) | ||||
# when | # when | ||||
origin_visit1 = self.storage.origin_visit_add( | origin_visit1 = self.storage.origin_visit_add( | ||||
origin_id, | origin_id, | ||||
date=self.date_visit2) | date=self.date_visit2) | ||||
self.storage.origin_visit_upsert([{ | self.storage.origin_visit_upsert([{ | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
# then | # then | ||||
self.assertEqual(origin_visit1['origin'], origin_id) | self.assertEqual(origin_visit1['origin'], origin_id) | ||||
self.assertIsNotNone(origin_visit1['visit']) | self.assertIsNotNone(origin_visit1['visit']) | ||||
actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | ||||
self.assertEqual(actual_origin_visits, | self.assertEqual(actual_origin_visits, | ||||
[{ | [{ | ||||
'origin': origin_id, | 'origin': origin_id, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
}]) | }]) | ||||
expected_origin = self.origin2.copy() | expected_origin = self.origin2.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin2['type'], | |||||
'status': 'full', | 'status': 'full', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.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 92 Lines • ▼ Show 20 Lines | def test_snapshot_add_get_empty(self): | ||||
by_ov = self.storage.snapshot_get_by_origin_visit(origin_id, visit_id) | by_ov = self.storage.snapshot_get_by_origin_visit(origin_id, visit_id) | ||||
self.assertEqual(by_ov, self.empty_snapshot) | self.assertEqual(by_ov, self.empty_snapshot) | ||||
expected_origin = self.origin.copy() | expected_origin = self.origin.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.empty_snapshot['id'], | 'snapshot': self.empty_snapshot['id'], | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
('snapshot', self.empty_snapshot), | ('snapshot', self.empty_snapshot), | ||||
Show All 13 Lines | def test_snapshot_add_get_empty__legacy_add(self): | ||||
by_ov = self.storage.snapshot_get_by_origin_visit(origin_id, visit_id) | by_ov = self.storage.snapshot_get_by_origin_visit(origin_id, visit_id) | ||||
self.assertEqual(by_ov, self.empty_snapshot) | self.assertEqual(by_ov, self.empty_snapshot) | ||||
expected_origin = self.origin.copy() | expected_origin = self.origin.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.empty_snapshot['id'], | 'snapshot': self.empty_snapshot['id'], | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
('snapshot', self.empty_snapshot), | ('snapshot', self.empty_snapshot), | ||||
▲ Show 20 Lines • Show All 237 Lines • ▼ Show 20 Lines | def test_snapshot_add_twice(self): | ||||
visit2_id) | visit2_id) | ||||
self.assertEqual(by_ov2, self.snapshot) | self.assertEqual(by_ov2, self.snapshot) | ||||
expected_origin = self.origin.copy() | expected_origin = self.origin.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.snapshot['id'], | 'snapshot': self.snapshot['id'], | ||||
} | } | ||||
data3 = { | data3 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data4 = { | data4 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.snapshot['id'], | 'snapshot': self.snapshot['id'], | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
('snapshot', self.snapshot), | ('snapshot', self.snapshot), | ||||
Show All 22 Lines | def test_snapshot_add_twice__legacy_add(self): | ||||
visit2_id) | visit2_id) | ||||
self.assertEqual(by_ov2, self.snapshot) | self.assertEqual(by_ov2, self.snapshot) | ||||
expected_origin = self.origin.copy() | expected_origin = self.origin.copy() | ||||
data1 = { | data1 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data2 = { | data2 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit1, | 'date': self.date_visit1, | ||||
'visit': origin_visit1['visit'], | 'visit': origin_visit1['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.snapshot['id'], | 'snapshot': self.snapshot['id'], | ||||
} | } | ||||
data3 = { | data3 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': None, | 'snapshot': None, | ||||
} | } | ||||
data4 = { | data4 = { | ||||
'origin': expected_origin, | 'origin': expected_origin, | ||||
'date': self.date_visit2, | 'date': self.date_visit2, | ||||
'visit': origin_visit2['visit'], | 'visit': origin_visit2['visit'], | ||||
'type': self.origin['type'], | |||||
'status': 'ongoing', | 'status': 'ongoing', | ||||
'metadata': None, | 'metadata': None, | ||||
'snapshot': self.snapshot['id'], | 'snapshot': self.snapshot['id'], | ||||
} | } | ||||
self.assertEqual(list(self.journal_writer.objects), | self.assertEqual(list(self.journal_writer.objects), | ||||
[('origin', expected_origin), | [('origin', expected_origin), | ||||
('origin_visit', data1), | ('origin_visit', data1), | ||||
('snapshot', self.snapshot), | ('snapshot', self.snapshot), | ||||
▲ Show 20 Lines • Show All 1,158 Lines • Show Last 20 Lines |