python - 在测试 PyQt 应用程序时，pytest 在终端和使用 VSCode pytest 插件时都会出现问题

import logging from PyQt6.QtCore import Qt from PyQt6.QtGui import QIcon, QAction from PyQt6.QtWidgets import QMainWindow, QStackedWidget, QToolButton, QToolBar, QWidget, QApplication import sys class MainWindow(QMainWindow): """ Main window class - this will house the main window for the desktop application. """ def __init__(self) -> None: # setup basic page components super().__init__() self.setWindowTitle("Demo App") self.setWindowState(Qt.WindowState.WindowMaximized) # call page setup here self.setup_application() @property def logger(self) -> logging.Logger: return logging.getLogger(f"{__name__}.{self.__class__.__name__}") def setup_application(self) -> None: self.logger.debug("Starting MainWindow setup") # setup stack of widgets (different views) self.setup_stack() # setup toolbar for view navigation self.setup_toolbar() def setup_stack(self) -> None: # all views to be added into application self.first = QWidget() self.second = QWidget() self.third = QWidget() self.fourth = QWidget() # add widgets to stack self.stack = QStackedWidget(self) self.stack.addWidget(self.first) self.stack.addWidget(self.second) self.stack.addWidget(self.third) self.stack.addWidget(self.fourth) # set indexes of views in stack self.FIRST_INDEX = 0 self.SECOND_INDEX = 1 self.THIRD_INDEX = 2 self.FOURTH_INDEX = 3 # finally, set central widget for application self.setCentralWidget(self.stack) def setup_toolbar(self) -> None: # setup actions for views' toolbuttons self.first_action = QAction("Page 1", self) self.second_action = QAction("Page 2", self) self.third_action = QAction("Page 3", self) self.fourth_action = QAction("Page 4", self) self.first_action.triggered.connect( lambda: self.change_window(self.FIRST_INDEX) ) self.second_action.triggered.connect(lambda: self.change_window(self.SECOND_INDEX)) self.third_action.triggered.connect( lambda: self.change_window(self.THIRD_INDEX) ) self.fourth_action.triggered.connect( lambda: self.change_window(self.FOURTH_INDEX) ) # setup buttons for toolbar self.first_button = QToolButton() self.first_button.setDefaultAction(self.first_action) self.second_button = QToolButton() self.second_button.setDefaultAction(self.second_action) self.third_button = QToolButton() self.third_button.setDefaultAction(self.third_action) self.fourth_button = QToolButton() self.fourth_button.setDefaultAction(self.fourth_action) # finally, setup toolbar for top of screen self.toolbar = QToolBar() self.toolbar.addWidget(self.first_button) self.toolbar.addWidget(self.second_button) self.toolbar.addWidget(self.third_button) self.toolbar.addWidget(self.fourth_button) self.toolbar.setMovable(False) self.addToolBar(self.toolbar) def change_window(self, index: int) -> None: # TODO: add bold text/button highlighting to make it clear from # the toolbar which view you are on # set application view to desired view, just doing this for now self.stack.setCurrentIndex(index)

def main() -> None: # TODO: Figure way to get package name dynamically app = QApplication(sys.argv) window = MainWindow() window.show() sys.exit(app.exec()) # invoke application if __name__ == "__main__": main()

# add imports for any files to be tested in windows # package imports from PyQt6.QtCore import Qt import pytest from pytest_mock import MockerFixture from pytestqt.plugin import QtBot import sys # file imports from windows.main_window import MainWindow # tests should start with `test_` and be followed by function name # example: test_my_fancy_func() @pytest.fixture def mw() -> MainWindow: return MainWindow() class TestMainWindow: def test_setup_application(self, qtbot, mw: MainWindow, mocker) -> None: mocker.patch.object(MainWindow, attribute="setup_stack", return_value=None) mocker.patch.object(MainWindow, attribute="setup_toolbar", return_value=None) mw.setup_application() def test_setup_stack(self, qtbot, mw: MainWindow) -> None: mw.setup_stack() assert mw.stack.count() == 4 assert mw.stack.indexOf(mw.first) == 0 assert mw.stack.indexOf(mw.second) == 1 assert mw.stack.indexOf(mw.third) == 2 assert mw.stack.indexOf(mw.fourth) == 3 assert mw.centralWidget() == mw.stack def test_setup_toolbar(self, qtbot, mw: MainWindow) -> None: mw.setup_stack() mw.setup_toolbar() # check action text assert mw.first_action.text() == "Page 1" assert mw.second_action.text() == "Page 2" assert mw.third_action.text() == "Page 3" assert mw.fourth_action.text() == "Page 4" # check toolbuttons set up properly, and toolbar settings correct assert mw.first_button.defaultAction() == mw.first_action assert mw.second_button.defaultAction() == mw.second_action assert mw.third_button.defaultAction() == mw.third_action assert mw.fourth_button.defaultAction() == mw.fourth_action assert mw.toolbar.isMovable() is False assert mw.toolBarArea(mw.toolbar) == Qt.ToolBarArea.TopToolBarArea # try clicking buttons and checking responses mw.first_action.trigger() assert mw.stack.currentIndex() == 0 mw.second_action.trigger() assert mw.stack.currentIndex() == 1 mw.third_action.trigger() assert mw.stack.currentIndex() == 2 mw.fourth_action.trigger() assert mw.stack.currentIndex() == 3