def test_getattr(self):
set = _TaskSet('testparams')
set._set('func1', 'one', 'testexchange', 'testrouting')
set._set('func2', 'two', 'testexchange', 'testrouting')
self.assertEqual('func1', set.one.func)
self.assertEqual('func2', set.two.func)
self.assertRaises(AttributeError, getattr, set, 'three')
def test_set(self):
set = _TaskSet('testapp')
set._set('testfunc', 'testname', 'testexchange', 'testrouting')
self.assertEqual(1, len(set))
self.assertTrue('testname' in set)
self.assertEqual('testfunc', set._tasks['testname'].func)
self.assertEqual('testname', set._tasks['testname'].name)
self.assertEqual('testapp', set._tasks['testname'].app)
self.assertEqual('testrouting', set._tasks['testname'].routing_key)
self.assertEqual('testexchange', set._tasks['testname'].exchange)
def test_repr(self):
set = _TaskSet('testparams')
set._set('func1', 'one', 'testexchange', 'testrouting')
set._set('func2', 'two', 'testexchange', 'testrouting')
self.assertTrue(repr(set).startswith('<registered task set '))
self.assertTrue(repr(set).endswith('>'))
def test_declare(self):
set = _TaskSet('testapp')
set._set('testfunc1', 'testtask1', 'testexchange1', 'testrouting1')
set._declare('testtask1', 'testexchange1', 'testrouting1')
set._declare('testtask2', 'testexchange2', 'testrouting2')
self.assertEqual(2, len(set))
