githubmirror/AgentPilot

tests/test_gui.py
152 строки4 KB
0.4.1
20 янв 2025, 06:44
20 янв 2025, 06:446bf3b2c
import sys
import time
import unittest
import pyautogui

from PySide6.QtGui import Qt
from PySide6.QtWidgets import QApplication
from PySide6.QtTest import QTest
from src.gui.main import Main
from src.gui.widgets import IconButton


# TEST LIST
# - Open app with no internet/
# - Open convos page
# -   Double click context/
# -   Chat button/
# -   Delete button/
# -   Search bar/
# -   New folder/
# -   Right click context items/

# - Open agent page
# -   New agent button/
# -   Double click agent/
# -   Chat button/
# -   New folder/
# -   Delete button/
# -     Info tab settings
# -       Change plugin
# -       Change avatar & name
# -     Chat tab settings
# -       Message tab settings
# -       Preload tab settings
# -       Group tab settings
# -     Tools tab settings
# -     Voice tab settings

# - Open settings page
# -   System tab settings
# -     Dev mode
# -     Telemetry
# -     Always on top
# -     Default model
# -     Auto title
# -   Display tab settings
# -   Model tab settings
# -     Edit api
# -     New model
# -     Delete model
# -     Edit model
# -   Blocks tab settings
# -   Sandbox tab settings

# - Chat page
# -   Edit title
# -   Navigation buttons
# -   Openai agent
# -   Perplexity agent
# -   Multi agent mixed providers
# -   Context placeholders
# -   Hide responses
# -   Decoupled scroll
# -   Stop button

# - Plugins
# -   Open interpreter
# -   Openai assistant
# -   Memgpt
# -   Autogen agents (with and without context plugin)
# -   Autogen context plugin
#

# - Other
# -   Push/pull buttons


class TestApp(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.app = QApplication(sys.argv)
        cls.main = Main()

    @classmethod
    def tearDownClass(cls):
        # Clean up after all tests
        cls.main.close()
        cls.app.quit()

    def setUp(self):
        self.main.show()
        self.main.raise_()
        QTest.qWait(1000)  # Wait for the window to show

        self.btn_settings = self.main.main_menu.settings_sidebar.page_buttons['Settings']
        self.page_settings = self.main.main_menu.pages['Settings']

    def click_widget(self, widget):
        x, y = self.get_widget_coords(widget)
        pyautogui.moveTo(x, y, duration=0.3)
        QTest.mouseClick(widget, Qt.LeftButton)
        QTest.qWait(500)

    def goto_page(self, page_name):
        btn = self.main.main_menu.settings_sidebar.page_buttons.get(page_name, None)
        page = self.main.main_menu.pages.get(page_name, None)
        if btn:
            if not btn.isVisible():
                btn = None
                page = None
        if not btn:
            btn = self.page_settings.settings_sidebar.page_buttons.get(page_name, None)
            page = self.page_settings.settings_sidebar.pages.get(page_name, None)
            if not btn.isVisible():
                btn = None
                page = None
        if not btn:
            raise ValueError(f'Page {page_name} not found')

        self.click_widget(btn)
        return page

    def iterate_button_bar(self, button_bar):
        for attr_name, obj in button_bar.__dict__.items():
            if not isinstance(obj, IconButton) or not obj.isVisible() or not obj.isEnabled():
                continue
            self.click_widget(obj)

    def get_widget_coords(self, widget):
        center = widget.rect().center()
        global_center = widget.mapToGlobal(center)
        return global_center.x(), global_center.y()

    def test_chat_page(self):
        page_contexts = self.goto_page('Contexts')
        self.iterate_button_bar(page_contexts.tree_buttons)

        #
        # # Check UI state
        # # For example, check if a label text has changed
        # label = self.window.findChild(QLabel, "yourLabelName")
        # self.assertEqual(label.text(), "Expected Text")
        #
        # # Or check if a widget is visible
        # widget = self.window.findChild(QWidget, "someWidgetName")
        # self.assertTrue(widget.isVisible())
        # endregion


if __name__ == '__main__':
    unittest.main()