source: SHX/trunk/src/SeismicHandler/tests/test_events.py @ 153

# -*- coding: utf-8 -*-

from SeismicHandler.core.modules.Messages import Message, MessageService
import unittest

class eventsTestCase(unittest.TestCase):
    def setUp(self):
        self.ret = []
        
        self.test = Message("test")
        self.test2 = Message("test2")
        
        MessageService.subscribe([self.test, self.test2], self.returnValue)

    def returnValue(self, *c):
        """ helper function for event trigger """
        self.ret.append(c[1])

    def testSubscribe(self):
        self.assertTrue(self.test in MessageService.events)
        self.assertTrue(self.test2 in MessageService.events)

    def testTrigger(self):
        MessageService.trigger(self.test, "one")
        MessageService.trigger(self.test2, "two")

        self.assertEqual(self.ret, ["one", "two"])

    def testUnsubscribe(self):
        MessageService.unsubscribe(self.returnValue, self.test)
        self.assertFalse(self.returnValue in MessageService.events[self.test])

        MessageService.trigger(self.test, "one2")
        MessageService.trigger(self.test2, "two2")

        self.assertEqual(self.ret, ["two2",])

    def tearDown(self):
        MessageService.unsubscribe(self.returnValue)

def suite():
    return unittest.makeSuite(eventsTestCase, 'test')

if __name__ == "__main__":
    unittest.main(defaultTest='suite')