- platform/interface.py: PlatformClient Protocol + Pydantic models (User, MessageResponse, UserSettings) — no explicit session management, Master handles container lifecycle - platform/mock.py: MockPlatformClient with simulated latency, [MOCK] responses, is_new correctly True only on first creation - core/protocol.py: unified dataclasses for all events and responses (IncomingMessage/Command/Callback, OutgoingMessage/UI/Notification, AuthFlow, ChatContext, SettingsAction, etc.) - core/store.py: StateStore Protocol + InMemoryStore (tests) + SQLiteStore (prod) with JSON serialization - core/chat.py: ChatManager — chat metadata (C1/C2/C3), not container lifecycle (that's the platform's job) - core/auth.py: AuthManager — start_flow / confirm / is_authenticated - core/settings.py: SettingsManager — get/apply with store cache - core/handler.py: EventDispatcher — registry-based routing with keys (command name, action name, attachment type, "*" catch-all) - core/handlers/: register_all() + start/new/message/callback/settings handlers; voice slot falls back to stub text until voice_handler added - conftest.py: sys.path fix so local platform/ shadows stdlib platform - docs/api-contract.md: rewritten for Lambda Lab 3.0 container model 46 tests passing, 0 warnings.
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
# tests/core/test_store.py
|
|
from core.store import InMemoryStore, SQLiteStore
|
|
|
|
|
|
async def test_inmemory_get_missing_returns_none():
|
|
store = InMemoryStore()
|
|
assert await store.get("missing") is None
|
|
|
|
|
|
async def test_inmemory_set_and_get():
|
|
store = InMemoryStore()
|
|
await store.set("k", {"x": 1})
|
|
assert await store.get("k") == {"x": 1}
|
|
|
|
|
|
async def test_inmemory_delete():
|
|
store = InMemoryStore()
|
|
await store.set("k", {"x": 1})
|
|
await store.delete("k")
|
|
assert await store.get("k") is None
|
|
|
|
|
|
async def test_inmemory_keys_prefix():
|
|
store = InMemoryStore()
|
|
await store.set("chat:u1:C1", {"a": 1})
|
|
await store.set("chat:u1:C2", {"b": 2})
|
|
await store.set("auth:u1", {"c": 3})
|
|
keys = await store.keys("chat:u1:")
|
|
assert set(keys) == {"chat:u1:C1", "chat:u1:C2"}
|
|
|
|
|
|
async def test_sqlite_set_and_get(tmp_path):
|
|
store = SQLiteStore(str(tmp_path / "test.db"))
|
|
await store.set("k", {"hello": "world"})
|
|
assert await store.get("k") == {"hello": "world"}
|
|
|
|
|
|
async def test_sqlite_overwrite(tmp_path):
|
|
store = SQLiteStore(str(tmp_path / "test.db"))
|
|
await store.set("k", {"v": 1})
|
|
await store.set("k", {"v": 2})
|
|
assert await store.get("k") == {"v": 2}
|
|
|
|
|
|
async def test_sqlite_delete(tmp_path):
|
|
store = SQLiteStore(str(tmp_path / "test.db"))
|
|
await store.set("k", {"v": 1})
|
|
await store.delete("k")
|
|
assert await store.get("k") is None
|
|
|
|
|
|
async def test_sqlite_keys_prefix(tmp_path):
|
|
store = SQLiteStore(str(tmp_path / "test.db"))
|
|
await store.set("chat:u1:C1", {})
|
|
await store.set("chat:u1:C2", {})
|
|
await store.set("auth:u1", {})
|
|
keys = await store.keys("chat:u1:")
|
|
assert set(keys) == {"chat:u1:C1", "chat:u1:C2"}
|