soleprint init commit

station/tools/tester/base.py

@@ -0,0 +1,119 @@
+"""
+Pure HTTP Contract Tests - Base Class
+
+Framework-agnostic: works against ANY backend implementation.
+"""
+
+import unittest
+import httpx
+
+from .config import config
+
+
+class ContractTestCase(unittest.TestCase):
+    """
+    Base class for pure HTTP contract tests.
+
+    Features:
+        - Framework-agnostic (works with Django, FastAPI, Node, etc.)
+        - Pure HTTP via httpx library
+        - No database access - all data through API
+        - API Key authentication
+    """
+
+    _base_url = None
+    _api_key = None
+
+    @classmethod
+    def setUpClass(cls):
+        """Set up once per test class"""
+        super().setUpClass()
+        cls._base_url = config.get("CONTRACT_TEST_URL", "").rstrip("/")
+        if not cls._base_url:
+            raise ValueError("CONTRACT_TEST_URL required in environment")
+
+        cls._api_key = config.get("CONTRACT_TEST_API_KEY", "")
+        if not cls._api_key:
+            raise ValueError("CONTRACT_TEST_API_KEY required in environment")
+
+    @property
+    def base_url(self):
+        return self._base_url
+
+    @property
+    def api_key(self):
+        return self._api_key
+
+    def _auth_headers(self):
+        """Get authorization headers"""
+        return {"Authorization": f"Api-Key {self.api_key}"}
+
+    # =========================================================================
+    # HTTP helpers
+    # =========================================================================
+
+    def get(self, path: str, params: dict = None, **kwargs):
+        """GET request"""
+        url = f"{self.base_url}{path}"
+        headers = {**self._auth_headers(), **kwargs.pop("headers", {})}
+        response = httpx.get(url, params=params, headers=headers, timeout=30, **kwargs)
+        return self._wrap_response(response)
+
+    def post(self, path: str, data: dict = None, **kwargs):
+        """POST request with JSON"""
+        url = f"{self.base_url}{path}"
+        headers = {**self._auth_headers(), **kwargs.pop("headers", {})}
+        response = httpx.post(url, json=data, headers=headers, timeout=30, **kwargs)
+        return self._wrap_response(response)
+
+    def put(self, path: str, data: dict = None, **kwargs):
+        """PUT request with JSON"""
+        url = f"{self.base_url}{path}"
+        headers = {**self._auth_headers(), **kwargs.pop("headers", {})}
+        response = httpx.put(url, json=data, headers=headers, timeout=30, **kwargs)
+        return self._wrap_response(response)
+
+    def patch(self, path: str, data: dict = None, **kwargs):
+        """PATCH request with JSON"""
+        url = f"{self.base_url}{path}"
+        headers = {**self._auth_headers(), **kwargs.pop("headers", {})}
+        response = httpx.patch(url, json=data, headers=headers, timeout=30, **kwargs)
+        return self._wrap_response(response)
+
+    def delete(self, path: str, **kwargs):
+        """DELETE request"""
+        url = f"{self.base_url}{path}"
+        headers = {**self._auth_headers(), **kwargs.pop("headers", {})}
+        response = httpx.delete(url, headers=headers, timeout=30, **kwargs)
+        return self._wrap_response(response)
+
+    def _wrap_response(self, response):
+        """Add .data attribute for consistency with DRF responses"""
+        try:
+            response.data = response.json()
+        except Exception:
+            response.data = None
+        return response
+
+    # =========================================================================
+    # Assertion helpers
+    # =========================================================================
+
+    def assert_status(self, response, expected_status: int):
+        """Assert response has expected status code"""
+        self.assertEqual(
+            response.status_code,
+            expected_status,
+            f"Expected {expected_status}, got {response.status_code}. "
+            f"Response: {response.data if hasattr(response, 'data') else response.content[:500]}"
+        )
+
+    def assert_has_fields(self, data: dict, *fields: str):
+        """Assert dictionary has all specified fields"""
+        missing = [f for f in fields if f not in data]
+        self.assertEqual(missing, [], f"Missing fields: {missing}. Got: {list(data.keys())}")
+
+    def assert_is_list(self, data, min_length: int = 0):
+        """Assert data is a list with minimum length"""
+        self.assertIsInstance(data, list)
+        self.assertGreaterEqual(len(data), min_length)