"""Step definitions for ``server_lifecycle.feature``. Tests the ASGI application factory, health/agent-card endpoints, A2A JSON-RPC dispatch, ServerLifecycle construction and shutdown, and Settings-based server configuration. """ from __future__ import annotations from typing import Any from unittest.mock import MagicMock from behave import given, then, use_step_matcher, when # type: ignore[import-untyped] from cleveragents.a2a.facade import A2aLocalFacade from cleveragents.config.settings import Settings from cleveragents.infrastructure.server.asgi_app import create_asgi_app from cleveragents.infrastructure.server.server_lifecycle import ServerLifecycle # --------------------------------------------------------------------------- # ASGI application creation # --------------------------------------------------------------------------- @when("I create an ASGI application with default settings") def step_create_asgi_default(context: Any) -> None: context.asgi_app = create_asgi_app() @given("an A2aLocalFacade with no services") def step_given_facade(context: Any) -> None: context.facade = A2aLocalFacade() @when("I create an ASGI application with that facade") def step_create_asgi_with_facade(context: Any) -> None: context.asgi_app = create_asgi_app(facade=context.facade) @then("the ASGI app should be a FastAPI instance") def step_check_fastapi_instance(context: Any) -> None: from fastapi import FastAPI assert isinstance(context.asgi_app, FastAPI), ( f"Expected FastAPI, got {type(context.asgi_app)}" ) @when('I try to create an ASGI application with facade "{value}"') def step_create_asgi_bad_facade(context: Any, value: str) -> None: context.caught_exception = None try: create_asgi_app(facade=value) # type: ignore[arg-type] except TypeError as exc: context.caught_exception = exc @then("a TypeError should be raised for invalid facade") def step_check_type_error_facade(context: Any) -> None: assert context.caught_exception is not None, "Expected TypeError" assert isinstance(context.caught_exception, TypeError) @when("I try to create an ASGI application with port {port:d}") def step_create_asgi_bad_port(context: Any, port: int) -> None: context.caught_exception = None try: create_asgi_app(port=port) except ValueError as exc: context.caught_exception = exc @then("a ValueError should be raised for invalid port") def step_check_value_error_port(context: Any) -> None: assert context.caught_exception is not None, "Expected ValueError" assert isinstance(context.caught_exception, ValueError) use_step_matcher("re") @when('I try to create an ASGI application with host "(?P[^"]*)"') def step_create_asgi_bad_host(context: Any, host: str) -> None: context.caught_exception = None try: create_asgi_app(host=host) except ValueError as exc: context.caught_exception = exc use_step_matcher("parse") @then("a ValueError should be raised for invalid host") def step_check_value_error_host(context: Any) -> None: assert context.caught_exception is not None, "Expected ValueError" assert isinstance(context.caught_exception, ValueError) # --------------------------------------------------------------------------- # Test client helpers # --------------------------------------------------------------------------- @given("a running ASGI test client") def step_given_test_client(context: Any) -> None: from fastapi.testclient import TestClient app = create_asgi_app() context.test_client = TestClient(app) @when("I request GET /health") def step_get_health(context: Any) -> None: context.response = context.test_client.get("/health") @when("I request GET /.well-known/agent.json") def step_get_agent_card(context: Any) -> None: context.response = context.test_client.get("/.well-known/agent.json") @when('I POST a JSON-RPC request to /a2a with operation "{operation}"') def step_post_a2a(context: Any, operation: str) -> None: payload = {"operation": operation, "params": {}} context.response = context.test_client.post("/a2a", json=payload) @when("I POST a malformed JSON body to /a2a") def step_post_a2a_malformed(context: Any) -> None: # Send a body that cannot be parsed into A2aRequest (missing operation) context.response = context.test_client.post("/a2a", json={"bad": "data"}) @then("the response status code should be {code:d}") def step_check_status_code(context: Any, code: int) -> None: assert context.response.status_code == code, ( f"Expected {code}, got {context.response.status_code}: {context.response.text}" ) @then('the response body should contain "{text}"') def step_check_response_body(context: Any, text: str) -> None: body = context.response.text assert text in body, f"Expected '{text}' in response body: {body}" @then('the A2A response status should be "{expected_status}"') def step_check_a2a_status(context: Any, expected_status: str) -> None: data = context.response.json() assert data.get("status") == expected_status, ( f"Expected A2A status '{expected_status}', got {data}" ) # --------------------------------------------------------------------------- # ServerLifecycle construction # --------------------------------------------------------------------------- @when('I create a ServerLifecycle with host "{host}" and port {port:d}') def step_create_lifecycle(context: Any, host: str, port: int) -> None: context.lifecycle = ServerLifecycle(host=host, port=port) @then('the lifecycle host should be "{expected}"') def step_check_lifecycle_host(context: Any, expected: str) -> None: assert context.lifecycle.host == expected @then("the lifecycle port should be {expected:d}") def step_check_lifecycle_port(context: Any, expected: int) -> None: assert context.lifecycle.port == expected @then("the lifecycle should not be started") def step_check_not_started(context: Any) -> None: assert not context.lifecycle.is_started @then("the lifecycle should not be stopped") def step_check_not_stopped(context: Any) -> None: assert not context.lifecycle.is_stopped use_step_matcher("re") @when('I try to create a ServerLifecycle with host "(?P[^"]*)"') def step_create_lifecycle_bad_host(context: Any, host: str) -> None: context.caught_exception = None try: ServerLifecycle(host=host) except ValueError as exc: context.caught_exception = exc use_step_matcher("parse") @when("I try to create a ServerLifecycle with port {port:d}") def step_create_lifecycle_bad_port(context: Any, port: int) -> None: context.caught_exception = None try: ServerLifecycle(port=port) except ValueError as exc: context.caught_exception = exc @when("I try to create a ServerLifecycle with empty log_level") def step_create_lifecycle_bad_log_level(context: Any) -> None: context.caught_exception = None try: ServerLifecycle(log_level="") except ValueError as exc: context.caught_exception = exc @then("a ValueError should be raised for invalid log_level") def step_check_value_error_log_level(context: Any) -> None: assert context.caught_exception is not None, "Expected ValueError" assert isinstance(context.caught_exception, ValueError) # --------------------------------------------------------------------------- # Settings # --------------------------------------------------------------------------- @then('the default server host from Settings should be "{expected}"') def step_check_settings_host(context: Any, expected: str) -> None: settings = Settings() assert settings.server_host == expected @then("the default server port from Settings should be {expected:d}") def step_check_settings_port(context: Any, expected: int) -> None: settings = Settings() assert settings.server_port == expected # --------------------------------------------------------------------------- # Graceful shutdown # --------------------------------------------------------------------------- @given("a ServerLifecycle with a mock uvicorn server") def step_lifecycle_with_mock_server(context: Any) -> None: lifecycle = ServerLifecycle(host="127.0.0.1", port=9999) mock_server = MagicMock() mock_server.should_exit = False lifecycle._server = mock_server context.lifecycle = lifecycle context.mock_server = mock_server @when("I call request_shutdown on the lifecycle") def step_call_request_shutdown(context: Any) -> None: context.lifecycle.request_shutdown() @then("the mock server should_exit flag should be true") def step_check_should_exit(context: Any) -> None: assert context.mock_server.should_exit is True @given("a ServerLifecycle that has already been started") def step_lifecycle_already_started(context: Any) -> None: lifecycle = ServerLifecycle(host="127.0.0.1", port=9998) lifecycle._started = True context.lifecycle = lifecycle @when("I try to start the lifecycle again") def step_try_double_start(context: Any) -> None: context.caught_exception = None try: context.lifecycle.start() except RuntimeError as exc: context.caught_exception = exc @then("a RuntimeError should be raised for double start") def step_check_runtime_error(context: Any) -> None: assert context.caught_exception is not None, "Expected RuntimeError" assert isinstance(context.caught_exception, RuntimeError)