Page MenuHomeSoftware Heritage
Files F7124411

D3530.id12467.diff
No OneTemporary
Actions

D3530.id12467.diff
View Options

diff --git a/swh/storage/algos/origin.py b/swh/storage/algos/origin.py
--- a/swh/storage/algos/origin.py
+++ b/swh/storage/algos/origin.py
@@ -4,7 +4,7 @@
# See top-level LICENSE file for more information
from typing import Any, Dict, Optional, Iterable, Tuple
-from swh.model.model import OriginVisit, OriginVisitStatus
+from swh.model.model import Origin, OriginVisit, OriginVisitStatus
def iter_origins(storage, origin_from=1, origin_to=None, batch_size=10000):
@@ -33,7 +33,7 @@
start = origins[-1]["id"] + 1
for origin in origins:
del origin["id"]
- yield origin
+ yield Origin.from_dict(origin)
if origin_to and start > origin_to:
break
diff --git a/swh/storage/tests/algos/test_origin.py b/swh/storage/tests/algos/test_origin.py
--- a/swh/storage/tests/algos/test_origin.py
+++ b/swh/storage/tests/algos/test_origin.py
@@ -22,14 +22,16 @@
@pytest.fixture
def swh_storage_backend_config():
- yield {"cls": "validate", "storage": {"cls": "memory",}}
+ yield {
+ "cls": "memory",
+ }
def test_iter_origins(swh_storage):
origins = [
- {"url": "bar"},
- {"url": "qux"},
- {"url": "quuz"},
+ Origin(url="bar"),
+ Origin(url="qux"),
+ Origin(url="quuz"),
]
assert swh_storage.origin_add(origins) == {"origin:add": 3}
assert_list_eq(iter_origins(swh_storage), origins)
@@ -82,7 +84,7 @@
mock_origin_get_range.assert_called_with(origin_from=1, origin_count=42)
-def test_origin_get_latest_visit_status_none(swh_storage):
+def test_origin_get_latest_visit_status_none(swh_storage, sample_data_model):
"""Looking up unknown objects should return nothing
"""
@@ -90,11 +92,13 @@
assert origin_get_latest_visit_status(swh_storage, "unknown-origin") is None
# unknown type so no result
- origin = Origin.from_dict(data.origin)
+ origin = sample_data_model["origin"][0]
+ origin_visit = sample_data_model["origin_visit"][0]
+ assert origin_visit.origin == origin.url
+
swh_storage.origin_add_one(origin)
- swh_storage.origin_visit_add(
- [OriginVisit(origin=origin.url, date=data.date_visit1, type="git",),]
- )[0]
+ swh_storage.origin_visit_add([origin_visit])[0]
+ assert origin_visit.type != "unknown"
actual_origin_visit = origin_get_latest_visit_status(
swh_storage, origin.url, type="unknown"
)

File Metadata

Mime Type
text/plain
Expires
Dec 21 2024, 10:31 AM (11 w, 4 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3232562

Event Timeline

About · Developer information · Legal