Przeglądaj źródła

Fixes PublisherPool __exit__ test

Ask Solem 13 lat temu
rodzic
commit
944d988e1b
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      celery/tests/test_app/test_app_amqp.py

+ 2 - 2
celery/tests/test_app/test_app_amqp.py

@@ -24,10 +24,10 @@ class test_TaskPublisher(AppCase):
     def test__exit__(self):
 
         publisher = self.app.amqp.TaskPublisher(self.app.broker_connection())
-        publisher.close = Mock()
+        publisher.release = Mock()
         with publisher:
             pass
-        publisher.close.assert_called_with()
+        publisher.release.assert_called_with()
 
     def test_ensure_declare_queue(self, q="x1242112"):
         publisher = self.app.amqp.TaskPublisher(Mock())