|
@@ -335,6 +335,7 @@ class TestCeleryTasks(unittest.TestCase):
|
|
request.foo = 32
|
|
request.foo = 32
|
|
self.assertEqual(request.get("foo"), 32)
|
|
self.assertEqual(request.get("foo"), 32)
|
|
self.assertEqual(request.get("bar", 36), 36)
|
|
self.assertEqual(request.get("bar", 36), 36)
|
|
|
|
+ request.clear()
|
|
|
|
|
|
def test_task_class_repr(self):
|
|
def test_task_class_repr(self):
|
|
task = self.createTaskCls("T1", "c.unittest.t.repr")
|
|
task = self.createTaskCls("T1", "c.unittest.t.repr")
|
|
@@ -343,9 +344,10 @@ class TestCeleryTasks(unittest.TestCase):
|
|
def test_after_return(self):
|
|
def test_after_return(self):
|
|
task = self.createTaskCls("T1", "c.unittest.t.after_return")()
|
|
task = self.createTaskCls("T1", "c.unittest.t.after_return")()
|
|
task.backend = Mock()
|
|
task.backend = Mock()
|
|
- task.request.chord = 123
|
|
|
|
|
|
+ task.request.chord = {}
|
|
task.after_return("SUCCESS", 1.0, "foobar", (), {}, None)
|
|
task.after_return("SUCCESS", 1.0, "foobar", (), {}, None)
|
|
task.backend.on_chord_part_return.assert_called_with(task)
|
|
task.backend.on_chord_part_return.assert_called_with(task)
|
|
|
|
+ task.request.clear()
|
|
|
|
|
|
def test_send_task_sent_event(self):
|
|
def test_send_task_sent_event(self):
|
|
T1 = self.createTaskCls("T1", "c.unittest.t.t1")
|
|
T1 = self.createTaskCls("T1", "c.unittest.t.t1")
|