def test_return_value_class_method_resolution_order(self): given_task_class = Task test_cls = celery_retry.autoretry_task_factory(given_task_class) self.assertListEqual(test_cls.mro(), [ test_cls, celery_retry.AutoRetryTaskMixin, given_task_class, object ])
def setUp(self): super(AutoRetryTaskMixinTestCase, self).setUp() self.test_class = celery_retry.autoretry_task_factory(Task)
def test_return_value_is_subclass_of_given_task_class(self): given_task_class = Task test_cls = celery_retry.autoretry_task_factory(given_task_class) self.assertTrue(issubclass(test_cls, given_task_class))
def test_return_value_type_is_type(self): test_cls = celery_retry.autoretry_task_factory(SimpleTask) self.assertIs(type(test_cls), type)
def test_return_value_is_class(self): test_cls = celery_retry.autoretry_task_factory(Task) self.assertTrue(inspect.isclass(test_cls))
class ExceptionTask(celery_retry.autoretry_task_factory(Task)): def run(self): raise test_exception
class SuccessfulTask(celery_retry.autoretry_task_factory(Task)): def run(self): return True