You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.9 KiB
82 lines
2.9 KiB
import unittest
|
|
import mock
|
|
|
|
from os.path import dirname, realpath
|
|
from sys import argv, path
|
|
path.append(dirname(dirname(realpath(__file__))) + '/lib')
|
|
|
|
from Event.EventHandler import EventHandler
|
|
|
|
class HandlerOne(EventHandler):
|
|
def __init__(self):
|
|
super(HandlerOne, self).__init__()
|
|
|
|
self._event_methods = {
|
|
1 : self._handleOne,
|
|
2 : self._handleTwo }
|
|
|
|
def _handleOne(self, event):
|
|
return 'one'
|
|
|
|
def _handleTwo(self, event):
|
|
return 'two'
|
|
|
|
class TestEventHandler(unittest.TestCase):
|
|
def setUp(self):
|
|
self._handler = EventHandler()
|
|
self._handler_one = HandlerOne()
|
|
|
|
self._event_mock1 = mock.Mock()
|
|
self._event_mock2 = mock.Mock()
|
|
self._event_source_mock = mock.Mock()
|
|
self._dispatcher_mock1 = mock.Mock()
|
|
self._dispatcher_mock2 = mock.Mock()
|
|
|
|
self._event_mock1.name = 'a'
|
|
self._event_mock1.type = 1
|
|
self._event_mock1.subject = self._event_source_mock
|
|
self._event_mock1.data = None
|
|
|
|
self._event_mock2.name = 'b'
|
|
self._event_mock2.type = 2
|
|
self._event_mock2.subject = self._event_source_mock
|
|
self._event_mock2.data = 'arbitrary data'
|
|
|
|
self._event_source_mock.emit.return_value = self._event_mock1
|
|
|
|
def testEmptyHandlerSetDispatcher(self):
|
|
self._handler.setDispatcher(self._dispatcher_mock1)
|
|
self._handler.setDispatcher(self._dispatcher_mock2)
|
|
self.assertIn(self._dispatcher_mock1, self._handler._dispatcher)
|
|
self.assertIn(self._dispatcher_mock2, self._handler._dispatcher)
|
|
|
|
def testEmptyHandlerGetHandledIds(self):
|
|
self.assertEqual(self._handler.getHandledIds(), [])
|
|
|
|
def testEmptyHandlerNoDispatcherIssueEvent(self):
|
|
self._handler.issueEvent(self._event_source_mock, 'a', None)
|
|
self._event_source_mock.emit.assert_called_once_with('a', None)
|
|
|
|
def testEmptyHandlerIssueEvent(self):
|
|
self._handler.setDispatcher(self._dispatcher_mock1)
|
|
self._handler.setDispatcher(self._dispatcher_mock2)
|
|
self._handler.issueEvent(self._event_source_mock, 'a', None)
|
|
self._event_source_mock.emit.assert_called_once_with('a', None)
|
|
self._dispatcher_mock1.queueEvent.called_once_with(self._event_mock1)
|
|
self._dispatcher_mock2.queueEvent.called_once_with(self._event_mock1)
|
|
|
|
def testEmptyHandlerHandleEvent(self):
|
|
self.assertFalse(self._handler.handleEvent(self._event_mock1))
|
|
self.assertFalse(self._handler.handleEvent(self._event_mock2))
|
|
|
|
def testHandlerOneHandleEvent(self):
|
|
self.assertEqual(self._handler_one.handleEvent(self._event_mock1), 'one')
|
|
self.assertEqual(self._handler_one.handleEvent(self._event_mock2), 'two')
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromTestCase(TestEventHandler)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|
|
|
|
# vim: set ft=python et ts=8 sw=4 sts=4:
|