Changeset 9e426c1 in sasview for src/sas/qtgui/UnitTesting/MainWindowTest.py

Timestamp:

Jun 21, 2016 8:07:33 AM (8 years ago)

Author:

Piotr Rozyczko <piotr.rozyczko@…>

Branches:

ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc

Children:

f82ab8c

Parents:

1042dba

Message:

More main window items, system close, update checker, doc viewer etc.

File:

• src/sas/qtgui/UnitTesting/MainWindowTest.py (modified) (2 diffs)

src/sas/qtgui/UnitTesting/MainWindowTest.py

@@ -2 +2 @@
 import unittest
 
-from PyQt4.QtGui import *
-from PyQt4.QtTest import QTest
-from PyQt4.QtCore import *
+from PyQt4 import QtGui
+from PyQt4 import QtTest
+from PyQt4 import QtCore
 from mock import MagicMock
 
@@ -11 +11 @@
 from MainWindow import SplashScreen
 
-app = QApplication(sys.argv)
+app = QtGui.QApplication(sys.argv)
 
 class MainWindowTest(unittest.TestCase):
-    '''Test the Main Window GUI'''
+    """Test the Main Window GUI"""
     def setUp(self):
-        '''Create the GUI'''
+        """Create the GUI"""
 
         self.widget = MainSasViewWindow(None)
 
     def tearDown(self):
-        '''Destroy the GUI'''
-        self.widget.close()
+        """Destroy the GUI"""
         self.widget = None
 
     def testDefaults(self):
-        '''Test the GUI in its default state'''
-        self.assertIsInstance(self.widget, QMainWindow)
-        self.assertIsInstance(self.widget.centralWidget(), QWorkspace)
+        """Test the GUI in its default state"""
+        self.assertIsInstance(self.widget, QtGui.QMainWindow)
+        self.assertIsInstance(self.widget.centralWidget(), QtGui.QWorkspace)
         
     def testSplashScreen(self):
+        """ Test the splash screen """
+        splash = SplashScreen()
+        self.assertIsInstance(splash, QtGui.QSplashScreen)
+
+    def testExit(self):
         """
+        Test that the custom exit method is called on shutdown
         """
-        splash = SplashScreen()
-        self.assertIsInstance(splash, QSplashScreen)
+        # Must mask sys.exit, otherwise the whole testing process stops.
+        sys.exit = MagicMock()
+        QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes)
+
+        # Open, then close the main window
+        tmp_main = MainSasViewWindow(None)
+        tmp_main.close()
+
+        # See that the MessageBox method got called
+        self.assertTrue(QtGui.QMessageBox.question.called)
        
 if __name__ == "__main__":