101 lines
3 KiB
Python
101 lines
3 KiB
Python
|
from datetime import UTC, datetime
|
||
|
|
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
|
||
|
|
from tenant_engine.app import create_app
|
||
|
|
from tenant_engine.domain import CapabilityRole, Tenant, create_role_grant
|
||
|
|
from tenant_engine.store import InMemoryTenantStore, TenantStore
|
||
|
|
|
||
|
|
|
||
|
|
class _BrokenStore:
|
||
|
|
"""Test double: every active_roles() call raises, simulating an outage."""
|
||
|
|
|
||
|
|
def __init__(self, delegate: TenantStore) -> None:
|
||
|
|
self._delegate = delegate
|
||
|
|
|
||
|
|
def create_tenant(self, tenant):
|
||
|
|
return self._delegate.create_tenant(tenant)
|
||
|
|
|
||
|
|
def get_tenant(self, tenant_id):
|
||
|
|
return self._delegate.get_tenant(tenant_id)
|
||
|
|
|
||
|
|
def grant_role(self, grant):
|
||
|
|
return self._delegate.grant_role(grant)
|
||
|
|
|
||
|
|
def revoke_role(self, **kwargs):
|
||
|
|
return self._delegate.revoke_role(**kwargs)
|
||
|
|
|
||
|
|
def active_roles(self, tenant_id):
|
||
|
|
from tenant_engine.store import StoreUnavailableError
|
||
|
|
|
||
|
|
raise StoreUnavailableError("simulated outage")
|
||
|
|
|
||
|
|
def assign_plan(self, assignment):
|
||
|
|
return self._delegate.assign_plan(assignment)
|
||
|
|
|
||
|
|
def events(self):
|
||
|
|
return self._delegate.events()
|
||
|
|
|
||
|
|
|
||
|
|
def _seeded_store() -> InMemoryTenantStore:
|
||
|
|
store = InMemoryTenantStore()
|
||
|
|
tenant = Tenant.create(tenant_id="t-binky", identifier="tenant:friendly:binky")
|
||
|
|
store.create_tenant(tenant)
|
||
|
|
store.grant_role(
|
||
|
|
create_role_grant(
|
||
|
|
tenant=tenant,
|
||
|
|
grant_id="g-1",
|
||
|
|
role=CapabilityRole.CUS,
|
||
|
|
grant_reason="manual_grant",
|
||
|
|
plan_id=None,
|
||
|
|
granted_by="ops",
|
||
|
|
correlation_id="corr-1",
|
||
|
|
granted_at=datetime.now(UTC),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
return store
|
||
|
|
|
||
|
|
|
||
|
|
def test_cache_read_roles_returns_active_roles() -> None:
|
||
|
|
client = TestClient(create_app(store=_seeded_store()))
|
||
|
|
response = client.get("/tenants/t-binky/roles")
|
||
|
|
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.json() == {"tenant_id": "t-binky", "roles": ["CUS"]}
|
||
|
|
|
||
|
|
|
||
|
|
def test_cache_read_roles_unknown_tenant_is_404() -> None:
|
||
|
|
client = TestClient(create_app(store=_seeded_store()))
|
||
|
|
response = client.get("/tenants/does-not-exist/roles")
|
||
|
|
|
||
|
|
assert response.status_code == 404
|
||
|
|
|
||
|
|
|
||
|
|
def test_live_lookup_roles_returns_active_roles() -> None:
|
||
|
|
client = TestClient(create_app(store=_seeded_store()))
|
||
|
|
response = client.get("/tenants/t-binky/roles/live")
|
||
|
|
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.json()["roles"] == ["CUS"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_live_lookup_fails_closed_on_store_outage() -> None:
|
||
|
|
broken = _BrokenStore(_seeded_store())
|
||
|
|
client = TestClient(create_app(store=broken))
|
||
|
|
|
||
|
|
response = client.get("/tenants/t-binky/roles/live")
|
||
|
|
|
||
|
|
assert response.status_code == 503
|
||
|
|
assert response.json() != {"tenant_id": "t-binky", "roles": []}, (
|
||
|
|
"outage must not be indistinguishable from a legitimate empty role list"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_cache_read_also_fails_closed_on_store_outage() -> None:
|
||
|
|
broken = _BrokenStore(_seeded_store())
|
||
|
|
client = TestClient(create_app(store=broken))
|
||
|
|
|
||
|
|
response = client.get("/tenants/t-binky/roles")
|
||
|
|
|
||
|
|
assert response.status_code == 503
|