from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys

from .base import FunctionalTest
from .note_page import NotePage


class ItemValidationTest(FunctionalTest):
    # Helper functions
    def get_error_element(self):
        return self.browser.find_element(By.CSS_SELECTOR, ".invalid-feedback")

    # Test methods
    def test_cannot_add_empty_note_items(self):
        self.create_pre_authenticated_session("disco@test.io")
        self.browser.get(self.live_server_url)
        note_page = NotePage(self)
        note_page.get_item_input_box().send_keys(Keys.ENTER)

        self.wait_for(
            lambda: self.browser.find_element(By.CSS_SELECTOR, "#id_text:invalid")
        )

        note_page.get_item_input_box().send_keys("Purchase milk")
        self.wait_for(
            lambda: self.browser.find_element(By.CSS_SELECTOR, "#id_text:valid")
        )

        note_page.get_item_input_box().send_keys(Keys.ENTER)
        note_page.wait_for_row_in_note_table("Purchase milk", 1)

        note_page.get_item_input_box().send_keys(Keys.ENTER)

        note_page.wait_for_row_in_note_table("Purchase milk", 1)
        self.wait_for(
            lambda: self.browser.find_element(By.CSS_SELECTOR, "#id_text:invalid")
        )

        note_page.get_item_input_box().send_keys("Make tea")
        self.wait_for(
            lambda: self.browser.find_element(
                By.CSS_SELECTOR,
                "#id_text:valid",
            )
        )
        note_page.get_item_input_box().send_keys(Keys.ENTER)
        note_page.wait_for_row_in_note_table("Make tea", 2)

    def test_cannot_add_duplicate_items(self):
        self.create_pre_authenticated_session("disco@test.io")
        self.browser.get(self.live_server_url)
        note_page = NotePage(self)
        note_page.add_note_item("Witness divinity")

        note_page.get_item_input_box().send_keys("Witness divinity")
        note_page.get_item_input_box().send_keys(Keys.ENTER)

        self.wait_for(
            lambda: self.assertEqual(
                self.get_error_element().text,
                "You've already logged this to your note",
            )
        )

    def test_error_messages_are_cleared_on_input(self):
        self.create_pre_authenticated_session("disco@test.io")
        self.browser.get(self.live_server_url)
        note_page = NotePage(self)
        note_page.add_note_item("Gobbledygook")
        note_page.get_item_input_box().send_keys("Gobbledygook")
        note_page.get_item_input_box().send_keys(Keys.ENTER)
        self.wait_for(
            lambda: self.assertTrue(self.get_error_element().is_displayed())
        )

        note_page.get_item_input_box().send_keys("a")

        self.wait_for(
            lambda: self.assertFalse(self.get_error_element().is_displayed())
        )