"""Модуль компонента тулбара (панели инструментов). Содержит класс ToolbarComponent для работы с элементами тулбара: - Управление кнопками и их подсказками - Проверка видимости элементов - Взаимодействие с панелью инструментов """ from playwright.sync_api import Page, expect, Locator from tools.logger import get_logger from locators.toolbar_locators import ToolbarLocators from elements.tooltip_button_element import TooltipButton from components.base_component import BaseComponent logger = get_logger("TOOLBAR") class ToolbarComponent(BaseComponent): """Компонент тулбара. Предоставляет методы для работы с панелью инструментов. Args: page (Page): Экземпляр страницы Playwright title (str): Заголовок тулбара """ def __init__(self, page: Page, title: str): """Инициализирует компонент тулбара с указанным заголовком.""" super().__init__(page) self.title = title self.buttons = [] def add_title(self, title: str) -> None: """Устанавливает новый заголовок тулбара. Args: title (str): Новый заголовок """ self.title = title def add_button(self, locator: Locator, name: str) -> None: """Добавляет кнопку в тулбар. Args: locator (Locator): Локатор кнопки name (str): Уникальное имя кнопки """ self.buttons.append(TooltipButton(self.page, locator, name)) def get_button_by_name(self, name: str) -> TooltipButton | None: """Возвращает кнопку по имени. Args: name (str): Имя кнопки Returns: TooltipButton | None: Найденная кнопка или None """ for button in self.buttons: if button.name == name: return button return None def click_button(self, name: str) -> None: """Кликает по кнопке тулбара. Args: name (str): Имя кнопки Raises: AssertionError: Если кнопка не найдена """ button = self.get_button_by_name(name) if button is None: raise AssertionError(f"Unsupported button name {name}") button.click() def is_button_present(self, name: str) -> bool: """Проверяет наличие кнопки. Args: name (str): Имя кнопки Returns: bool: True если кнопка присутствует Raises: AssertionError: Если имя кнопки не поддерживается """ button = self.get_button_by_name(name) if button is None: raise AssertionError(f"Unsupported button name {name}") return button.is_present(timeout=1000) # Ожидание 1 секунда def is_button_not_present(self, name: str) -> bool: """Проверяет отсутствие кнопки. Args: name (str): Имя кнопки Returns: bool: True если кнопка отсутствует Raises: AssertionError: Если имя кнопки не поддерживается """ button = self.get_button_by_name(name) if button is None: raise AssertionError(f"Unsupported button name {name}") return button.is_not_present(timeout=1000) # Ожидание 1 секунда def check_toolbar_presence(self, message: str) -> None: """Проверяет видимость тулбара. Args: message (str): Сообщение об ошибке если тулбар не виден """ locator = self.get_locator(ToolbarLocators.TITLE).filter(has_text=self.title) expect(locator).to_be_visible(), message def check_button_visibility(self, name: str) -> None: """Проверяет наличие и видимость кнопки с предварительной прокруткой к элементу. Args: name (str): Имя кнопки Raises: AssertionError: Если кнопка не найдена или не видна """ button = self.get_button_by_name(name) if button is None: raise AssertionError(f"Unsupported button name {name}") button.locator.scroll_into_view_if_needed() button.check_visibility(f"Button with name {name} is missing") def check_button_tooltip(self, name: str, tooltip: str) -> None: """Проверяет текст подсказки кнопки. Args: name (str): Имя кнопки tooltip (str): Ожидаемый текст подсказки Raises: AssertionError: Если текст подсказки не совпадает """ button = self.get_button_by_name(name) if button is None: raise AssertionError(f"Unsupported button name {name}") button.check_tooltip_with_text(ToolbarLocators.TOOLTIP, tooltip)