def setUp(self):
self.widget = wx.Window()
self.registry = get_widget_registry()
self.target = TargetWithControl(self.widget)
self.good_wrapper = UIWrapper(
target=self.target,
registries=[self.registry],
)
class TestQtControlWidgetRegistry(unittest.TestCase):
""" Test the interface of AbstractTargetRegistry for QWidget's registry
"""
def setUp(self):
self.widget = QtGui.QWidget()
self.registry = get_widget_registry()
self.target = TargetWithControl(self.widget)
self.good_wrapper = UIWrapper(
target=self.target,
registries=[self.registry],
)
def test_is_enabled(self):
self.assertTrue(self.good_wrapper.inspect(IsEnabled()))
def test_is_disabled(self):
self.widget.setEnabled(False)
self.assertFalse(self.good_wrapper.inspect(IsEnabled()))
def test_is_visible(self):
self.widget.setVisible(True)
self.assertTrue(self.good_wrapper.inspect(IsVisible()))
def test_is_invisible(self):
self.widget.setVisible(False)
self.assertFalse(self.good_wrapper.inspect(IsVisible()))
def test_get_interactions_good_target(self):
self.assertEqual(
self.registry._get_interactions(self.target),
set([IsEnabled, IsVisible])
)
def test_get_interactions_bad_target(self):
self.assertEqual(self.registry._get_interactions(None), set())
def test_get_interaction_doc(self):
self.assertGreater(
len(self.registry._get_interaction_doc(self.target, IsEnabled)), 0
)
self.assertGreater(
len(self.registry._get_interaction_doc(self.target, IsVisible)), 0
)
def test_get_location_solver(self):
# There are currently no solvers
with self.assertRaises(LocationNotSupported):
self.registry._get_solver(self.target, None)
def test__get_locations(self):
self.assertEqual(self.registry._get_locations(self.target), set())
def test_error_get_location_doc(self):
with self.assertRaises(LocationNotSupported):
self.registry._get_location_doc(self.target, None)
class TestWxControlWidgetRegistry(unittest.TestCase):
def setUp(self):
self.widget = wx.Window()
self.registry = get_widget_registry()
self.target = TargetWithControl(self.widget)
self.good_wrapper = UIWrapper(
target=self.target,
registries=[self.registry],
)
def test_is_enabled(self):
self.widget.Enable(True)
self.assertTrue(self.good_wrapper.inspect(IsEnabled()))
def test_is_visible(self):
self.widget.Show(True)
self.assertTrue(self.good_wrapper.inspect(IsVisible()))
def test_is_invisible(self):
self.widget.Hide()
self.assertFalse(self.good_wrapper.inspect(IsVisible()))
def test_get_interactions_good_target(self):
self.assertEqual(
self.registry._get_interactions(self.target),
set([IsEnabled, IsVisible])
)
def test_get_interactions_bad_target(self):
self.assertEqual(self.registry._get_interactions(None), set())
def test_get_interaction_doc(self):
self.assertGreater(
len(self.registry._get_interaction_doc(self.target, IsEnabled)), 0
)
self.assertGreater(
len(self.registry._get_interaction_doc(self.target, IsVisible)), 0
)
def test_get_location_solver(self):
# There are currently no solvers
with self.assertRaises(LocationNotSupported):
self.registry._get_solver(self.target, None)
def test_get_locations(self):
self.assertEqual(self.registry._get_locations(self.target), set())
def test_error_get_location_doc(self):
with self.assertRaises(LocationNotSupported):
self.registry._get_location_doc(self.target, None)
def example_ui_wrapper(**kwargs):
""" Return an instance of UIWrapper for testing purposes.
Parameters
----------
**kwargs
Values to use instead of the default values.
Returns
-------
wrapper: UIWrapper
"""
values = dict(
target=None,
registries=[],
)
values.update(kwargs)
return UIWrapper(**values)
def _get_wrapper(self, ui):
"""Return a new UIWrapper wrapping the given traitsui.ui.UI.
Parameters
----------
ui : traitsui.ui.UI
The UI created, e.g. by ``create_ui``.
Returns
-------
wrapper : UIWrapper
"""
return UIWrapper(
target=ui,
registries=self._registries,
delay=self.delay,
auto_process_events=self._auto_process_events,
)
def find_by_id(self, ui, id):
""" Find the TraitsUI editor with the given identifier and return a new
``UIWrapper`` object for further interactions with the editor.
Parameters
----------
ui : traitsui.ui.UI
The UI created, e.g. by ``create_ui``.
id : str
Id for finding an item in the UI.
Returns
-------
wrapper : UIWrapper
"""
return UIWrapper(
target=ui,
registries=self._registries,
delay=self.delay,
).find_by_id(id=id)
def find_by_name(self, ui, name):
""" Find the TraitsUI editor with the given name and return a new
``UIWrapper`` object for further interactions with the editor.
Parameters
----------
ui : traitsui.ui.UI
The UI created, e.g. by ``create_ui``.
name : str
A single name for retrieving a target on a UI.
Returns
-------
wrapper : UIWrapper
"""
return UIWrapper(
target=ui,
registries=self._registries,
delay=self.delay,
).find_by_name(name=name)
