First commit

tests/unit/test_resilient_checkpointer.py

@@ -0,0 +1,48 @@
+import pytest
+
+from agent_framework.checkpoints.checkpoint_repository import (
+    CheckpointRecoveryError,
+    InMemoryCheckpointRepository,
+    ResilientCheckpointRepository,
+)
+
+
+@pytest.mark.asyncio
+async def test_integrity_envelope_and_recovery_skips_corrupt_latest():
+    raw = InMemoryCheckpointRepository()
+    repo = ResilientCheckpointRepository(raw, compact_every=100, keep_last=10, recovery_scan_limit=5)
+
+    await repo.put("thread-1", {"checkpoint_id": "ok-1", "checkpoint": {"id": "ok-1", "value": 1}})
+    await repo.put("thread-1", {"checkpoint_id": "ok-2", "checkpoint": {"id": "ok-2", "value": 2}})
+
+    # Simula corrupção no último registro persistido.
+    raw._data["thread-1"][-1]["payload"]["checkpoint"]["value"] = 999
+
+    recovered = await repo.get_latest("thread-1")
+    assert recovered["checkpoint_id"] == "ok-1"
+    assert recovered["checkpoint"]["value"] == 1
+
+
+@pytest.mark.asyncio
+async def test_compaction_keeps_last_n_checkpoints():
+    raw = InMemoryCheckpointRepository()
+    repo = ResilientCheckpointRepository(raw, compact_every=1, keep_last=3, recovery_scan_limit=10)
+
+    for i in range(7):
+        await repo.put("thread-compact", {"checkpoint_id": f"cp-{i}", "checkpoint": {"id": f"cp-{i}"}})
+
+    assert len(raw._data["thread-compact"]) <= 3
+    latest = await repo.get_latest("thread-compact")
+    assert latest["checkpoint_id"] == "cp-6"
+
+
+@pytest.mark.asyncio
+async def test_recovery_raises_when_only_corrupt_checkpoints_exist():
+    raw = InMemoryCheckpointRepository()
+    repo = ResilientCheckpointRepository(raw, compact_every=100, keep_last=10, recovery_scan_limit=5)
+
+    await repo.put("thread-bad", {"checkpoint_id": "bad", "checkpoint": {"id": "bad", "value": 1}})
+    raw._data["thread-bad"][-1]["payload"]["checkpoint"]["value"] = 2
+
+    with pytest.raises(CheckpointRecoveryError):
+        await repo.get_latest("thread-bad")