62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
"""
|
|
Integration tests for evaluation router endpoints.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
import pytest
|
|
|
|
pytestmark = [pytest.mark.asyncio, pytest.mark.integration]
|
|
|
|
|
|
async def _upload_test_dataset(test_client, admin_headers: dict[str, str], db_id: str) -> tuple[str, str]:
|
|
dataset_name = f"pytest_dataset_{uuid.uuid4().hex[:8]}"
|
|
line = '{"query":"什么是单元测试？","gold_answer":"用于验证代码行为的自动化测试"}\n'
|
|
|
|
response = await test_client.post(
|
|
f"/api/evaluation/databases/{db_id}/datasets/upload",
|
|
data={"name": dataset_name, "description": "pytest dataset for download"},
|
|
files={"file": ("pytest_dataset.jsonl", line.encode("utf-8"), "application/x-ndjson")},
|
|
headers=admin_headers,
|
|
)
|
|
assert response.status_code == 200, response.text
|
|
|
|
payload = response.json()
|
|
assert payload.get("message") == "success"
|
|
dataset_id = payload.get("data", {}).get("dataset_id")
|
|
assert dataset_id
|
|
return dataset_id, line
|
|
|
|
|
|
async def test_download_dataset_requires_admin(test_client, standard_user):
|
|
response = await test_client.get(
|
|
"/api/evaluation/datasets/dataset_fake/download",
|
|
headers=standard_user["headers"],
|
|
)
|
|
assert response.status_code == 403
|
|
|
|
|
|
async def test_admin_can_download_dataset(test_client, admin_headers, knowledge_database):
|
|
dataset_id, expected_line = await _upload_test_dataset(test_client, admin_headers, knowledge_database["db_id"])
|
|
|
|
response = await test_client.get(
|
|
f"/api/evaluation/datasets/{dataset_id}/download",
|
|
headers=admin_headers,
|
|
)
|
|
assert response.status_code == 200, response.text
|
|
assert "application/x-ndjson" in response.headers.get("content-type", "")
|
|
assert "attachment" in response.headers.get("content-disposition", "").lower()
|
|
|
|
content = response.content.decode("utf-8")
|
|
assert expected_line.strip() in content
|
|
|
|
|
|
async def test_download_dataset_not_found(test_client, admin_headers):
|
|
response = await test_client.get(
|
|
f"/api/evaluation/datasets/dataset_not_found_{uuid.uuid4().hex[:8]}/download",
|
|
headers=admin_headers,
|
|
)
|
|
assert response.status_code == 404, response.text
|