Tianyang
542361c63a
Also add a reindex endpoint and button to rebuild the Meilisearch index from files already on disk, without needing to re-upload them.
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.main import app
|
|
from app.dependencies import get_document_service, get_search_client
|
|
|
|
|
|
class FakeSearchClient:
|
|
def __init__(self):
|
|
self.queries = []
|
|
|
|
def search(self, query):
|
|
self.queries.append(query)
|
|
if query == "检索":
|
|
return [
|
|
{
|
|
"key": "产品文档/架构设计.md",
|
|
"title": "架构设计.md",
|
|
"path": "产品文档",
|
|
"snippet": "支持全文<em>检索</em>功能",
|
|
}
|
|
]
|
|
return []
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
fake_client = FakeSearchClient()
|
|
app.dependency_overrides[get_search_client] = lambda: fake_client
|
|
with TestClient(app) as test_client:
|
|
test_client.fake_client = fake_client
|
|
yield test_client
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
def test_search_returns_matching_results(client):
|
|
response = client.get("/api/search", params={"q": "检索"})
|
|
assert response.status_code == 200
|
|
body = response.json()
|
|
assert body["results"][0]["key"] == "产品文档/架构设计.md"
|
|
assert "<em>检索</em>" in body["results"][0]["snippet"]
|
|
|
|
|
|
def test_search_returns_empty_results_for_no_match(client):
|
|
response = client.get("/api/search", params={"q": "不存在的关键词"})
|
|
assert response.status_code == 200
|
|
assert response.json() == {"results": []}
|
|
|
|
|
|
def test_search_passes_query_string_to_search_client(client):
|
|
client.get("/api/search", params={"q": "Garage"})
|
|
assert client.fake_client.queries == ["Garage"]
|
|
|
|
|
|
class FakeDocumentService:
|
|
def __init__(self):
|
|
self.reindex_called = False
|
|
|
|
def reindex_all(self):
|
|
self.reindex_called = True
|
|
return {"indexed": 3, "failed": 0, "total": 3}
|
|
|
|
|
|
@pytest.fixture
|
|
def reindex_client():
|
|
fake_service = FakeDocumentService()
|
|
app.dependency_overrides[get_document_service] = lambda: fake_service
|
|
with TestClient(app) as test_client:
|
|
test_client.fake_service = fake_service
|
|
yield test_client
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
def test_reindex_triggers_full_reindex_and_returns_summary(reindex_client):
|
|
response = reindex_client.post("/api/search/reindex")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"indexed": 3, "failed": 0, "total": 3}
|
|
assert reindex_client.fake_service.reindex_called
|