Cleanups so hooke/ui/gui/ passes unittests.
[hooke.git] / hooke / ui / gui / handler / selection.py
index ee0e5e0d10f43c0a5aec8e5a4854a4cb317147cd..d8330b5000fa7522c0e6e6de364a556e190e58e9 100644 (file)
 
 import wx
 
-from ..dialog.selection import SelectionDialog
+from ..dialog.selection import Selection
 from . import Handler
 
 
 class SelectionHandler (Handler):
     def __init__(self):
-        super(StringHandler, self).__init__(name='selection')
+        super(SelectionHandler, self).__init__(name='selection')
 
     def run(self, hooke_frame, msg):
         self._canceled = True
         while self._canceled:
-            s = SelectionDialog(
+            s = Selection(
                 options=msg.options,
                 message=msg.msg,
                 button_id=wxID_OK,
@@ -44,8 +44,7 @@ class SelectionHandler (Handler):
                 selection_style='single',
                 parent=self,
                 label='Selection handler',
-                style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER),
-            )
+                style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
         return self._selected
 
     def _selection(self, _class, method, options, selected):