53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
 
|
"""Namespaced plugin KV store round-trips (SPEC §11 plugin_kv)."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from sqlalchemy import delete
|
||
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
|
||
|
|
from sqlalchemy.pool import NullPool
|
||
|
|
|
||
|
|
from app.config import get_settings
|
||
|
|
from app.db import session_scope
|
||
|
|
from app.identity.passwords import hash_password
|
||
|
|
from app.models import User
|
||
|
|
from app.plugins.sdk import sdk
|
||
|
|
|
||
|
|
|
||
|
|
async def test_plugin_kv_crud_is_namespaced() -> None:
|
||
|
|
engine = create_async_engine(get_settings().database_url, poolclass=NullPool)
|
||
|
|
sm = async_sessionmaker(engine, expire_on_commit=False)
|
||
|
|
session = sm()
|
||
|
|
user = User(handle=f"kv{uuid.uuid4().hex[:8]}@test.local", password_hash=hash_password("pw"))
|
||
|
|
session.add(user)
|
||
|
|
await session.commit()
|
||
|
|
|
||
|
|
try:
|
||
|
|
kv = sdk("filters").kv(user.id)
|
||
|
|
other = sdk("labels").kv(user.id) # different plugin, same user
|
||
|
|
|
||
|
|
assert await kv.get("rules", default=[]) == [] # missing key -> default
|
||
|
|
await kv.set("rules", [{"match": "boss", "flag": "important"}])
|
||
|
|
await kv.set("count", 3)
|
||
|
|
assert await kv.get("rules") == [{"match": "boss", "flag": "important"}]
|
||
|
|
assert await kv.get("count") == 3
|
||
|
|
|
||
|
|
# Namespacing: the labels plugin sees nothing filters wrote.
|
||
|
|
assert await other.get("rules") is None
|
||
|
|
assert await other.items() == {}
|
||
|
|
|
||
|
|
assert await kv.items() == {"rules": [{"match": "boss", "flag": "important"}], "count": 3}
|
||
|
|
|
||
|
|
await kv.set("count", 4) # upsert overwrites
|
||
|
|
assert await kv.get("count") == 4
|
||
|
|
|
||
|
|
await kv.delete("count")
|
||
|
|
assert await kv.get("count") is None
|
||
|
|
assert set((await kv.items()).keys()) == {"rules"}
|
||
|
|
finally:
|
||
|
|
async with session_scope() as cleanup: # cascade drops plugin_kv rows
|
||
|
|
await cleanup.execute(delete(User).where(User.id == user.id))
|
||
|
|
await session.close()
|
||
|
|
await engine.dispose()
|