[wxWidgets.git] / wxPython / samples / wxPIA_book / Chapter-05 / testEventExample.py

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')
Commit	Line	Data
be05b434 RD	1	import unittest
	2	import modelExample
	3	import wx
	4
	5	class TestExample(unittest.TestCase):
	6
	7	def setUp(self):
	8	self.app = wx.PySimpleApp()
	9	self.frame = modelExample.ModelExample(parent=None, id=-1)
	10
	11	def tearDown(self):
	12	self.frame.Destroy()
	13
	14	def testModel(self):
	15	self.frame.OnBarney(None)
	16	self.assertEqual("Barney", self.frame.model.first,
	17	msg="First is wrong")
	18	self.assertEqual("Rubble", self.frame.model.last)
	19
	20	def testEvent(self):
	21	panel = self.frame.GetChildren()[0]
	22	for each in panel.GetChildren():
	23	if each.GetLabel() == "Wilmafy":
	24	wilma = each
	25	break
	26	event = wx.CommandEvent(wx.wxEVT_COMMAND_BUTTON_CLICKED, wilma.GetId())
	27	wilma.GetEventHandler().ProcessEvent(event)
	28	self.assertEqual("Wilma", self.frame.model.first)
	29	self.assertEqual("Flintstone", self.frame.model.last)
	30
	31	def suite():
	32	suite = unittest.makeSuite(TestExample, 'test')
	33	return suite
	34
	35	if __name__ == '__main__':
	36	unittest.main(defaultTest='suite')
	37