# tests/front_end_tests.py

import time
from flask import url_for

class TestLogin(TestBase):

    def test_login(self):
        self.driver.find_element_by_id("login_link").click()
        time.sleep(1)
        self.driver.find_element_by_id("email").send_keys(test_employee1_email)
        self.driver.find_element_by_id("password").send_keys(
            test_employee1_password)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        assert url_for('home.dashboard') in self.driver.current_url
        username_greeting = self.driver.find_element_by_id(
            "username_greeting").text
        assert "Olá, usuário1 !" in username_greeting

    def test_admin_login(self):
        self.driver.find_element_by_id("login_link").click()
        time.sleep(1)
        self.driver.find_element_by_id("email").send_keys(test_admin_email)
        self.driver.find_element_by_id("password").send_keys(
            test_admin_password)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        assert url_for('home.admin_dashboard') in self.driver.current_url
        username_greeting = self.driver.find_element_by_id(
            "username_greeting").text
        assert "Olá, admin!" in username_greeting

    def test_login_invalid_email_format(self):
        self.driver.find_element_by_id("login_link").click()
        time.sleep(1)
        self.driver.find_element_by_id("email").send_keys("invalid")
        self.driver.find_element_by_id("password").send_keys(
            test_employee1_password)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        error_message = self.driver.find_element_by_class_name(
            "help-block").text
        assert "Email inválido." in error_message

    def test_login_wrong_email(self):
        # Click login menu link
        self.driver.find_element_by_id("login_link").click()
        time.sleep(1)
        self.driver.find_element_by_id("email").send_keys(test_employee2_email)
        self.driver.find_element_by_id("password").send_keys(
            test_employee1_password)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Invalid email or password" in error_message

    def test_login_wrong_password(self):
        self.driver.find_element_by_id("login_link").click()
        time.sleep(1)
        self.driver.find_element_by_id("email").send_keys(test_employee1_email)
        self.driver.find_element_by_id("password").send_keys(
            "invalid")
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Invalid email or password" in error_message

class CreateObjects(object):

    def login_admin_user(self):
        login_link = self.get_server_url() + url_for('auth.login')
        self.driver.get(login_link)
        self.driver.find_element_by_id("email").send_keys(test_admin_email)
        self.driver.find_element_by_id("password").send_keys(
            test_admin_password)
        self.driver.find_element_by_id("submit").click()

    def login_test_user(self):
        login_link = self.get_server_url() + url_for('auth.login')
        self.driver.get(login_link)
        self.driver.find_element_by_id("email").send_keys(test_employee1_email)
        self.driver.find_element_by_id("password").send_keys(
            test_employee1_password)
        self.driver.find_element_by_id("submit").click()

class TestDados(CreateObjects, TestBase):
    # equipamento existe
    def test_view_existing_equipment(self):
        self.login_admin_user()
        self.driver.find_element_by_id("equipments_link").click()
        time.sleep(1)
        self.driver.find_element_by_class_name("btn").click()
        time.sleep(1)
        self.driver.find_element_by_id("codigo").send_keys(test_equipment1_name)
        self.driver.find_element_by_id("description").send_keys(test_equipment1_description)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Error: equipment name don't exists" in error_message
        self.assertEqual(Equipment.query.count(), 1)

    def test_view_equipment_data(self): # B e C
        # dados existem e estão no mapa
        self.login_test_user()
        self.driver.find_element_by_id("equipments_link").click()
        time.sleep(1)
        self.driver.find_element_by_class_name("btn").click()
        time.sleep(1)
        self.driver.find_element_by_id("codigo").send_keys(test_equipment1_name)
        self.driver.find_element_by_id("description").send_keys(test_equipment1_description)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        self.driver.find_element_by_id("data_equipment_link").click()
        time.sleep(1)
        success_message = self.driver.find_element_by_class_name("alert").text
        assert "Equipment data plotted in map." in success_message
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Error: no data." in error_message

    def test_view_equipment_log_data(self): # A
        # dados chegando
        self.login_admin_user()
        self.driver.find_element_by_id("equipments_link").click()
        time.sleep(1)
        self.driver.find_element_by_class_name("btn").click()
        time.sleep(1)
        self.driver.find_element_by_id("codigo").send_keys(test_equipment1_name)
        time.sleep(2)
        self.driver.find_element_by_id("datalog_equipment_link").click()
        time.sleep(1)
        success_message = self.driver.find_element_by_class_name("alert").text
        assert "Receiving equipment data." in success_message
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Error: no data." in error_message

    def test_view_equipment_data(self): # B e D
        # dados existem e estão no mapa. Admin seleciona equipamentos
        self.login_admin_user()
        self.driver.find_element_by_id("equipments_link").click()
        time.sleep(1)
        self.driver.find_element_by_class_name("btn").click()
        time.sleep(1)
        self.driver.find_element_by_id("codigo").send_keys(test_equipment1_name)
        self.driver.find_element_by_id("description").send_keys(test_equipment1_description)
        self.driver.find_element_by_id("submit").click()
        time.sleep(2)
        self.driver.find_element_by_id("data_equipment_link").click()
        time.sleep(1)
        success_message = self.driver.find_element_by_class_name("alert").text
        assert "Equipment data plotted in map." in success_message
        error_message = self.driver.find_element_by_class_name("alert").text
        assert "Error: no data." in error_message