Browse Source

Fixes build

Ask Solem 9 years ago
parent
commit
ebbfa84cc1
2 changed files with 3 additions and 3 deletions
  1. 2 2
      celery/tests/tasks/test_trace.py
  2. 1 1
      celery/tests/worker/test_loops.py

+ 2 - 2
celery/tests/tasks/test_trace.py

@@ -274,7 +274,7 @@ class test_trace(TraceCase):
 
     def test_trace_task_ret__no_content_type(self):
         _trace_task_ret(
-            self.add.name, 'id1', {}, ((2, 2), {}), None, None,
+            self.add.name, 'id1', {}, ((2, 2), {}, {}), None, None,
             app=self.app,
         )
 
@@ -283,7 +283,7 @@ class test_trace(TraceCase):
             self.add.name, self.add, app=self.app,
         )
         _fast_trace_task(
-            self.add.name, 'id1', {}, ((2, 2), {}), None, None,
+            self.add.name, 'id1', {}, ((2, 2), {}, {}), None, None,
             app=self.app, _loc=[self.app.tasks, {}, 'hostname']
         )
 

+ 1 - 1
celery/tests/worker/test_loops.py

@@ -177,7 +177,7 @@ class test_asynloop(AppCase):
         x, on_task, msg, strategy = self.task_context(self.add.s(2, 2))
         msg.headers.pop('task')
         on_task(msg)
-        x.on_unknown_message.assert_called_with(msg.payload, msg)
+        x.on_unknown_message.assert_called_with(msg.decode(), msg)
 
     def test_on_task_not_registered(self):
         x, on_task, msg, strategy = self.task_context(self.add.s(2, 2))