diff --git a/qubesmanager/tests/test_qube_manager.py b/qubesmanager/tests/test_qube_manager.py index a02a508..f876a1d 100644 --- a/qubesmanager/tests/test_qube_manager.py +++ b/qubesmanager/tests/test_qube_manager.py @@ -1396,7 +1396,7 @@ class QubeManagerThreadTest(unittest.TestCase): class VMShutdownMonitorTest(unittest.TestCase): - @unittest.mock.patch('PyQt5.QtWidgets.QMessageBox') + @unittest.mock.patch('qubesmanager.qube_manager.QMessageBox') @unittest.mock.patch('PyQt5.QtCore.QTimer') def test_01_vm_shutdown_correct(self, mock_timer, mock_question): mock_vm = unittest.mock.Mock() @@ -1411,7 +1411,7 @@ class VMShutdownMonitorTest(unittest.TestCase): self.assertEqual(mock_timer.call_count, 0) monitor.restart_vm_if_needed.assert_called_once_with() - @unittest.mock.patch('PyQt5.QtWidgets.QMessageBox') + @unittest.mock.patch('qubesmanager.qube_manager.QMessageBox') @unittest.mock.patch('PyQt5.QtCore.QTimer.singleShot') def test_02_vm_not_shutdown_wait(self, mock_timer, mock_question): mock_question().clickedButton.return_value = 1 @@ -1428,7 +1428,7 @@ class VMShutdownMonitorTest(unittest.TestCase): self.assertEqual(mock_timer.call_count, 1) - @unittest.mock.patch('PyQt5.QtWidgets.QMessageBox') + @unittest.mock.patch('qubesmanager.qube_manager.QMessageBox') @unittest.mock.patch('PyQt5.QtCore.QTimer.singleShot') def test_03_vm_kill(self, mock_timer, mock_question): mock_question().clickedButton.return_value = 1 @@ -1448,7 +1448,7 @@ class VMShutdownMonitorTest(unittest.TestCase): mock_vm.kill.assert_called_once_with() monitor.restart_vm_if_needed.assert_called_once_with() - @unittest.mock.patch('PyQt5.QtWidgets.QMessageBox') + @unittest.mock.patch('qubesmanager.qube_manager.QMessageBox') @unittest.mock.patch('PyQt5.QtCore.QTimer.singleShot') def test_04_check_later(self, mock_timer, mock_question): mock_vm = unittest.mock.Mock()