X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/bb2775b9e8bc1b882d9a5c7ffe68371717b9e5dc..be05b434519887d3e8c1634120d52e052b0530c1:/wxPython/samples/wxPIA_book/Chapter-05/testEventExample.py

diff --git a/wxPython/samples/wxPIA_book/Chapter-05/testEventExample.py b/wxPython/samples/wxPIA_book/Chapter-05/testEventExample.py
new file mode 100644
index 0000000000..321a4b4af1
--- /dev/null
+++ b/wxPython/samples/wxPIA_book/Chapter-05/testEventExample.py
@@ -0,0 +1,37 @@
+import unittest
+import modelExample
+import wx
+
+class TestExample(unittest.TestCase):
+
+    def setUp(self):
+        self.app = wx.PySimpleApp()
+        self.frame = modelExample.ModelExample(parent=None, id=-1)
+
+    def tearDown(self):
+        self.frame.Destroy()
+
+    def testModel(self):
+        self.frame.OnBarney(None)
+        self.assertEqual("Barney", self.frame.model.first,
+                msg="First is wrong")
+        self.assertEqual("Rubble", self.frame.model.last)
+
+    def testEvent(self):
+        panel = self.frame.GetChildren()[0]
+        for each in panel.GetChildren():
+            if each.GetLabel() == "Wilmafy":
+                wilma = each
+                break
+        event = wx.CommandEvent(wx.wxEVT_COMMAND_BUTTON_CLICKED, wilma.GetId())
+        wilma.GetEventHandler().ProcessEvent(event)
+        self.assertEqual("Wilma", self.frame.model.first)
+        self.assertEqual("Flintstone", self.frame.model.last)
+
+def suite():
+    suite = unittest.makeSuite(TestExample, 'test')
+    return suite
+
+if __name__ == '__main__':
+    unittest.main(defaultTest='suite')
+