wehub-resource-sync
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
"""Session delete purges agent_tasks and memory_durable_receipts rows."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from opensquilla.session.models import (
|
|
AgentTaskRecord,
|
|
AgentTaskStatus,
|
|
MemoryDurableReceipt,
|
|
SessionNode,
|
|
)
|
|
from opensquilla.session.storage import SessionStorage
|
|
|
|
KEY = "agent:main:webchat:default"
|
|
|
|
|
|
async def test_delete_session_purges_agent_tasks(tmp_path) -> None:
|
|
storage = SessionStorage(str(tmp_path / "sessions.db"))
|
|
await storage.connect()
|
|
try:
|
|
await storage.upsert_session(SessionNode(session_key=KEY, session_id="sid-old"))
|
|
await storage.create_agent_task(
|
|
AgentTaskRecord(
|
|
task_id="task-1",
|
|
session_key=KEY,
|
|
source_kind="webui",
|
|
queue_mode="followup",
|
|
run_kind="web_turn",
|
|
status=AgentTaskStatus.RUNNING,
|
|
)
|
|
)
|
|
|
|
await storage.delete_session(KEY)
|
|
|
|
assert await storage.get_session(KEY) is None
|
|
assert await storage.list_agent_tasks(session_key=KEY) == []
|
|
finally:
|
|
await storage.close()
|
|
|
|
|
|
async def test_delete_session_purges_memory_durable_receipts(tmp_path) -> None:
|
|
storage = SessionStorage(str(tmp_path / "sessions.db"))
|
|
await storage.connect()
|
|
try:
|
|
await storage.upsert_session(SessionNode(session_key=KEY, session_id="sid-old"))
|
|
await storage.upsert_memory_durable_receipt(
|
|
MemoryDurableReceipt(
|
|
session_key=KEY,
|
|
session_id="sid-old",
|
|
scope="checkpoint",
|
|
source_path="memory/2026-01-01.md",
|
|
idempotency_key="idem-1",
|
|
status="failed",
|
|
reason="disk full",
|
|
)
|
|
)
|
|
|
|
await storage.delete_session(KEY)
|
|
|
|
assert await storage.get_session(KEY) is None
|
|
assert await storage.list_memory_durable_receipts(session_key=KEY) == []
|
|
finally:
|
|
await storage.close()
|
|
|
|
|
|
async def test_recreated_session_key_does_not_inherit_deleted_tasks(tmp_path) -> None:
|
|
storage = SessionStorage(str(tmp_path / "sessions.db"))
|
|
await storage.connect()
|
|
try:
|
|
await storage.upsert_session(SessionNode(session_key=KEY, session_id="sid-old"))
|
|
await storage.create_agent_task(
|
|
AgentTaskRecord(
|
|
task_id="task-1",
|
|
session_key=KEY,
|
|
source_kind="webui",
|
|
queue_mode="followup",
|
|
run_kind="web_turn",
|
|
status=AgentTaskStatus.FAILED,
|
|
terminal_reason="error",
|
|
)
|
|
)
|
|
await storage.delete_session(KEY)
|
|
|
|
await storage.upsert_session(SessionNode(session_key=KEY, session_id="sid-new"))
|
|
|
|
grouped = await storage.list_agent_tasks_for_sessions([KEY])
|
|
assert grouped[KEY] == []
|
|
finally:
|
|
await storage.close()
|