test: cover operation run + bookmark parsing

tests/test_operations.py

@@ -0,0 +1,98 @@
+import time
+
+import pytest
+
+from skywipe.operations import Operation, BookmarkStrategy
+
+
+class FakeClient:
+    class Me:
+        did = "did:plc:fake"
+
+    me = Me()
+
+
+class FakeResponse:
+    def __init__(self, items, cursor=None):
+        self.items = items
+        self.cursor = cursor
+
+
+class FakeStrategy:
+    def __init__(self, responses, fail_on=None):
+        self._responses = list(responses)
+        self._fail_on = fail_on
+
+    def fetch(self, context, cursor=None):
+        return self._responses.pop(0)
+
+    def extract_items(self, response):
+        return response.items
+
+    def process_item(self, item, context):
+        if self._fail_on is not None and item == self._fail_on:
+            raise ValueError("boom")
+
+    def get_cursor(self, response):
+        return response.cursor
+
+
+def test_operation_run_batches_filters_and_sleeps(monkeypatch):
+    responses = [
+        FakeResponse(items=[1, 2, 3], cursor="next"),
+        FakeResponse(items=[4], cursor=None),
+    ]
+    operation = Operation(
+        "Testing",
+        strategy_type="feed",
+        client=FakeClient(),
+        config_data={"batch_size": 2, "delay": 1},
+        filter_fn=lambda item: item != 2,
+    )
+    operation.strategy = FakeStrategy(responses, fail_on=3)
+
+    slept = []
+
+    def fake_sleep(seconds):
+        slept.append(seconds)
+
+    monkeypatch.setattr(time, "sleep", fake_sleep)
+
+    total = operation.run()
+
+    assert total == 2
+    assert slept == [1]
+
+
+def test_bookmark_strategy_extracts_uri_from_shapes():
+    strategy = BookmarkStrategy()
+
+    class Obj:
+        pass
+
+    direct = Obj()
+    direct.uri = "direct"
+    assert strategy._extract_bookmark_uri(direct) == "direct"
+
+    subject = Obj()
+    subject.subject = Obj()
+    subject.subject.uri = "subject"
+    assert strategy._extract_bookmark_uri(subject) == "subject"
+
+    record = Obj()
+    record.record = Obj()
+    record.record.uri = "record"
+    assert strategy._extract_bookmark_uri(record) == "record"
+
+    post = Obj()
+    post.post = Obj()
+    post.post.uri = "post"
+    assert strategy._extract_bookmark_uri(post) == "post"
+
+    item = Obj()
+    item.item = Obj()
+    item.item.uri = "item"
+    assert strategy._extract_bookmark_uri(item) == "item"
+
+    missing = Obj()
+    assert strategy._extract_bookmark_uri(missing) is None