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 @@ -3031,6 +3031,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() @@ -3038,8 +3039,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()