Browse Source

Fixes build

Ask Solem 8 years ago
parent
commit
f7bd26aba6

+ 10 - 5
celery/tests/bin/test_beat.py

@@ -11,11 +11,16 @@ from celery.apps import beat as beatapp
 from celery.tests.case import AppCase, Mock, mock, patch
 
 
-class MockBeat(beatapp.Beat):
-    Service = Mock(
-        name='MockBeat.Service',
-        return_value=Mock(name='MockBeat()', max_interval=3.3),
-    )
+def MockBeat(*args, **kwargs):
+    class _Beat(beatapp.Beat):
+        Service = Mock(
+            name='MockBeat.Service',
+            return_value=Mock(name='MockBeat()', max_interval=3.3),
+        )
+    b = _Beat(*args, **kwargs)
+    sched = b.Service.return_value.get_scheduler = Mock()
+    sched.return_value.max_interval = 3.3
+    return b
 
 
 class test_Beat(AppCase):

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

@@ -616,7 +616,7 @@ class test_Consumer(AppCase):
         self.assertTrue(connections[0].closed)
 
     @patch('kombu.connection.Connection._establish_connection')
-    @patch('kombu.utils.sleep')
+    @patch('kombu.utils.functional.sleep')
     def test_connect_errback(self, sleep, connect):
         c = self.NoopConsumer()
         Transport.connection_errors = (ChannelError,)

+ 1 - 0
requirements/extras/mongodb.txt

@@ -0,0 +1 @@
+pymongo>=3.3.0