Add a GUI test for Enum.create_editor

traits/testing/optional_dependencies.py

@@ -42,6 +42,9 @@ def optional_import(name):
 numpy = optional_import("numpy")
 requires_numpy = unittest.skipIf(numpy is None, "NumPy not available")
 
+pyface = optional_import("pyface")
+requires_pyface = unittest.skipIf(pyface is None, "Pyface not available")
+
 sphinx = optional_import("sphinx")
 requires_sphinx = unittest.skipIf(sphinx is None, "Sphinx not available")
 

traits/tests/test_enum.py

@@ -14,6 +14,15 @@
 from traits.api import (
     Any, BaseEnum, Enum, HasTraits, List, Property, TraitError)
 
+from traits.testing.optional_dependencies import pyface, requires_traitsui
+
+if pyface is not None:
+    GuiTestAssistant = pyface.toolkit.toolkit_object(
+        "util.gui_test_assistant:GuiTestAssistant")
+else:
+    class GuiTestAssistant:
+        pass
+
 
 class FooEnum(enum.Enum):
     foo = 0
@@ -110,6 +119,13 @@ class EnumCollectionExample(HasTraits):
     slow_enum = BaseEnum("yes", "no", "maybe")
 
 
+class EnumCollectionGUIExample(EnumCollectionExample):
+    # Override attributes that may fail GUI test
+    # until traitsui #781 is fixed.
+    int_set_enum = Enum("int", "set")
+    correct_int_set_enum = Enum("int", "set")
+
+
 class EnumTestCase(unittest.TestCase):
     def test_valid_enum(self):
         example_model = ExampleModel(root="model1")
@@ -296,3 +312,27 @@ def test_base_enum(self):
         with self.assertRaises(TraitError):
             obj.slow_enum = "perhaps"
         self.assertEqual(obj.slow_enum, "no")
+
+
+@requires_traitsui
+class TestGui(GuiTestAssistant, unittest.TestCase):
+
+    def test_create_editor(self):
+        obj = EnumCollectionGUIExample()
+        user_editable_traits = obj.class_editable_traits()
+
+        # Create a UI window
+        ui = obj.edit_traits()
+
+        for t in user_editable_traits:
+
+            with self.subTest(t=t):
+                editor = getattr(ui.info, t)
+                values = obj.trait(t).trait_type.values
+
+                # Try setting all valid values for the Enum trait
+                for value in values:
+                    with self.assertTraitChangesInEventLoop(
+                            obj, t, lambda instance: getattr(
+                                instance, t) == value, 0, 3):
+                        self.gui.set_trait_later(editor, 'value', value)