Browse Source

Fixes build

Ask Solem 9 years ago
parent
commit
b00e3b802d

+ 1 - 6
celery/tests/backends/test_mongodb.py

@@ -10,7 +10,7 @@ from celery import uuid
 from celery import states
 from celery import states
 from celery.backends import mongodb as module
 from celery.backends import mongodb as module
 from celery.backends.mongodb import (
 from celery.backends.mongodb import (
-    InvalidDocument, MongoBackend, Bunch, pymongo,
+    InvalidDocument, MongoBackend, pymongo,
 )
 )
 from celery.exceptions import ImproperlyConfigured
 from celery.exceptions import ImproperlyConfigured
 from celery.tests.case import (
 from celery.tests.case import (
@@ -49,11 +49,6 @@ class test_MongoBackend(AppCase):
         module.Binary = self._reset['Binary']
         module.Binary = self._reset['Binary']
         datetime.datetime = self._reset['datetime']
         datetime.datetime = self._reset['datetime']
 
 
-    def test_Bunch(self):
-        x = Bunch(foo='foo', bar=2)
-        self.assertEqual(x.foo, 'foo')
-        self.assertEqual(x.bar, 2)
-
     def test_init_no_mongodb(self):
     def test_init_no_mongodb(self):
         prev, module.pymongo = module.pymongo, None
         prev, module.pymongo = module.pymongo, None
         try:
         try:

+ 14 - 0
celery/tests/utils/test_objects.py

@@ -0,0 +1,14 @@
+from __future__ import absolute_import, unicode_literals
+
+from celery.utils.objects import Bunch
+
+from celery.tests.case import Case
+
+
+class test_Bunch(Case):
+
+    def test(self):
+        x = Bunch(foo='foo', bar=2)
+        self.assertEqual(x.foo, 'foo')
+        self.assertEqual(x.bar, 2)
+

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

@@ -339,7 +339,7 @@ class test_Gossip(AppCase):
 
 
     def test_callbacks(self):
     def test_callbacks(self):
         c = self.Consumer()
         c = self.Consumer()
-        c.app.connection = _amqp_connection()
+        c.app.connection_for_read = _amqp_connection()
         g = Gossip(c)
         g = Gossip(c)
         on_node_join = Mock(name='on_node_join')
         on_node_join = Mock(name='on_node_join')
         on_node_join2 = Mock(name='on_node_join2')
         on_node_join2 = Mock(name='on_node_join2')