|
@@ -6,6 +6,8 @@ from mock import Mock, patch
|
|
|
|
|
|
from celery import current_app
|
|
from celery import current_app
|
|
from celery import states
|
|
from celery import states
|
|
|
|
+from celery.registry import tasks
|
|
|
|
+from celery.task import subtask
|
|
from celery.utils import cached_property, uuid
|
|
from celery.utils import cached_property, uuid
|
|
from celery.utils.timeutils import timedelta_seconds
|
|
from celery.utils.timeutils import timedelta_seconds
|
|
|
|
|
|
@@ -91,7 +93,7 @@ class test_RedisBackend(unittest.TestCase):
|
|
self.assertEqual(b.expires, 60)
|
|
self.assertEqual(b.expires, 60)
|
|
|
|
|
|
def test_on_chord_apply(self):
|
|
def test_on_chord_apply(self):
|
|
- self.Backend().on_chord_apply()
|
|
|
|
|
|
+ self.Backend().on_chord_apply("setid")
|
|
|
|
|
|
def test_mget(self):
|
|
def test_mget(self):
|
|
b = self.MockBackend()
|
|
b = self.MockBackend()
|
|
@@ -105,8 +107,6 @@ class test_RedisBackend(unittest.TestCase):
|
|
|
|
|
|
@patch("celery.result.TaskSetResult")
|
|
@patch("celery.result.TaskSetResult")
|
|
def test_on_chord_part_return(self, setresult):
|
|
def test_on_chord_part_return(self, setresult):
|
|
- from celery.registry import tasks
|
|
|
|
- from celery.task import subtask
|
|
|
|
b = self.MockBackend()
|
|
b = self.MockBackend()
|
|
deps = Mock()
|
|
deps = Mock()
|
|
deps.total = 10
|
|
deps.total = 10
|
|
@@ -117,6 +117,7 @@ class test_RedisBackend(unittest.TestCase):
|
|
try:
|
|
try:
|
|
tasks["foobarbaz"] = task
|
|
tasks["foobarbaz"] = task
|
|
task.request.chord = subtask(task)
|
|
task.request.chord = subtask(task)
|
|
|
|
+ task.request.taskset = "setid"
|
|
|
|
|
|
b.on_chord_part_return(task)
|
|
b.on_chord_part_return(task)
|
|
self.assertTrue(b.client.incr.call_count)
|
|
self.assertTrue(b.client.incr.call_count)
|