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

from .base import FunctionalTest

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_list_items(self):
        self.browser.get(self.live_server_url)
        self.get_item_input_box().send_keys(Keys.ENTER)

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

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

        self.get_item_input_box().send_keys(Keys.ENTER)
        self.wait_for_row_in_list_table("1. Purchase milk")

        self.get_item_input_box().send_keys(Keys.ENTER)

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

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

    def test_cannot_add_duplicate_items(self):
        self.browser.get(self.live_server_url)
        self.add_list_item("Witness divinity")

        self.get_item_input_box().send_keys("Witness divinity")
        self.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 list",
            )
        )

    def test_error_messages_are_cleared_on_input(self):
        self.browser.get(self.live_server_url)
        self.add_list_item("Gobbledygook")
        self.get_item_input_box().send_keys("Gobbledygook")
        self.get_item_input_box().send_keys(Keys.ENTER)
        self.wait_for(
            lambda: self.assertTrue(self.get_error_element().is_displayed())
        )

        self.get_item_input_box().send_keys("a")

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