yunke/yunke-server/tests/test_analytics_api.py

"""/api/analytics/health + /overview:E0-lite 与 A1 冒烟。"""
from __future__ import annotations

import datetime as dt

from app.analytics import metrics
from app.db.models import Customer, MsgDailyStats, StatDailyStaff, WechatAccount

# naive:updated_at 列是无时区 DateTime;fromisoformat 构造不触发 DTZ 检查
_SEEN = "2026-09-04T12:00:00"

# overview 的 today_key 是真实今天——固定 day 键的 seed 必须 monkeypatch
_TODAY = 20260904


async def test_health_empty(make_client):
    async def seed(session):
        ...

    client = await make_client(seed)
    r = await client.get("/api/analytics/health")
    assert r.status_code == 200
    body = r.json()
    assert body["data_as_of"] is None
    assert len(body["entities"]) == 6
    assert all(e["last_seen"] is None for e in body["entities"])


async def test_health_with_stats(make_client):
    async def seed(session):
        session.add(
            MsgDailyStats(
                channel="wechat",
                owner_account_id="wx1",
                peer_kind="friend",
                peer_id="t1",
                day=20260904,
                send_cnt=1,
                recv_cnt=2,
                last_ts=1756900000000,
                last_is_send=False,
            )
        )
        session.add(
            Customer(
                friend_id="f1",
                channel="wechat",
                updated_at=dt.datetime.fromisoformat(_SEEN),
            )
        )
        await session.commit()

    client = await make_client(seed)
    r = await client.get("/api/analytics/health")
    assert r.status_code == 200
    assert r.json()["data_as_of"] == 1756900000000
    entities = {e["table"]: e["last_seen"] for e in r.json()["entities"]}
    # 前端契约:last_seen 必须是 ISO 字符串(fromisoformat 可解析)
    assert dt.datetime.fromisoformat(entities["customer"]) == dt.datetime.fromisoformat(
        _SEEN
    )


async def test_overview_endpoint(make_client, monkeypatch):
    monkeypatch.setattr(metrics, "today_key", lambda: _TODAY)

    async def seed(session):
        session.add(MsgDailyStats(channel="wechat", owner_account_id="wx1",
                                  peer_kind="friend", peer_id="t1", day=_TODAY,
                                  send_cnt=4, recv_cnt=1, last_ts=1788500000000,
                                  last_is_send=True))
        await session.commit()

    client = await make_client(seed)
    r = await client.get(
        "/api/analytics/overview", params={"channel": "wechat", "scope": "today"}
    )
    assert r.status_code == 200
    body = r.json()
    assert body["cards"]["msg_total"] == 5
    assert isinstance(body["trend"], list) and body["trend"]
    # scope 非法 → 422
    r2 = await client.get(
        "/api/analytics/overview", params={"channel": "wechat", "scope": "year"}
    )
    assert r2.status_code == 422
    # channel 非法 → 422(同一 Literal 校验路径)
    r3 = await client.get(
        "/api/analytics/overview", params={"channel": "sms", "scope": "today"}
    )
    assert r3.status_code == 422


async def test_overview_unmapped_accounts(make_client, monkeypatch):
    """控制器追加:cards 带 unmapped_accounts 计数(spec §8「未映射账号按账号计并标注」的标注通道)。"""
    monkeypatch.setattr(metrics, "today_key", lambda: _TODAY)

    async def seed(session):
        session.add(WechatAccount(wx_id="wx1", user_id="u1"))
        # wx2 无账号表记录 → 未映射
        session.add(MsgDailyStats(channel="wechat", owner_account_id="wx1",
                                  peer_kind="friend", peer_id="t1", day=_TODAY,
                                  send_cnt=2, recv_cnt=0, last_ts=1788500000000,
                                  last_is_send=True))
        session.add(MsgDailyStats(channel="wechat", owner_account_id="wx2",
                                  peer_kind="friend", peer_id="t2", day=_TODAY,
                                  send_cnt=1, recv_cnt=0, last_ts=1788501000000,
                                  last_is_send=True))
        await session.commit()

    client = await make_client(seed)
    r = await client.get(
        "/api/analytics/overview", params={"channel": "wechat", "scope": "today"}
    )
    body = r.json()
    assert body["cards"]["active_staff"] == 2       # u1 + wx2(按账号计)
    assert body["cards"]["unmapped_accounts"] == 1  # wx2 标注


