unknown
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()
|