Changeset db5cd8d in sasview for src/sas/qtgui/UnitTesting/PlotPropertiesTest.py

Timestamp:

Jan 12, 2017 4:09:23 AM (7 years ago)

Author:

Piotr Rozyczko <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:

2e3e959

Parents:

87cc73a

Message:

Added tests for PlotProperties?

File:

• src/sas/qtgui/UnitTesting/PlotPropertiesTest.py (modified) (4 diffs)

src/sas/qtgui/UnitTesting/PlotPropertiesTest.py

@@ -1 +1 @@
 import sys
 import unittest
+from mock import MagicMock
 
 from PyQt4 import QtGui
@@ -6 +7 @@
 # set up import paths
 import path_prepare
-
-from UnitTesting.TestUtils import WarningNotImplemented
 
 # Local
@@ -19 +18 @@
         '''Create the PlotProperties'''
 
-        self.widget = PlotProperties(None)
+        self.widget = PlotProperties(None, 
+                         color=1,
+                         marker=3,
+                         marker_size=10,
+                         legend="LL")
 
     def tearDown(self):
@@ -30 +33 @@
         self.assertIsInstance(self.widget, QtGui.QDialog)
         self.assertEqual(self.widget.windowTitle(), "Modify Plot Properties")
+
+        # Check the combo boxes
+        self.assertEqual(self.widget.cbColor.currentText(), "Green")
+        self.assertEqual(self.widget.cbShape.currentText(), "Triangle Down")
+        self.assertEqual(self.widget.txtLegend.text(), "LL")
+        self.assertEqual(self.widget.sbSize.value(), 10)
         
     def testOnColorChange(self):
         '''Test the response to color change event'''
-        WarningNotImplemented(sys._getframe().f_code.co_name)
+        # Accept the new color
+        QtGui.QColorDialog.getColor = MagicMock(return_value=QtGui.QColor(255, 0, 255))
+
+        self.widget.onColorChange(None)
+
+        self.assertEqual(self.widget.color(), "#ff00ff")
+        self.assertTrue(self.widget.custom_color)
+        self.assertEqual(self.widget.cbColor.currentIndex(), 7)
+        self.assertEqual(self.widget.cbColor.currentText(), "Custom")
+
+        # Reset the color - this will remove "Custom" from the combobox
+        # and set its index to "Green"
+        self.widget.cbColor.setCurrentIndex(1)
+        self.assertEqual(self.widget.cbColor.currentText(), "Green")
+
+        # Cancel the dialog now
+        bad_color = QtGui.QColor() # constructs an invalid color
+        QtGui.QColorDialog.getColor = MagicMock(return_value=bad_color)
+        self.widget.onColorChange(None)
+
+        self.assertEqual(self.widget.color(), 1)
+        self.assertFalse(self.widget.custom_color)
+        self.assertEqual(self.widget.cbColor.currentIndex(), 1)
+        self.assertEqual(self.widget.cbColor.currentText(), "Green")
+
 
     def testOnColorIndexChange(self):
         '''Test the response to color index change event'''
-        WarningNotImplemented(sys._getframe().f_code.co_name)
-    
+        # Intitial population of the color combo box
+        self.widget.onColorIndexChange(0)
+        self.assertEqual(self.widget.cbColor.count(), 7)
+        # Block the callback so we can update the cb
+        self.widget.cbColor.blockSignals(True)
+        # Add the Custom entry
+        self.widget.cbColor.addItems(["Custom"])
+        # Unblock the callback
+        self.widget.cbColor.blockSignals(False)
+        # Assert the new CB
+        self.assertEqual(self.widget.cbColor.count(), 8)
+        # Call the method
+        self.widget.onColorIndexChange(0)
+        # see that the Custom entry disappeared
+        self.assertEqual(self.widget.cbColor.count(), 7)
+        self.assertEqual(self.widget.cbColor.findText("Custom"), -1)
 
 if __name__ == "__main__":