async def test_overview_unmapped_rollup(make_client, monkeypatch):
    """rollup 路径未映射口径与回退路径一致:当窗 user_id 不在映射值集 → 未映射。

    recompute 以 amap.get(owner, owner) 落表,映射不上的 owner 以原值进
    stat_daily_staff(user_id=wx2);account_map.values() 只含在册员工 user_id,
    wx2 落在值集之外 → 被识别并标注。
    """
    monkeypatch.setattr(metrics, "today_key", lambda: _TODAY)

    async def seed(session):
        session.add(WechatAccount(wx_id="wx1", user_id="u1"))
        session.add(StatDailyStaff(day=_TODAY, channel="wechat", user_id="u1",
                                   msg_sent=2, msg_recv=0,
                                   active_friends=1, active_rooms=0))
        session.add(StatDailyStaff(day=_TODAY, channel="wechat", user_id="wx2",
                                   msg_sent=1, msg_recv=0,
                                   active_friends=1, active_rooms=0))
        session.add(MsgDailyStats(channel="wechat", owner_account_id="wx1",
                                  peer_kind="friend", peer_id="t1", day=_TODAY,
                                  send_cnt=2, recv_cnt=0, last_ts=1788500000000,
                                  last_is_send=True))
        await session.commit()

    client = await make_client(seed)
    r = await client.get(
        "/api/analytics/overview", params={"channel": "wechat", "scope": "today"}
    )
    body = r.json()
    assert body["cards"]["active_staff"] == 2
    assert body["cards"]["unmapped_accounts"] == 1


async def test_overview_custom_scope_api(make_client, monkeypatch):
    """C1 自定义日期:正常窗 200 + customer_assets 恒在;缺参/start>end/
    跨度超 366 天/非法日历 全 422;闰年整 366 天放行。"""
    monkeypatch.setattr(metrics, "today_key", lambda: _TODAY)

    async def seed(session):
        session.add(MsgDailyStats(channel="wechat", owner_account_id="wx1",
                                  peer_kind="friend", peer_id="t1", day=20260815,
                                  send_cnt=10, recv_cnt=0, last_ts=1786000000000,
                                  last_is_send=True))
        await session.commit()

    client = await make_client(seed)
    r = await client.get("/api/analytics/overview", params={
        "channel": "wechat", "scope": "custom",
        "start": 20260801, "end": 20260827,
    })
    assert r.status_code == 200
    body = r.json()
    assert body["scope"] == "custom"
    assert len(body["days"]) == 27
    assert body["cards"]["msg_total"] == 10
    # B1 客户资产总览恒在(全库快照,不随 scope)
    assert body["customer_assets"]["total"] >= 0

    # 缺 start/end → 422(不让静默落到空窗)
    r2 = await client.get("/api/analytics/overview",
                          params={"scope": "custom"})
    assert r2.status_code == 422
    # start > end → 422
    r3 = await client.get("/api/analytics/overview", params={
        "scope": "custom", "start": 20260827, "end": 20260801})
    assert r3.status_code == 422
    # 跨度 367 天 → 422(2024-01-01..2025-01-01 含端点 367 天)
    r4 = await client.get("/api/analytics/overview", params={
        "scope": "custom", "start": 20240101, "end": 20250101})
    assert r4.status_code == 422
    # 非法日历(2 月 30)→ 422——Query ge/le 只拦数值范围不拦日历
    r5 = await client.get("/api/analytics/overview", params={
        "scope": "custom", "start": 20260230, "end": 20260301})
    assert r5.status_code == 422
    # 闰年整年 366 天(含端点)→ 放行
    r6 = await client.get("/api/analytics/overview", params={
        "scope": "custom", "start": 20240101, "end": 20241231})
    assert r6.status_code == 200
    assert len(r6.json()["days"]) == 366