浏览代码

Fixes PublisherPool __exit__ test

Ask Solem 13 年之前
父节点
当前提交
944d988e1b
共有 1 个文件被更改,包括 2 次插入2 次删除
  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())