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 @@ -1189,6 +1189,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'] @@ -1274,6 +1277,10 @@ for visit in visits: origin_id = visit['origin'] visit_id = visit['visit'] + + self._objects[(origin_id, visit_id)].append( + ('origin_visit', None)) + while len(self._origin_visits[origin_id-1]) < visit_id: self._origin_visits[origin_id-1].append(None) visit = self._origin_visits[origin_id-1][visit_id-1] = visit 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 @@ -3038,6 +3038,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)