diff --git a/tests/test_dispatcher.py b/tests/test_dispatcher.py index 75b3f64..b82ea70 100644 --- a/tests/test_dispatcher.py +++ b/tests/test_dispatcher.py @@ -6,18 +6,25 @@ import unittest from router.dispatcher import Dispatcher -def comparator(params): - return 'param' in params +def comparator1(params): + return 'param1' in params +def comparator2(params): + return params['param2']=='value2' + def action(params): - return params['param'] + return params['param3'] class TestDispatcher(unittest.TestCase): def test_dispatcher(self): dispatcher = Dispatcher() - params ={'param': 'value'} + params = { + 'param1': 'value1', + 'param2': 'value2', + 'param3': 'value3' + } dispatcher.register('name', comparator, action) method = dispatcher(params) result = method(params) @@ -34,4 +41,4 @@ class TestDispatcher(unittest.TestCase): method = dispatcher(params) result = method(params) sub_result = result(params) - self.assertEqual(sub_result, 'value') + self.assertEqual(sub_result, 'value3')