from allure import step
from assertpy import assert_that
from playwright.sync_api import Page
from playwright.sync_api import expect
from src.ui_elements.base_ui_element import BaseUIElement
from src.pages.categories_page import CategoriesPage
from src.locators.category_locators import CategoryLocators


class CategoryPage(CategoriesPage):

    def __init__(self, page: Page):
        super().__init__(page)
        self.page = page

        self.category_container = BaseUIElement(self.page, CategoryLocators.CATEGORY_CONTAINER, 'Category container')
        self.entity_card_img = BaseUIElement(self.page, CategoryLocators.ENTITY_CARD_IMG, 'Category Entity image')
        self.entity_card_dsp = BaseUIElement(self.page, CategoryLocators.ENTITY_CARD_DSP, 'Category Entity dsp')

    @step("Category page is loaded")
    def is_page_loaded(self):
        self.category_container.exists()