from app.search_provider_registry import SearchProviderRegistry, StaticArchiveProvider from app.search_gateway import SearchGateway, SearchRequest def test_provider_registry_has_default_layers(): registry = SearchProviderRegistry() status = registry.status() names = {item["name"] for item in status["providers"]} assert "official_site" in names assert "general_web" in names assert "internal_archive" in names def test_internal_archive_can_supply_evidence(): registry = SearchProviderRegistry() registry.register_provider( "internal_archive", StaticArchiveProvider([ { "title": "AIOS document", "snippet": "search gateway architecture", "tags": "search gateway", "confidence": "high", } ]), ) gateway = SearchGateway(registry.ordered()) result = gateway.search(SearchRequest(query="search gateway", requester="test")) assert result.status == "found"