ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter4/collaboration-tools/test_load_expired_timer_callback.py

@@ -0,0 +1,58 @@
+import asyncio
+import json
+from datetime import datetime, timedelta
+from pathlib import Path
+import pytest
+import sys
+
+sys.path.insert(0, str(Path(__file__).parent / "src"))
+
+import timer_tools
+from config import config
+
+
+@pytest.mark.asyncio
+async def test_load_timers_expired_active_timer_triggers_callback():
+    storage = Path(config.timer.storage_path).expanduser()
+    storage.parent.mkdir(parents=True, exist_ok=True)
+
+    past_expiry = (datetime.now() - timedelta(seconds=10)).isoformat()
+    timer_id = "test-expired-timer-restore"
+    callback_fired = False
+
+    async def mock_callback(timer_data):
+        nonlocal callback_fired
+        callback_fired = True
+
+    timer_tools._active_timers.clear()
+    timer_tools._trigger_timer_callback = mock_callback
+
+    timer_data = {
+        "timer_id": timer_id,
+        "name": "Expired Active Timer",
+        "duration_seconds": 10,
+        "start_time": (datetime.now() - timedelta(seconds=20)).isoformat(),
+        "expiry_time": past_expiry,
+        "callback_message": "Timer expired message!",
+        "status": "active",
+        "created_at": (datetime.now() - timedelta(seconds=20)).isoformat()
+    }
+
+    with open(storage, "w") as f:
+        json.dump({timer_id: timer_data}, f)
+
+    await timer_tools._load_timers()
+
+    loaded = timer_tools._active_timers.get(timer_id)
+    assert loaded is not None
+    assert loaded["status"] == "expired"
+    assert loaded.get("completed_at") is not None
+    assert callback_fired is True
+
+    with open(storage, "r") as f:
+        on_disk = json.load(f)
+    assert on_disk[timer_id]["status"] == "expired"
+
+
+if __name__ == "__main__":
+    asyncio.run(test_load_timers_expired_active_timer_triggers_callback())