Browse Source

More tests passing

Ask Solem 11 years ago
parent
commit
c56e13ee65
1 changed files with 2 additions and 2 deletions
  1. 2 2
      celery/tests/tasks/test_tasks.py

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

@@ -245,7 +245,7 @@ class test_tasks(AppCase):
 
     def assertNextTaskDataEqual(self, consumer, presult, task_name,
                                 test_eta=False, test_expires=False, **kwargs):
-        next_task = consumer.queues[0].get()
+        next_task = consumer.queues[0].get(accept=['pickle'])
         task_data = next_task.decode()
         self.assertEqual(task_data['id'], presult.id)
         self.assertEqual(task_data['task'], task_name)
@@ -501,7 +501,7 @@ class test_TaskSet(AppCase):
         taskset_id = taskset_res.taskset_id
         consumer = increment_counter.get_consumer()
         for subtask in subtasks:
-            m = consumer.queues[0].get().payload
+            m = consumer.queues[0].get(accept=['pickle']).payload
             self.assertDictContainsSubset({'taskset': taskset_id,
                                            'task': increment_counter.name,
                                            'id': subtask.id}, m)