Added the sample code from wxPython In Action to the samples dir

[wxWidgets.git] / 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 (file)
index 0000000..321a4b4
--- /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')
+