diff --git a/swh/storage/in_memory.py b/swh/storage/in_memory.py --- a/swh/storage/in_memory.py +++ b/swh/storage/in_memory.py @@ -1180,6 +1180,9 @@ 'visit': visit_id, } + self._objects[(origin_id, visit_id)].append( + ('origin_visit', None)) + if self.journal_writer: origin = self.origin_get([{'id': origin_id}])[0] del origin['id'] @@ -1255,6 +1258,10 @@ for visit in visits: origin_id = visit['origin'] visit_id = visit['visit'] + + self._objects[(origin_id, visit_id)].append( + ('origin_visit', None)) + if isinstance(visit['date'], str): visit['date'] = dateutil.parser.parse(visit['date']) while len(self._origin_visits[origin_id-1]) < visit_id: diff --git a/swh/storage/tests/test_storage.py b/swh/storage/tests/test_storage.py --- a/swh/storage/tests/test_storage.py +++ b/swh/storage/tests/test_storage.py @@ -2427,6 +2427,7 @@ self.assertEqual(counters['directory'], 1) self.assertEqual(counters['snapshot'], 1) self.assertEqual(counters['origin'], 1) + self.assertEqual(counters['origin_visit'], 1) self.assertEqual(counters['revision'], 1) self.assertEqual(counters['person'], 2)