diff --git a/components/modal_window_component.py b/components/modal_window_component.py index 0929ba0..33511cf 100644 --- a/components/modal_window_component.py +++ b/components/modal_window_component.py @@ -40,7 +40,7 @@ class ModalWindowComponent(BaseComponent): def add_toolbar_button(self, locator: str, name: str) -> None: """Добавляет кнопку в панель инструментов модального окна.""" - self.toolbar.add_button(locator, name) + self.toolbar.add_tooltip_button(locator, name) def add_button(self, locator: str, name: str) -> None: """Добавляет кнопку в модальное окно.""" diff --git a/components/toolbar_component.py b/components/toolbar_component.py index 8cd8083..4372c6b 100644 --- a/components/toolbar_component.py +++ b/components/toolbar_component.py @@ -10,6 +10,8 @@ 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 elements.tab_button_element import TabButton +from elements.button_element import Button from components.base_component import BaseComponent logger = get_logger("TOOLBAR") @@ -39,6 +41,26 @@ class ToolbarComponent(BaseComponent): self.title = title + def add_tooltip_button(self, locator: Locator, name: str) -> None: + """Добавляет кнопку с подсказкой в тулбар. + + Args: + locator (Locator): Локатор кнопки + name (str): Уникальное имя кнопки + """ + + self.buttons.append(TooltipButton(self.page, locator, name)) + + def add_tab_button(self, locator: Locator, name: str) -> None: + """Добавляет кнопку типа v-tab в тулбар. + + Args: + locator (Locator): Локатор кнопки + name (str): Уникальное имя кнопки + """ + + self.buttons.append(TabButton(self.page, locator, name)) + def add_button(self, locator: Locator, name: str) -> None: """Добавляет кнопку в тулбар. @@ -47,7 +69,7 @@ class ToolbarComponent(BaseComponent): name (str): Уникальное имя кнопки """ - self.buttons.append(TooltipButton(self.page, locator, name)) + self.buttons.append(Button(self.page, locator, name)) def get_button_by_name(self, name: str) -> TooltipButton | None: """Возвращает кнопку по имени. diff --git a/pages/users_tab.py b/pages/users_tab.py index c9893c1..f5c17c4 100644 --- a/pages/users_tab.py +++ b/pages/users_tab.py @@ -40,9 +40,9 @@ class UsersTab(BasePage): ).get_by_role("button").nth(1) self.toolbar = ToolbarComponent(page, "Пользователи") - self.toolbar.add_button(locator_button_1, "edit") - self.toolbar.add_button(locator_button_1, "add_user") - self.toolbar.add_button(locator_button_2, "close") + self.toolbar.add_tooltip_button(locator_button_1, "edit") + self.toolbar.add_tooltip_button(locator_button_1, "add_user") + self.toolbar.add_tooltip_button(locator_button_2, "close") self.users_table = TableComponent(page) self.modal_windows = {}