Ask Solem 11 years ago
parent
commit
174bc56826
1 changed files with 3 additions and 2 deletions
  1. 3 2
      celery/tests/app/test_app.py

+ 3 - 2
celery/tests/app/test_app.py

@@ -174,12 +174,13 @@ class test_App(AppCase):
 
     def test_maybe_close_pool(self):
         cpool = self.app._pool = Mock()
-        ppool = self.app.amqp._producer_pool = Mock()
+        amqp = self.app.__dict__['amqp'] = Mock()
+        ppool = amqp._producer_pool
         self.app._maybe_close_pool()
         cpool.force_close_all.assert_called_with()
         ppool.force_close_all.assert_called_with()
         self.assertIsNone(self.app._pool)
-        self.assertIsNone(self.app.amqp._producer_pool)
+        self.assertIsNone(self.app.__dict__['amqp']._producer_pool)
 
         self.app._pool = Mock()
         self.app._maybe_close_pool()