30 lines
1015 B
Python
30 lines
1015 B
Python
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"
|