Ask Solem 9 лет назад
Родитель
Сommit
e3a8edbcf8
1 измененных файлов с 6 добавлено и 2 удалено
  1. 6 2
      celery/tests/tasks/test_tasks.py

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

@@ -428,8 +428,12 @@ class test_tasks(TasksCase):
         self.mytask.request.callbacks = 'callbacks'
         self.mytask.request.errbacks = 'errbacks'
 
-        mocked_signature = MagicMock(name='s')
-        accumulate_mock = MagicMock(name='accumulate', s=mocked_signature)
+        class JsonMagicMock(MagicMock):
+            def __json__(self):
+                return 'whatever'
+
+        mocked_signature = JsonMagicMock(name='s')
+        accumulate_mock = JsonMagicMock(name='accumulate', s=mocked_signature)
         self.mytask.app.tasks['celery.accumulate'] = accumulate_mock
 
         try: