from allure import step
import re
from playwright.sync_api import expect
from src.ui_elements.base_ui_element import BaseUIElement


class SwitchUIElement(BaseUIElement):
    @step("Is {0} checked - {checked}")
    def is_checked(self, checked):
        self.log.info(f"Does element {self.locator} contain class <switch--checked> - {checked}")
        if checked is True:
            expect(self._element).to_have_class(re.compile(r"switch--checked"))
        elif checked is False:
            expect(self._element).not_to_have_class(re.compile(r"switch--checked"))

    def __str__(self):
        return f"Switch {self.name}"