yunke/yunke-server/tests/test_api_messages.py

"""Tests for GET /api/messages — chat timeline.

Seeds 3 messages (2 for talker t_a across times 1000/3000, 1 for talker t_b at
2000) and asserts filters (channel, owner, talker, time window) + ordering
(``timestamp asc``) + pagination.
"""

from __future__ import annotations

from sqlalchemy.ext.asyncio import AsyncSession

from app.db.models import Message


async def _seed(session: AsyncSession) -> None:
    session.add_all(
        [
            Message(
                msg_id="m1",
                channel="wechat",
                owner_account_id="wx_owner_1",
                talker_id="t_a",
                room_id=None,
                text="hello",
                timestamp=1000,
                is_send=True,
                type="text",
                type_raw=1,
            ),
            Message(
                msg_id="m2",
                channel="wechat",
                owner_account_id="wx_owner_1",
                talker_id="t_b",
                room_id=None,
                text="hi",
                timestamp=2000,
                is_send=False,
                type="text",
                type_raw=1,
            ),
            Message(
                msg_id="m3",
                channel="wechat",
                owner_account_id="wx_owner_1",
                talker_id="t_a",
                room_id=None,
                text="bye",
                timestamp=3000,
                is_send=True,
                type="text",
                type_raw=1,
            ),
        ]
    )
    await session.commit()


async def test_default_page_returns_all_ordered_asc(make_client):
    """No filters -> 3 rows ordered by ``timestamp asc`` (timeline order)."""
    client = await make_client(_seed)
    r = await client.get("/api/messages")
    assert r.status_code == 200
    body = r.json()
    assert body["total"] == 3
    assert len(body["items"]) == 3
    assert [it["msg_id"] for it in body["items"]] == ["m1", "m2", "m3"]
    for key in (
        "msg_id",
        "channel",
        "owner_account_id",
        "talker_id",
        "room_id",
        "is_send",
        "type",
        "type_raw",
        "text",
        "file_url",
        "sender_name",
        "sender_head_url",
        "timestamp",
    ):
        assert key in body["items"][0]


async def test_filter_by_talker_id(make_client):
    """``?talker_id=t_a`` -> 2 messages for that 1:1 conversation."""
    client = await make_client(_seed)
    r = await client.get("/api/messages", params={"talker_id": "t_a"})
    assert r.status_code == 200
    body = r.json()
    assert body["total"] == 2
    assert {it["msg_id"] for it in body["items"]} == {"m1", "m3"}


async def test_filter_by_owner_account_id(make_client):
    """``?owner_account_id=wx_owner_1`` -> all 3 (all owned by this account)."""
    client = await make_client(_seed)
    r = await client.get("/api/messages", params={"owner_account_id": "wx_owner_1"})
    assert r.status_code == 200
    assert r.json()["total"] == 3


async def test_filter_by_owner_account_id_no_match(make_client):
    """A non-existent owner account -> 0 rows."""
    client = await make_client(_seed)
    r = await client.get("/api/messages", params={"owner_account_id": "nobody"})
    assert r.status_code == 200
    assert r.json()["total"] == 0


async def test_time_window_filter(make_client):
    """``?start=1500&end=2500`` -> only m2 (timestamp 2000)."""
    client = await make_client(_seed)
    r = await client.get(
        "/api/messages", params={"start": 1500, "end": 2500}
    )
    assert r.status_code == 200
    body = r.json()
    assert body["total"] == 1
    assert body["items"][0]["msg_id"] == "m2"


async def test_time_window_combined_with_talker(make_client):
    """``talker_id=t_a & start=1500`` -> only m3 (t_a's later message)."""
    client = await make_client(_seed)
    r = await client.get(
        "/api/messages", params={"talker_id": "t_a", "start": 1500}
    )
    assert r.status_code == 200
    body = r.json()
    assert body["total"] == 1
    assert body["items"][0]["msg_id"] == "m3"


async def test_channel_filter(make_client):
    """``?channel=whatsapp`` -> 0 (all seeded are wechat)."""
    client = await make_client(_seed)
    r = await client.get("/api/messages", params={"channel": "whatsapp"})
    assert r.status_code == 200
    assert r.json()["total"] == 0


async def test_pagination_page_size_one(make_client):
    """``page_size=1&page=1`` -> 1 item, total still 3; page 2 -> next item."""
    client = await make_client(_seed)
    r1 = await client.get(
        "/api/messages", params={"page": 1, "page_size": 1}
    )
    b1 = r1.json()
    assert b1["total"] == 3
    assert len(b1["items"]) == 1
    assert b1["items"][0]["msg_id"] == "m1"

    r2 = await client.get(
        "/api/messages", params={"page": 2, "page_size": 1}
    )
    b2 = r2.json()
    assert len(b2["items"]) == 1
    assert b2["items"][0]["msg_id"] == "m2"


async def test_group_chat_filter_by_room_id(make_client):
    """``?room_id=room_x`` -> only the group message."""
    async def seed(session: AsyncSession) -> None:
        await _seed(session)
        session.add(
            Message(
                msg_id="m4",
                channel="wechat",
                owner_account_id="wx_owner_1",
                talker_id=None,
                room_id="room_x",
                text="group hi",
                timestamp=4000,
                type="text",
                type_raw=1,
            )
        )
        await session.commit()

    client = await make_client(seed)
    r = await client.get("/api/messages", params={"room_id": "room_x"})
    assert r.status_code == 200
    body = r.json()
    assert body["total"] == 1
    assert body["items"][0]["msg_id"] == "m4"