34 lines
978 B
Python
34 lines
978 B
Python
|
"""Rights registry and DB sync tests."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import rights_registrations # noqa: F401
|
||
|
|
from db import get_connection
|
||
|
|
from rights_registry import get_registered_capabilities, sync_rights_registry_to_db
|
||
|
|
|
||
|
|
|
||
|
|
def test_registry_contains_initial_capabilities():
|
||
|
|
keys = {reg.key for reg in get_registered_capabilities()}
|
||
|
|
assert keys == {
|
||
|
|
"kairo.admin.access",
|
||
|
|
"kairo.tenant.manage",
|
||
|
|
"kairo.actor.manage",
|
||
|
|
"kairo.context.read",
|
||
|
|
"kairo.entitlements.read",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def test_sync_is_idempotent():
|
||
|
|
assert sync_rights_registry_to_db() == 0
|
||
|
|
assert sync_rights_registry_to_db() == 0
|
||
|
|
|
||
|
|
conn = get_connection()
|
||
|
|
try:
|
||
|
|
with conn.cursor() as cur:
|
||
|
|
cur.execute("SELECT COUNT(*) FROM capabilities")
|
||
|
|
assert cur.fetchone()[0] == 5
|
||
|
|
cur.execute("SELECT COUNT(*) FROM role_capability_grants")
|
||
|
|
assert cur.fetchone()[0] >= 5
|
||
|
|
finally:
|
||
|
|
conn.close()
|