4c3870bf3e
Add IngestionService chaining analysis chunks → EmbeddingClient → OpenSearchStore.
Idempotent: existing doc chunks deleted before re-indexing. REST API:
POST /api/ingestion/{jobId}, DELETE /api/ingestion/{docId}, GET /api/ingestion/status.
Wired in lifespan when OPENSEARCH_URL + EMBEDDING_URL are set. 25 new tests.
112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
"""Tests for the ingestion REST API endpoints."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from main import app
|
|
from services.ingestion_service import IngestionError, IngestionResult
|
|
|
|
|
|
@pytest.fixture()
|
|
def client():
|
|
return TestClient(app, raise_server_exceptions=False)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clear_ingestion_service():
|
|
"""Ensure ingestion_service is reset after each test."""
|
|
original = getattr(app.state, "ingestion_service", None)
|
|
yield
|
|
app.state.ingestion_service = original
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# GET /api/ingestion/status
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_status_available(client):
|
|
app.state.ingestion_service = MagicMock()
|
|
resp = client.get("/api/ingestion/status")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["available"] is True
|
|
|
|
|
|
def test_status_unavailable(client):
|
|
app.state.ingestion_service = None
|
|
resp = client.get("/api/ingestion/status")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["available"] is False
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# POST /api/ingestion/{job_id}
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_ingest_success(client):
|
|
svc = MagicMock()
|
|
svc.ingest = AsyncMock(
|
|
return_value=IngestionResult(doc_id="doc-1", chunks_indexed=5, embedding_dimension=384)
|
|
)
|
|
app.state.ingestion_service = svc
|
|
resp = client.post("/api/ingestion/job-1")
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["doc_id"] == "doc-1"
|
|
assert data["chunks_indexed"] == 5
|
|
assert data["embedding_dimension"] == 384
|
|
|
|
|
|
def test_ingest_no_service(client):
|
|
app.state.ingestion_service = None
|
|
resp = client.post("/api/ingestion/job-1")
|
|
assert resp.status_code == 503
|
|
|
|
|
|
def test_ingest_ingestion_error(client):
|
|
svc = MagicMock()
|
|
svc.ingest = AsyncMock(side_effect=IngestionError("job not found"))
|
|
app.state.ingestion_service = svc
|
|
resp = client.post("/api/ingestion/job-1")
|
|
assert resp.status_code == 422
|
|
assert "job not found" in resp.json()["detail"]
|
|
|
|
|
|
def test_ingest_unexpected_error(client):
|
|
svc = MagicMock()
|
|
svc.ingest = AsyncMock(side_effect=RuntimeError("boom"))
|
|
app.state.ingestion_service = svc
|
|
resp = client.post("/api/ingestion/job-1")
|
|
assert resp.status_code == 500
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# DELETE /api/ingestion/{doc_id}
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_delete_success(client):
|
|
svc = MagicMock()
|
|
svc.delete = AsyncMock(return_value=3)
|
|
app.state.ingestion_service = svc
|
|
resp = client.delete("/api/ingestion/doc-1")
|
|
assert resp.status_code == 204
|
|
|
|
|
|
def test_delete_no_service(client):
|
|
app.state.ingestion_service = None
|
|
resp = client.delete("/api/ingestion/doc-1")
|
|
assert resp.status_code == 503
|
|
|
|
|
|
def test_delete_ingestion_error(client):
|
|
svc = MagicMock()
|
|
svc.delete = AsyncMock(side_effect=IngestionError("opensearch error"))
|
|
app.state.ingestion_service = svc
|
|
resp = client.delete("/api/ingestion/doc-1")
|
|
assert resp.status_code == 422
|