from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from pages.desktop.base import Base


class ContentReview(Base):
    URL_TEMPLATE = "reviewers/queue/content_review"

    # Current tab
    _content_review_tab_locator = (By.XPATH, "//li[@class='selected']//a[contains(text(),'Content Review')]")

    # Queue viewing
    _addon_column_locator = (By.XPATH, "//th[contains(text(),'Add-on')]")
    _flags_column_locator = (By.XPATH, "//th[contains(text(),'Flags')]")
    _last_updated_column_locator = (By.XPATH, "//a[contains(text(),'Last Updated')]")

    def wait_for_page_to_load(self):
        self.wait.until(
            lambda _: self.find_element(*self._content_review_tab_locator).is_displayed(),
            message="Content Review Page was not loaded",
        )
        return self

    # Queue Viewing --------------------------------------------------------

    @property
    def addon_column(self):
        return self.find_element(*self._addon_column_locator)

    @property
    def flag_column(self):
        return self.find_element(*self._flags_column_locator)

    @property
    def last_updated_column(self):
        return self.find_element(*self._last_updated_column_locator)

    # Assert Methods ----------------------------------------------------------

    def assert_queue_viewing_content_review(self):
        assert (
            self.addon_column.is_displayed(),
            "Add-on" in self.addon_column.text
        )
        assert (
            self.flag_column.is_displayed(),
            "Flags" in self.flag_column.text
        )
        assert (
            self.last_updated_column.is_displayed(),
            "Last Updated" in self.last_updated_column.text
        )