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))