Files
Phoenix/unittests/test_modalhook.py
2016-12-05 16:19:11 -06:00

35 lines
948 B
Python

import unittest
from unittests import wtc
import wx
#---------------------------------------------------------------------------
class modalhook_Tests(wtc.WidgetTestCase):
def test_modalhook1(self):
class MyModalDialogHook(wx.ModalDialogHook):
def __init__(self):
wx.ModalDialogHook.__init__(self)
self.counter = 0
def Enter(self, dialog):
self.counter += 1
return wx.ID_OK
def Exit(self, dialog):
self.counter += 1 # not called because Enter didn't return wx.ID_NONE
myHook = MyModalDialogHook()
myHook.Register()
wx.MessageBox("This should be auto-dismissed...", style=wx.OK|wx.CANCEL)
self.assertEqual(myHook.counter, 1)
myHook.Unregister()
#---------------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()