mirror of
https://github.com/hoshikawa2/agent_platform_oci.git
synced 2026-07-09 14:04:19 +00:00
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
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")
|