from teams_bot.bot import RelayPlugin
from teams_bot.commands import get_crew_id


def test_is_relay_group(crew, outsider):
    crew.bot.relayplugin = RelayPlugin(crew.bot)
    crew.bot.add_account_plugin(crew.bot.relayplugin)
    assert not crew.bot.relayplugin.is_relay_group(crew)

    botcontact_outsider = outsider.create_contact(crew.bot.get_config("addr"))
    outsider_to_bot = outsider.create_chat(botcontact_outsider)
    outsider_to_bot.send_text("test message to bot")
    message_from_outsider = crew.bot.wait_next_incoming_message()
    assert not crew.bot.relayplugin.is_relay_group(message_from_outsider.chat)

    crew.user.wait_next_incoming_message()  # group added message
    forwarded_message_from_outsider = crew.user.wait_next_incoming_message()
    user_relay_group = forwarded_message_from_outsider.create_chat()
    user_relay_group.send_text("Harmless reply in relay group")
    message_in_relay_group = crew.bot.wait_next_incoming_message()
    assert message_in_relay_group.chat.get_name().startswith(
        "[%s] " % (crew.bot.get_config("addr").split("@")[0],)
    )
    assert (
        message_in_relay_group.chat.get_messages()[0].get_sender_contact()
        == crew.bot.get_self_contact()
    )
    assert not message_in_relay_group.chat.is_protected()
    assert (
        crew.bot.get_chat_by_id(get_crew_id(crew.bot)).get_contacts()
        == message_in_relay_group.chat.get_contacts()
    )
    assert crew.bot.relayplugin.is_relay_group(message_in_relay_group.chat)

    outsider_to_bot = outsider.create_group_chat(
        "test with outsider", contacts=[botcontact_outsider]
    )
    outsider_to_bot.send_text("test message to outsider group")
    message_from_outsider = crew.bot.wait_next_incoming_message()
    assert not crew.bot.relayplugin.is_relay_group(message_from_outsider.chat)

    botcontact_user = crew.user.create_contact(crew.bot.get_config("addr"))
    user_to_bot = crew.user.create_chat(botcontact_user)
    user_to_bot.send_text("test message to bot")
    message_from_user = crew.bot.wait_next_incoming_message()
    assert not crew.bot.relayplugin.is_relay_group(message_from_user.chat)

    user_group = crew.user.create_group_chat(
        "test with user", contacts=[botcontact_user]
    )
    user_group.send_text("testing message to user group")
    message_from_user = crew.bot.wait_next_incoming_message()
    assert not crew.bot.relayplugin.is_relay_group(message_from_user.chat)