diff --git a/swh/loader/core/loader.py b/swh/loader/core/loader.py --- a/swh/loader/core/loader.py +++ b/swh/loader/core/loader.py @@ -170,7 +170,7 @@ """ assert self.origin - self.storage.origin_add_one(self.origin) + self.storage.origin_add([self.origin]) if not self.visit_date: # now as default visit_date if not provided self.visit_date = datetime.datetime.now(tz=datetime.timezone.utc) diff --git a/swh/loader/core/tests/test_loader.py b/swh/loader/core/tests/test_loader.py --- a/swh/loader/core/tests/test_loader.py +++ b/swh/loader/core/tests/test_loader.py @@ -34,9 +34,9 @@ self.origin_url = ORIGIN.url self.visit_date = datetime.datetime.now(tz=datetime.timezone.utc) self.visit_type = "git" - origin_url = self.storage.origin_add_one(ORIGIN) + self.storage.origin_add([ORIGIN]) visit = OriginVisit( - origin=origin_url, date=self.visit_date, type=self.visit_type, + origin=self.origin_url, date=self.visit_date, type=self.visit_type, ) self.visit = self.storage.origin_visit_add([visit])[0] diff --git a/swh/loader/package/loader.py b/swh/loader/package/loader.py --- a/swh/loader/package/loader.py +++ b/swh/loader/package/loader.py @@ -299,7 +299,7 @@ # Prepare origin and origin_visit origin = Origin(url=self.url) try: - self.storage.origin_add_one(origin) + self.storage.origin_add([origin]) visit = self.storage.origin_visit_add( [ OriginVisit( diff --git a/swh/loader/package/tests/test_loader.py b/swh/loader/package/tests/test_loader.py --- a/swh/loader/package/tests/test_loader.py +++ b/swh/loader/package/tests/test_loader.py @@ -7,7 +7,7 @@ class FakeStorage: - def origin_add_one(self, origin): + def origin_add(self, origins): raise ValueError("We refuse to add an origin") def origin_visit_get_latest(self, origin): @@ -15,8 +15,8 @@ class FakeStorage2(FakeStorage): - def origin_add_one(self, origin): - return origin + def origin_add(self, origins): + pass def origin_visit_add(self, visits): raise ValueError("We refuse to add an origin visit")