Create tests/test_adapter_contract.py

tests/test_adapter_contract.py

@@ -1,43 +1,30 @@
-import inspect
-import asyncio
-import pytest
-from typing import Dict, List
-
-from ingest.generic_public_foia import GenericFOIAAdapter
-from ingest.registry import get_all_adapters
-
-
-@pytest.mark.asyncio
-async def test_adapters_implement_search():
-    """
-    Every adapter must implement an async search() method.
-    """
-    adapters = get_all_adapters()
-
-    assert adapters, "No adapters registered"
-
-    for name, adapter in adapters.items():
-        assert isinstance(adapter, GenericFOIAAdapter), f"{name} not subclassing GenericFOIAAdapter"
-
-        method = adapter.search
-        assert callable(method), f"{name}.search is not callable"
-        assert inspect.iscoroutinefunction(method), f"{name}.search must be async"
+REQUIRED_KEYS = {
+    "source",
+    "title",
+    "url",
+    "snippet",
+    "live",
+    "extended",
+}
 
 
 @pytest.mark.asyncio
-async def test_search_returns_list_of_dicts():
+async def test_result_schema():
     """
-    search() must always return List[Dict], even on failure.
+    Enforces normalized result format for UI, export, and citations.
     """
     adapters = get_all_adapters()
 
     for name, adapter in adapters.items():
-        try:
-            results = await adapter.search("test")
-        except Exception as e:
-            pytest.fail(f"{name}.search() raised exception: {e}")
-
-        assert isinstance(results, list), f"{name}.search did not return a list"
+        results = await adapter.search("compliance test")
 
         for r in results:
-            assert isinstance(r, dict), f"{name} returned non-dict result"
+            missing = REQUIRED_KEYS - set(r.keys())
+            assert not missing, f"{name} missing keys: {missing}"
+
+            assert isinstance(r["source"], str)
+            assert isinstance(r["title"], str)
+            assert isinstance(r["url"], str)
+            assert isinstance(r["snippet"], str)
+            assert isinstance(r["live"], bool)
+            assert isinstance(r["extended"], bool)