diff --git a/components/toolbar_component.py b/components/toolbar_component.py index e7d5a39..1e2c4e5 100644 --- a/components/toolbar_component.py +++ b/components/toolbar_component.py @@ -25,7 +25,7 @@ class ToolbarComponent(BaseComponent): title (str): Заголовок тулбара """ - def __init__(self, page: Page, title: str): + def __init__(self, page: Page, title: str) -> None: """Инициализирует компонент тулбара с указанным заголовком.""" super().__init__(page) self.title = title @@ -67,7 +67,8 @@ class ToolbarComponent(BaseComponent): """ self.buttons.append(Button(self.page, locator, name)) - def get_button_by_name(self, name: str) -> TooltipButton | TabButton | Button | None: + def get_button_by_name(self, name: str + ) -> TooltipButton | TabButton | Button | None: """Возвращает кнопку по имени. Args: @@ -95,12 +96,13 @@ class ToolbarComponent(BaseComponent): raise AssertionError(f"Unsupported button name {name}") button.click() - def get_toolbar_title_text(self, locator: str = 'ToolbarLocators.TITLE', - filter_text: str = None, timeout: int = 5000) -> str: + def get_toolbar_title_text(self, locator: str = ToolbarLocators.TITLE, + filter_text: str | None = None, + timeout: int = 5000) -> str: """Получает заголовок тулбара окна. Args: - locator: Локатор для заголовка тулбара (по умолчанию 'ToolbarLocators.TITLE') + locator: Локатор для заголовка тулбара filter_text: Текст для фильтрации заголовка (опционально) timeout: Таймаут ожидания в миллисекундах @@ -122,7 +124,7 @@ class ToolbarComponent(BaseComponent): # Получаем текст заголовка title_text = title_locator.text_content().strip() - logger.info("Заголовок тулбара: '%s'", title_text) + logger.info("Toolbar title: '%s'", title_text) return title_text @@ -167,32 +169,35 @@ class ToolbarComponent(BaseComponent): Args: message (str): Сообщение об ошибке если тулбар не виден """ - - locator = self.get_locator(ToolbarLocators.TITLE).filter(has_text=self.title) + locator = self.get_locator(ToolbarLocators.TITLE).filter( + has_text=self.title + ) expect(locator).to_be_visible(), message - def check_toolbar_presence_by_locator(self, locator: str|Locator, message: str) -> None: + def check_toolbar_presence_by_locator(self, locator: str | Locator, + message: str) -> None: """Проверяет видимость тулбара. Args: + locator: Локатор тулбара message (str): Сообщение об ошибке если тулбар не виден """ - locator = self.get_locator(locator) expect(locator).to_be_visible(), message - def check_toolbar_presence_by_locator_and_title(self, locator: str|Locator, message: str) -> None: + def check_toolbar_presence_by_locator_and_title(self, locator: str | Locator, + message: str) -> None: """Проверяет видимость тулбара. Args: + locator: Локатор тулбара message (str): Сообщение об ошибке если тулбар не виден """ - locator = self.get_locator(locator).filter(has_text=self.title) expect(locator).to_be_visible(), message def check_button_visibility(self, name: str) -> None: - """Проверяет наличие и видимость кнопки с предварительной прокруткой к элементу. + """Проверяет наличие и видимость кнопки с предварительной прокруткой. Args: name (str): Имя кнопки