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