51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
"""Shared fixtures for tests/tools/ web-provider tests.
|
|
|
|
Per-file subprocess isolation means each test file gets a fresh interpreter,
|
|
so module-level state (like the web-search-provider registry) is empty when
|
|
a file starts. The ``web_registry_populated`` fixture registers all bundled
|
|
providers before each test and resets the registry afterwards — tests that
|
|
depend on the registry being populated should use it explicitly or via
|
|
``@pytest.mark.usefixtures("web_registry_populated")``.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
|
|
def register_all_web_providers():
|
|
"""Register all bundled web-search providers into the global registry.
|
|
|
|
This is the single source of truth for the provider list used by
|
|
test classes that need the registry populated for dispatch checks.
|
|
"""
|
|
from agent.web_search_registry import register_provider, _reset_for_tests
|
|
from plugins.web.brave_free.provider import BraveFreeWebSearchProvider
|
|
from plugins.web.ddgs.provider import DDGSWebSearchProvider
|
|
from plugins.web.exa.provider import ExaWebSearchProvider
|
|
from plugins.web.firecrawl.provider import FirecrawlWebSearchProvider
|
|
from plugins.web.parallel.provider import ParallelWebSearchProvider
|
|
from plugins.web.searxng.provider import SearXNGWebSearchProvider
|
|
from plugins.web.tavily.provider import TavilyWebSearchProvider
|
|
from plugins.web.xai.provider import XAIWebSearchProvider
|
|
|
|
_reset_for_tests()
|
|
for cls in (
|
|
BraveFreeWebSearchProvider,
|
|
DDGSWebSearchProvider,
|
|
ExaWebSearchProvider,
|
|
FirecrawlWebSearchProvider,
|
|
ParallelWebSearchProvider,
|
|
SearXNGWebSearchProvider,
|
|
TavilyWebSearchProvider,
|
|
XAIWebSearchProvider,
|
|
):
|
|
register_provider(cls())
|
|
|
|
|
|
@pytest.fixture
|
|
def web_registry_populated():
|
|
"""Populate the web-search-provider registry for one test, then reset."""
|
|
register_all_web_providers()
|
|
yield
|
|
from agent.web_search_registry import _reset_for_tests
|
|
_reset_for_tests()
|