|
@@ -7,6 +7,7 @@ from celery import task
|
|
from celery import registry
|
|
from celery import registry
|
|
from celery.log import setup_logger
|
|
from celery.log import setup_logger
|
|
from celery import messaging
|
|
from celery import messaging
|
|
|
|
+from celery.backends import default_backend
|
|
|
|
|
|
|
|
|
|
# Task run functions can't be closures/lambdas, as they're pickled.
|
|
# Task run functions can't be closures/lambdas, as they're pickled.
|
|
@@ -93,7 +94,7 @@ class TestCeleryTasks(unittest.TestCase):
|
|
|
|
|
|
self.assertFalse(task.is_done(presult.task_id))
|
|
self.assertFalse(task.is_done(presult.task_id))
|
|
self.assertFalse(presult.is_done())
|
|
self.assertFalse(presult.is_done())
|
|
- task.mark_as_done(presult.task_id, result=None)
|
|
|
|
|
|
+ default_backend.mark_as_done(presult.task_id, result=None)
|
|
self.assertTrue(task.is_done(presult.task_id))
|
|
self.assertTrue(task.is_done(presult.task_id))
|
|
self.assertTrue(presult.is_done())
|
|
self.assertTrue(presult.is_done())
|
|
|
|
|