소스 검색

Tests passes again

Ask Solem 14 년 전
부모
커밋
dc07f346ae
2개의 변경된 파일9개의 추가작업 그리고 7개의 파일을 삭제
  1. 8 6
      celery/tests/test_bin/test_celeryd.py
  2. 1 1
      celery/tests/test_utils/test_utils_info.py

+ 8 - 6
celery/tests/test_bin/test_celeryd.py

@@ -126,16 +126,18 @@ class test_Worker(AppCase):
     def test_init_queues(self):
         app = current_app
         c = app.conf
-        p, app.amqp.queues = app.amqp.queues, {
+        p, app.amqp.queues = app.amqp.queues, app.amqp.Queues({
                 "celery": {"exchange": "celery",
                            "binding_key": "celery"},
                 "video": {"exchange": "video",
-                           "binding_key": "video"}}
+                           "binding_key": "video"}})
         try:
             worker = self.Worker(queues=["video"])
             worker.init_queues()
-            self.assertIn("video", worker.queues)
-            self.assertNotIn("celery", worker.queues)
+            self.assertIn("video", app.amqp.queues)
+            self.assertIn("video", app.amqp.queues.consume_from)
+            self.assertIn("celery", app.amqp.queues)
+            self.assertNotIn("celery", app.amqp.queues.consume_from)
 
             c.CELERY_CREATE_MISSING_QUEUES = False
             self.assertRaises(ImproperlyConfigured,
@@ -143,12 +145,12 @@ class test_Worker(AppCase):
             c.CELERY_CREATE_MISSING_QUEUES = True
             worker = self.Worker(queues=["image"])
             worker.init_queues()
-            self.assertIn("image", worker.queues)
+            self.assertIn("image", app.amqp.queues.consume_from)
             self.assertDictContainsSubset({"exchange": "image",
                                            "routing_key": "image",
                                            "binding_key": "image",
                                            "exchange_type": "direct"},
-                                            worker.queues["image"])
+                                            app.amqp.queues["image"])
         finally:
             app.amqp.queues = p
 

+ 1 - 1
celery/tests/test_utils/test_utils_info.py

@@ -57,6 +57,6 @@ class TestInfo(unittest.TestCase):
 
     def test_format_queues(self):
         celery = Celery(set_as_current=False)
-        celery.amqp.queues = QUEUES
+        celery.amqp.queues = celery.amqp.Queues(QUEUES)
         self.assertItemsEqual(celery.amqp.queues.format().split("\n"),
                               [QUEUE_FORMAT1, QUEUE_FORMAT2])