Сделано добавление в тулбар кнопки обычной и кнопки типа v-tabs

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:
         """Добавляет кнопку в модальное окно."""

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:
         """Возвращает кнопку по имени.

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 = {}