kplam
51dc8dc4f6
- FastAPI + Qdrant + Redis + Ollama 技术栈 - L1→L2→L3→chunk 四层分层检索（dense + sparse RRF 融合） - 文档三级总结与 2.5 级回退 - query 解析路由与分类 - /admin 管理页面
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
"""统一 API 响应包装与业务异常单元测试"""
|
|
|
|
import pytest
|
|
|
|
from app.api.response import ApiError, error, ok
|
|
|
|
|
|
class TestOk:
|
|
def test_ok_structure(self) -> None:
|
|
data = {"items": [1, 2, 3]}
|
|
assert ok(data) == {"code": 0, "data": {"items": [1, 2, 3]}, "message": "ok"}
|
|
|
|
def test_ok_none_data(self) -> None:
|
|
assert ok(None) == {"code": 0, "data": None, "message": "ok"}
|
|
|
|
|
|
class TestError:
|
|
def test_error_structure(self) -> None:
|
|
assert error(1004, "文档不存在") == {"code": 1004, "data": None, "message": "文档不存在"}
|
|
|
|
def test_error_code_passthrough(self) -> None:
|
|
result = error(2000, "服务器内部错误")
|
|
assert result["code"] == 2000
|
|
assert result["message"] == "服务器内部错误"
|
|
|
|
|
|
class TestApiError:
|
|
def test_attributes(self) -> None:
|
|
exc = ApiError(1004, "文档不存在")
|
|
assert exc.code == 1004
|
|
assert exc.message == "文档不存在"
|
|
|
|
def test_is_exception_with_message(self) -> None:
|
|
exc = ApiError(1004, "文档不存在")
|
|
assert isinstance(exc, Exception)
|
|
assert str(exc) == "文档不存在"
|
|
|
|
def test_raise_and_catch(self) -> None:
|
|
with pytest.raises(ApiError) as exc_info:
|
|
raise ApiError(1001, "请求参数校验失败")
|
|
assert exc_info.value.code == 1001
|
|
assert exc_info.value.message == "请求参数校验失败"
|