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 @@ -2420,6 +2420,7 @@ self.snapshot) self.storage.directory_add([self.dir]) self.storage.revision_add([self.revision]) + self.storage.release_add([self.release]) self.storage.refresh_stat_counters() counters = self.storage.stat_counters() @@ -2427,8 +2428,11 @@ 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) + self.assertEqual(counters['release'], 1) + self.assertEqual(counters['snapshot'], 1) + self.assertEqual(counters['person'], 3) def test_content_find_ctime(self): cont = self.cont.copy()