import json import httpx import pytest from qonto_assistant.flex_auth_client import CheckRequest, FlexAuthCheckClient, new_request_id def _request() -> CheckRequest: return CheckRequest( request_id=new_request_id(), tenant="tenant:friendly:binky", subject_id="agent-harness-binky", subject_type="agent", ) def _client(handler) -> FlexAuthCheckClient: return FlexAuthCheckClient( base_url="https://flex-auth.example.test", timeout_seconds=1, transport=httpx.MockTransport(handler), ) def test_allow_effect_authorizes() -> None: def handler(request: httpx.Request) -> httpx.Response: return httpx.Response(200, json={"id": "d-1", "effect": "allow", "resource": {}, "subject": {}, "provenance": {}}) assert _client(handler).is_allowed(_request()) is True @pytest.mark.parametrize("effect", ["deny", "redact", "audit_only", "not_applicable"]) def test_non_allow_effects_deny(effect: str) -> None: def handler(request: httpx.Request) -> httpx.Response: return httpx.Response(200, json={"id": "d-1", "effect": effect, "resource": {}, "subject": {}, "provenance": {}}) assert _client(handler).is_allowed(_request()) is False def test_non_200_status_denies() -> None: def handler(request: httpx.Request) -> httpx.Response: return httpx.Response(500, json={"error": "internal"}) assert _client(handler).is_allowed(_request()) is False def test_malformed_json_body_denies() -> None: def handler(request: httpx.Request) -> httpx.Response: return httpx.Response(200, content=b"not json") assert _client(handler).is_allowed(_request()) is False def test_non_object_json_body_denies() -> None: def handler(request: httpx.Request) -> httpx.Response: return httpx.Response(200, json=["not", "an", "object"]) assert _client(handler).is_allowed(_request()) is False def test_connection_failure_denies() -> None: def handler(request: httpx.Request) -> httpx.Response: raise httpx.ConnectError("connection refused", request=request) assert _client(handler).is_allowed(_request()) is False def test_timeout_denies() -> None: def handler(request: httpx.Request) -> httpx.Response: raise httpx.TimeoutException("timed out", request=request) assert _client(handler).is_allowed(_request()) is False def test_request_body_matches_schema_shape() -> None: seen: dict[str, object] = {} def handler(request: httpx.Request) -> httpx.Response: seen.update(json.loads(request.content)) return httpx.Response(200, json={"id": "d-1", "effect": "allow", "resource": {}, "subject": {}, "provenance": {}}) _client(handler).is_allowed(_request()) assert seen["tenant"] == "tenant:friendly:binky" assert seen["action"] == "finance.qonto.read" assert seen["subject"] == {"id": "agent-harness-binky", "type": "agent"} assert seen["resource"] == { "id": "finance-snapshot", "type": "finance-snapshot", "system": "qonto-assistant", }