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