def test_registry(self): self.assertTrue('queuecmd_test_task_one' in registry) self.assertTrue('queuecmd_test_task_two' in registry) self.assertFalse('MyTaskClass' in registry) registry.register(MyTaskClass) self.assertTrue('MyTaskClass' in registry) self.assertFalse('another' in registry)
def test_registry(self): self.assertTrue('queue_task_test_task_one' in registry) self.assertTrue('queue_task_test_task_two' in registry) self.assertFalse('MyTaskClass' in registry) registry.register(MyTaskClass) self.assertTrue('MyTaskClass' in registry) self.assertFalse('another' in registry)
def test_registry(self): self.assertTrue('huey.tests.test_registry.test_task_one' in registry) self.assertTrue('huey.tests.test_registry.test_task_two' in registry) self.assertTrue('huey.tests.test_registry.three' in registry) self.assertFalse('huey.tests.test_registry.MyTaskClass' in registry) registry.register(MyTaskClass) self.assertTrue('huey.tests.test_registry.MyTaskClass' in registry) self.assertFalse('huey.tests.test_registry.another' in registry) self.assertFalse( 'huey.tests.test_registry.test_task_three' in registry)
def __init__(cls, name, bases, attrs): """ Metaclass to ensure that all command classes are registered """ registry.register(cls)
@huey.task(include_task=True) def put_data_ctx(key, value, task=None): state['last_task_class'] = type(task).__name__ @huey_results.task(include_task=True) def error_testing_task_with_ctx(key, value, task=None): bad = 1/0 state['last_task_class'] = type(task).__name__ class PutTask(QueueTask): def execute(self): k, v = self.data state[k] = v registry.register(PutTask) class TestException(Exception): pass def _throw_error_task(message=None): raise TestException(message or 'bampf') throw_error_task = huey.task()(_throw_error_task) throw_error_task_res = huey_results.task()(_throw_error_task) @huey_results.task() def add_values(a, b): return a + b @huey_results.task()