bumblebee-status/tests/core/test_event.py

import unittest

import core.event

class event(unittest.TestCase):
    def setUp(self):
        self.someEvent = 'event'
        self.called = {}
        self.params = []
        core.event.clear()

    def callback1(self):
        self.called['callback1'] = True

    def callback2(self):
        self.called['callback2'] = True

    def callback_args(self, val1, val2):
        self.called['callback_args'] = True
        self.params = [val1, val2]

    def callback_kwargs(self, val1, val2, key1=None, key2=None):
        self.called['callback_kwargs'] = True
        self.params = [val1, val2, key1, key2]

    def test_simple_callback(self):
        core.event.register(self.someEvent, self.callback1)
        core.event.register(self.someEvent, self.callback2)

        core.event.trigger(self.someEvent)

        self.assertEqual(2, len(self.called.keys()))

    def test_arg_callback(self):
        core.event.register(self.someEvent, self.callback_args, 'a', 'b')
        core.event.trigger(self.someEvent)
        self.assertEqual(1, len(self.called.keys()))
        self.assertEqual(['a', 'b'], self.params)

    def test_kwargs_callback(self):
        core.event.register(self.someEvent, self.callback_kwargs, 'a', 'b', key1='test', key2='x')
        core.event.trigger(self.someEvent)
        self.assertEqual(1, len(self.called.keys()))
        self.assertEqual(['a', 'b', 'test', 'x'], self.params)

    def test_arg_trigger(self):
        core.event.register(self.someEvent, self.callback_args)
        core.event.trigger(self.someEvent, 'a', 'b')
        self.assertEqual(1, len(self.called.keys()))
        self.assertEqual(['a', 'b'], self.params)

    def test_kwargs_trigger(self):
        core.event.register(self.someEvent, self.callback_kwargs)
        core.event.trigger(self.someEvent, 'a', 'b', key1='test', key2='x')
        self.assertEqual(1, len(self.called.keys()))
        self.assertEqual(['a', 'b', 'test', 'x'], self.params)

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4