Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
| Show First 20 Lines • Show All 1,505 Lines • ▼ Show 20 Lines | def test_origin_visit_get_by(self): | ||||
| # 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']) | ||||
| # then | # then | ||||
| self.assertEqual(actual_origin_visit1, expected_origin_visit) | self.assertEqual(actual_origin_visit1, expected_origin_visit) | ||||
| def test_origin_visit_upsert_new(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 | |||||
| self.storage.origin_visit_upsert([{ | |||||
| 'origin': origin_id, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': 123, | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| }]) | |||||
| # then | |||||
| actual_origin_visits = list(self.storage.origin_visit_get(origin_id)) | |||||
| self.assertEqual(actual_origin_visits, | |||||
| [{ | |||||
| 'origin': origin_id, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': 123, | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| }]) | |||||
| expected_origin = self.origin2.copy() | |||||
| data = { | |||||
| 'origin': expected_origin, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': 123, | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| } | |||||
| self.assertEqual(list(self.journal_writer.objects), | |||||
| [('origin', expected_origin), | |||||
| ('origin_visit', data)]) | |||||
| def test_origin_visit_upsert_existing(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) | |||||
| self.storage.origin_visit_upsert([{ | |||||
| 'origin': origin_id, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': origin_visit1['visit'], | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| }]) | |||||
| # then | |||||
| self.assertEqual(origin_visit1['origin'], origin_id) | |||||
| self.assertIsNotNone(origin_visit1['visit']) | |||||
| 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'], | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| }]) | |||||
| expected_origin = self.origin2.copy() | |||||
| data1 = { | |||||
| 'origin': expected_origin, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': origin_visit1['visit'], | |||||
| 'status': 'ongoing', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| } | |||||
| data2 = { | |||||
| 'origin': expected_origin, | |||||
| 'date': self.date_visit2, | |||||
| 'visit': origin_visit1['visit'], | |||||
| 'status': 'full', | |||||
| 'metadata': None, | |||||
| 'snapshot': None, | |||||
| } | |||||
| self.assertEqual(list(self.journal_writer.objects), | |||||
| [('origin', expected_origin), | |||||
| ('origin_visit', data1), | |||||
| ('origin_visit', data2)]) | |||||
| def test_origin_visit_get_by_no_result(self): | def test_origin_visit_get_by_no_result(self): | ||||
| # No result | # No result | ||||
| actual_origin_visit = self.storage.origin_visit_get_by( | actual_origin_visit = self.storage.origin_visit_get_by( | ||||
| 10, 999) | 10, 999) | ||||
| self.assertIsNone(actual_origin_visit) | self.assertIsNone(actual_origin_visit) | ||||
| def test_person_get(self): | def test_person_get(self): | ||||
| ▲ Show 20 Lines • Show All 1,495 Lines • Show Last 20 Lines | |||||