"""
Tests for the SQLite-backed chat message store.
"""

import pytest
import pytest_asyncio
from pathlib import Path

from meshbay_node.chat.store import ChatStore


@pytest_asyncio.fixture
async def store(tmp_path):
    s = ChatStore(db_path=tmp_path / "test_chat.db")
    await s.open()
    yield s
    await s.close()


@pytest.mark.asyncio
async def test_save_and_retrieve(store):
    row_id = await store.save_message(
        sender_id="alice", iteration=0, payload=b"hello",
    )
    assert row_id == 1

    msgs = await store.get_messages()
    assert len(msgs) == 1
    assert msgs[0].sender_id == "alice"
    assert msgs[0].iteration == 0
    assert msgs[0].payload == b"hello"
    assert msgs[0].thread_id is None


@pytest.mark.asyncio
async def test_message_count(store):
    assert await store.message_count() == 0
    await store.save_message("alice", 0, b"msg1")
    await store.save_message("bob", 1, b"msg2")
    assert await store.message_count() == 2


@pytest.mark.asyncio
async def test_get_messages_since(store):
    await store.save_message("alice", 0, b"old")
    all_msgs = await store.get_messages()
    cutoff = all_msgs[0].timestamp
    await store.save_message("bob", 1, b"new")

    msgs = await store.get_messages(since=cutoff)
    assert len(msgs) == 1
    assert msgs[0].sender_id == "bob"


@pytest.mark.asyncio
async def test_thread_messages(store):
    await store.save_message("alice", 0, b"root", thread_id="t1")
    await store.save_message("bob", 1, b"reply", thread_id="t1")
    await store.save_message("carol", 2, b"other")

    thread = await store.get_thread("t1")
    assert len(thread) == 2
    assert thread[0].sender_id == "alice"
    assert thread[1].sender_id == "bob"


@pytest.mark.asyncio
async def test_message_ordering(store):
    for i in range(5):
        await store.save_message(f"user-{i}", i, f"msg-{i}".encode())

    msgs = await store.get_messages()
    assert len(msgs) == 5
    for i, m in enumerate(msgs):
        assert m.sender_id == f"user-{i}"


@pytest.mark.asyncio
async def test_limit(store):
    for i in range(10):
        await store.save_message("alice", i, f"msg-{i}".encode())

    msgs = await store.get_messages(limit=3)
    assert len(msgs) == 3


@pytest.mark.asyncio
async def test_context_manager(tmp_path):
    async with ChatStore(db_path=tmp_path / "ctx_test.db") as store:
        await store.save_message("alice", 0, b"test")
        assert await store.message_count() == 1