38 lines
910 B
Python
38 lines
910 B
Python
__author__ = 'RemiZOffAlex'
|
|
__email__ = 'remizoffalex@mail.ru'
|
|
|
|
import unittest
|
|
|
|
from router.dispatcher import Dispatcher
|
|
|
|
|
|
def comparator(params):
|
|
return 'param' in params
|
|
|
|
|
|
def action(params):
|
|
return params['param']
|
|
|
|
|
|
class TestDispatcher(unittest.TestCase):
|
|
def test_dispatcher(self):
|
|
dispatcher = Dispatcher()
|
|
params ={'param': 'value'}
|
|
dispatcher.register('name', comparator, action)
|
|
method = dispatcher(params)
|
|
result = method(params)
|
|
self.assertEqual(result, 'value')
|
|
|
|
def test_sub(self):
|
|
dispatcher = Dispatcher()
|
|
params ={'param': 'value'}
|
|
|
|
sub = Dispatcher()
|
|
sub.register('name', comparator, action)
|
|
|
|
dispatcher.register('name', comparator, sub)
|
|
method = dispatcher(params)
|
|
result = method(params)
|
|
sub_result = result(params)
|
|
self.assertEqual(sub_result, 'value')
|