import pytest

from takopi.telegram.chat_prefs import ChatPrefsStore


@pytest.mark.anyio
async def test_chat_prefs_store_roundtrip(tmp_path) -> None:
    path = tmp_path / "telegram_chat_prefs_state.json"
    store = ChatPrefsStore(path)
    await store.set_default_engine(123, "codex")
    await store.set_default_engine(123, "codex")
    await store.clear_default_engine(456)

    assert await store.get_default_engine(123) == "codex"

    store2 = ChatPrefsStore(path)
    assert await store2.get_default_engine(123) == "codex"

    await store2.clear_default_engine(123)
    assert await store2.get_default_engine(123) is None