jorgasisten

from types import SimpleNamespace

from jorgasisten.bot.main import message_user_id


def test_message_user_id_prefers_human_from_user() -> None:
    message = SimpleNamespace(
        from_user=SimpleNamespace(id=7389477424, is_bot=False),
        chat=SimpleNamespace(id=999111222, type="private"),
        sender_chat=None,
    )

    assert message_user_id(message) == 7389477424


def test_message_user_id_falls_back_to_private_chat_when_from_user_is_bot() -> None:
    message = SimpleNamespace(
        from_user=SimpleNamespace(id=8733064349, is_bot=True),
        chat=SimpleNamespace(id=7389477424, type="private"),
        sender_chat=None,
    )

    assert message_user_id(message) == 7389477424