linkeyu
9286b82b6d
## 变更内容 - 新增比价记录概览聚合接口，主耗时均值及 P5/P50/P95/P99 仅统计 success - 比价列表支持按北京自然日过滤，概览和列表复用同一筛选口径 - 后端聚合成功率、成本、低价率及中途退出指标 - 修复单条样本分位数错误返回 0 的边界问题 - 增加成功/失败/退出及日期分页回归测试 ## 验证 - Python compileall - SQLite 聚合、日期边界和分页检查 --------- Co-authored-by: unknown <798648091@qq.com> Reviewed-on: #152 Co-authored-by: linkeyu <linkeyu@wonderable.ai> Co-committed-by: linkeyu <linkeyu@wonderable.ai>
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""比价记录页后端分页与概览聚合。"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import UTC, date, datetime
|
|
|
|
import pytest
|
|
|
|
from app.admin.repositories import queries
|
|
from app.db.session import SessionLocal
|
|
from app.models.comparison import ComparisonRecord
|
|
|
|
|
|
def test_summary_uses_only_success_durations_and_filters_beijing_date() -> None:
|
|
db = SessionLocal()
|
|
try:
|
|
rows = [
|
|
("summary-success-a", "success", 1000, 1.0, 100),
|
|
("summary-success-b", "success", 3000, 2.0, 0),
|
|
("summary-failed", "failed", 100_000, 3.0, 0),
|
|
("summary-cancelled", "cancelled", 5000, 4.0, 0),
|
|
]
|
|
for trace_id, status, total_ms, cost, saved in rows:
|
|
db.add(ComparisonRecord(
|
|
trace_id=trace_id,
|
|
status=status,
|
|
total_ms=total_ms,
|
|
llm_cost_yuan=cost,
|
|
saved_amount_cents=saved,
|
|
created_at=datetime(2038, 1, 15, 12, tzinfo=UTC),
|
|
))
|
|
db.add(ComparisonRecord(
|
|
trace_id="summary-outside-day",
|
|
status="success",
|
|
total_ms=999_999,
|
|
created_at=datetime(2038, 1, 16, 16, tzinfo=UTC),
|
|
))
|
|
db.flush()
|
|
|
|
summary = queries.comparison_records_summary(
|
|
db, date_from=date(2038, 1, 15), date_to=date(2038, 1, 15)
|
|
)
|
|
|
|
assert summary["started"] == 4
|
|
assert summary["completed"] == 3
|
|
assert summary["success"] == 2
|
|
assert summary["success_rate"] == pytest.approx(2 / 3)
|
|
assert summary["avg_token_cost"] == pytest.approx(2.5)
|
|
assert summary["lower_price_rate"] == 0.5
|
|
assert summary["avg_duration_ms"] == 2000
|
|
assert summary["p5_duration_ms"] == 1100
|
|
assert summary["p50_duration_ms"] == 2000
|
|
assert summary["p95_duration_ms"] == 2900
|
|
assert summary["p99_duration_ms"] == 2980
|
|
assert summary["cancelled"] == 1
|
|
assert summary["cancelled_rate"] == 0.25
|
|
assert summary["cancelled_p50_ms"] == 5000
|
|
|
|
items, _next_cursor, total = queries.list_comparison_records(
|
|
db, date_from=date(2038, 1, 15), date_to=date(2038, 1, 15), limit=20
|
|
)
|
|
assert total == 4
|
|
assert {item.trace_id for item in items} == {row[0] for row in rows}
|
|
finally:
|
|
db.rollback()
|
|
db.close()
|