import time
import pytest
import numpy
from tests.base_test_case import SingleDeviceTestCase
from random import randint
from views.sign_in_view import SignInView

running_time = int(pytest.config.getoption('running_time'))
messages_number = int(pytest.config.getoption('messages_number'))
stop = int(time.time()) + running_time
public_keys = pytest.config.getoption('public_keys').split()
repeats = 24 / len(public_keys) if public_keys else 0


@pytest.mark.chatbot
class TestChatBot(SingleDeviceTestCase):

    def setup_method(self, method, **kwargs):
        super(TestChatBot, self).setup_method(method, maxDuration=10800)

    @pytest.mark.parametrize('key', numpy.repeat(public_keys, repeats))
    def test_one_to_one_chatbot(self, key):
        sign_in = SignInView(self.driver)
        home = sign_in.create_user()
        chat = home.add_contact(key)

        message_text = 'test'
        chat.chat_message_input.send_keys(message_text)
        chat.send_message_button.click()

        message = 'Message # %s, sent by e2e test, to %s '
        counter = 0
        while int(time.time()) < stop:
            counter += 1
            time.sleep(randint(60, 120))
            chat.chat_message_input.send_keys(message % (counter, key))
            chat.send_message_button.click()

    @pytest.mark.parametrize('number', list(range(int(pytest.config.getoption('device_number')))))
    def test_chatbot_public_chat(self, number):
        sign_in = SignInView(self.driver)
        home = sign_in.create_user()
        chat_name = pytest.config.getoption('chat_name')
        chat = home.join_public_chat(chat_name)
        counter = 0
        while counter <= messages_number / 24:
            counter += 1
            chat.chat_message_input.send_keys('Test message #%s from sender %s' % (counter, number))
            chat.send_message_button.click()