from __future__ import annotations
import pytest
from unittest.mock import AsyncMock, MagicMock, patch
from src.api.schemas import SearchRequest, SearchResponse, SearchResult, SearchMetadata
from src.config import SearchEngine
from src.core.search import SearchError, do_search
def _mock_pool(started: bool = True) -> MagicMock:
pool = MagicMock()
pool._started = started
page = AsyncMock()
pool.acquire.return_value.__aenter__ = AsyncMock(return_value=page)
pool.acquire.return_value.__aexit__ = AsyncMock(return_value=False)
return pool
class TestDoSearch:
@pytest.mark.asyncio
async def test_raises_when_pool_not_started(self):
pool = _mock_pool(started=False)
req = SearchRequest(query="test")
with pytest.raises(SearchError, match="not initialized"):
await do_search(pool, req)
@pytest.mark.asyncio
@patch("src.core.search.crawl_results", new_callable=AsyncMock)
@patch("src.core.search.ENGINES")
async def test_returns_response(self, mock_engines, mock_crawl):
pool = _mock_pool()
results = [SearchResult(title="R1", url="https://example.com", snippet="S1")]
mock_engine = AsyncMock()
mock_engine.search.return_value = results
mock_engines.__getitem__ = MagicMock(return_value=mock_engine)
mock_crawl.return_value = results
req = SearchRequest(query="test", engine=SearchEngine.DUCKDUCKGO)
resp = await do_search(pool, req)
assert resp.query == "test"
assert resp.total == 1
assert resp.engine == SearchEngine.DUCKDUCKGO
assert resp.results[0].title == "R1"
@pytest.mark.asyncio
@patch("src.core.search.crawl_results", new_callable=AsyncMock)
@patch("src.core.search.ENGINES")
@patch("src.core.search.FALLBACK_ORDER", {
SearchEngine.GOOGLE: [SearchEngine.DUCKDUCKGO, SearchEngine.BING],
})
async def test_fallback_on_empty_results(self, mock_engines, mock_crawl):
pool = _mock_pool()
results = [SearchResult(title="FB", url="https://fb.com", snippet="Fallback")]
primary = AsyncMock()
primary.search.return_value = []
fallback = AsyncMock()
fallback.search.return_value = results
mock_engines.__getitem__ = MagicMock(side_effect=lambda k: {
SearchEngine.GOOGLE: primary,
SearchEngine.DUCKDUCKGO: fallback,
}[k])
mock_crawl.return_value = results
req = SearchRequest(query="test", engine=SearchEngine.GOOGLE)
resp = await do_search(pool, req)
assert resp.engine == SearchEngine.DUCKDUCKGO
assert resp.total == 1
class TestFetchUrlContent:
@pytest.mark.asyncio
@patch("src.core.search.extract_main_content_markdown")
@patch("src.core.search.fetch_page_content", new_callable=AsyncMock)
async def test_returns_markdown(self, mock_fetch, mock_extract):
pool = _mock_pool()
mock_fetch.return_value = "<html><body>Hello</body></html>"
mock_extract.return_value = "# Hello"
from src.core.search import fetch_url_content
result = await fetch_url_content(pool, "https://example.com")
assert result == "# Hello"
@pytest.mark.asyncio
@patch("src.core.search.fetch_page_content", new_callable=AsyncMock)
async def test_returns_empty_on_failure(self, mock_fetch):
pool = _mock_pool()
mock_fetch.return_value = ""
from src.core.search import fetch_url_content
result = await fetch_url_content(pool, "https://example.com")
assert result == ""
