Browse Source

Attempt to fix py3 build

Ask Solem 9 years ago
parent
commit
e3a8edbcf8
1 changed files with 6 additions and 2 deletions
  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: