added tests

tests/base.py

@@ -0,0 +1,82 @@
+"""Test base — two execution modes from one test suite.
+
+Modes (set via CONTRACT_TEST_MODE env var):
+
+  inprocess  (default) — httpx.AsyncClient with ASGI transport, full FastAPI
+                          stack in-process. Fast, no server needed.
+
+  live       — httpx.AsyncClient against CONTRACT_TEST_URL.
+               Tests against a running deployment (Kind, EC2, etc).
+
+Usage:
+    # In-process (default)
+    pytest tests/
+
+    # Against live server
+    CONTRACT_TEST_MODE=live CONTRACT_TEST_URL=http://unt.local.ar pytest tests/
+"""
+
+import asyncio
+import os
+
+import httpx
+import pytest_asyncio
+
+
+def get_mode() -> str:
+    return os.getenv("CONTRACT_TEST_MODE", "inprocess")
+
+
+def get_base_url() -> str:
+    return os.getenv("CONTRACT_TEST_URL", "http://localhost:8040")
+
+
+@pytest_asyncio.fixture
+async def client() -> httpx.AsyncClient:
+    """Async HTTP client for API tests."""
+    if get_mode() == "live":
+        async with httpx.AsyncClient(base_url=get_base_url(), timeout=30.0) as c:
+            yield c
+    else:
+        from httpx import ASGITransport
+        from api.main import app
+
+        transport = ASGITransport(app=app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://test") as c:
+            yield c
+
+
+# Alias for agent tests
+agent_client = client
+
+
+class ContractHelpers:
+    """Reusable assertion and utility methods."""
+
+    @staticmethod
+    def assert_status(response: httpx.Response, expected: int):
+        assert response.status_code == expected, (
+            f"Expected {expected}, got {response.status_code}: {response.text[:200]}"
+        )
+
+    @staticmethod
+    def assert_has_fields(data: dict, *fields: str):
+        for f in fields:
+            assert f in data, f"Missing field: {f}. Keys: {list(data.keys())}"
+
+    @staticmethod
+    def assert_is_list(data, min_length: int = 0):
+        assert isinstance(data, list), f"Expected list, got {type(data)}"
+        assert len(data) >= min_length, f"Expected >= {min_length} items, got {len(data)}"
+
+    @staticmethod
+    async def poll_run(client: httpx.AsyncClient, run_id: str, timeout: int = 60) -> dict:
+        """Poll an agent run until completion."""
+        deadline = asyncio.get_event_loop().time() + timeout
+        while asyncio.get_event_loop().time() < deadline:
+            res = await client.get(f"/agents/runs/{run_id}")
+            data = res.json()
+            if data.get("status") in ("completed", "error"):
+                return data
+            await asyncio.sleep(1)
+        raise TimeoutError(f"Run {run_id} did not complete in {timeout}s")