import logging

import allure
from assertpy import assert_that
from ui_automation_framework.core import BasePage, TestStatus
from ui_automation_framework.utils import logger as cl

from app.src.pages.login_page import LoginPage


class PlaylistCategoriesPage(BasePage):
    log = cl.Logger(logging.DEBUG)

    def __init__(self, driver):
        super().__init__(driver)
        self.driver = driver
        self.ts = TestStatus(self.driver)
        self.loginPage = LoginPage(self.driver)

    @allure.step("Playlist Categories Page is present")
    def playlist_categories_page_is_present(self):
        locator = "//span[contains(text(),'Add Category')]"
        self.wait_for_element_visible(locator=locator, locator_type="xpath")
        button = self.is_element_displayed(locator=locator, locator_type="xpath")
        assert_that(button).is_true()