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 @@ -574,6 +574,21 @@ self.assertEqual(journal_objects, [('content', expected_cont)]) + def test_content_get_missing(self): + cont = self.cont + + self.storage.content_add([cont]) + + results = list(self.storage.content_get( + [self.cont['sha1'], self.cont2['sha1']])) + self.assertEqual(results, + [{'sha1': cont['sha1'], 'data': cont['data']}, None]) + + results = list(self.storage.content_get( + [self.cont2['sha1'], self.cont['sha1']])) + self.assertEqual(results, + [None, {'sha1': cont['sha1'], 'data': cont['data']}]) + def test_content_add_same_input(self): cont = self.cont