diff --git a/swh/indexer/tests/test_origin_head.py b/swh/indexer/tests/test_origin_head.py --- a/swh/indexer/tests/test_origin_head.py +++ b/swh/indexer/tests/test_origin_head.py @@ -52,9 +52,7 @@ def test_git_partial_snapshot(self): """Checks partial snapshots are ignored.""" origin_url = "https://github.com/SoftwareHeritage/swh-core" - self.indexer.storage.origin_add_one( - {"url": origin_url,} - ) + self.indexer.storage.origin_add([{"url": origin_url,}]) visit = self.indexer.storage.origin_visit_add( [ OriginVisit( @@ -95,9 +93,7 @@ def test_pypi_missing_branch(self): origin_url = "https://pypi.org/project/abcdef/" - self.indexer.storage.origin_add_one( - {"url": origin_url,} - ) + self.indexer.storage.origin_add([{"url": origin_url,}]) visit = self.indexer.storage.origin_visit_add( [ OriginVisit( diff --git a/swh/indexer/tests/utils.py b/swh/indexer/tests/utils.py --- a/swh/indexer/tests/utils.py +++ b/swh/indexer/tests/utils.py @@ -505,7 +505,7 @@ def fill_storage(storage): visit_types = {} for visit in ORIGIN_VISITS: - storage.origin_add_one({"url": visit["url"]}) + storage.origin_add([{"url": visit["url"]}]) visit_types[visit["url"]] = visit["type"] for snap in SNAPSHOTS: origin_url = snap["origin"]