Python3模拟设置

时间:2016-02-04 18:40:14

标签: python unit-testing mocking

我正在尝试在设置上模拟一些第三方库,这样我就可以假装它在我的代码上按预期工作。

我能够在本地模拟它,我在其上配置了函数本身的所有返回值

    class MockConnecton:
        def __init__(self):
            self._ch = Mock()

        def channel(self):
            return self._ch

    class QEmiterTest(unittest.TestCase):
        @patch('task_queues.queue.pika.BlockingConnection')
        @patch('task_queues.queue.pika.ConnectionParameters')
        def test_emiter(self,mock_params,mock_block):
            config = {
                'host':'mq',
                'exchange':'test'
            }
            params =  {"FOO":"BAR"}
            mock_params.return_value = params
            conn = MockConnecton()
            mock_conn = Mock(wraps=conn)
            mock_block.return_value = mock_conn
            emitter = QEmitter(config['host'],config['exchange'])
            mock_params.assert_called_with(config['host'])
            mock_block.assert_called_with(params)
            mock_conn.channel.assert_called_with()
            conn._ch.exchange_declare.assert_called_with(exchange=config['exchange'],type='topic')

但是当我尝试从这种方法转向使用mock start/stop的更干净的方法时,我收到了关于断言的错误:

  

AttributeError:'_ patch'对象没有属性'assert_called_with'

我试图像这样移植它

    class QEmiterTest(unittest.TestCase):
        def setUp(self):
            mock_params = patch('task_queues.queue.pika.ConnectionParameters')
            mock_block = patch('task_queues.queue.pika.BlockingConnection')
            self.params_ret =  {"FOO":"BAR"}
            mock_params.return_value = self.params_ret
            conn = MockConnecton()
            self.mock_conn = Mock(wraps=conn)
            mock_block.return_value = self.mock_conn
            self.patch_params = mock_params
            self.patch_block = mock_block
            self.patch_params.start()
            self.patch_block.start()

        def test_emiter(self):
            config = {
                'host':'mq',
                'exchange':'test'
            }
            emitter = QEmitter(config['host'],config['exchange'])
            self.patch_params.assert_called_with(config['host'])
            self.patch_block.assert_called_with(self.params_ret)
            self.mock_conn.channel.assert_called_with()
            self.mock_conn._ch.exchange_declare.assert_called_with(exchange=config['exchange'],type='topic')

        def tearDown(self):
            self.patch_params.stop()
            self.patch_block.stop()

我可能不完全理解开始和停止,我是在设置它将应用补丁和它的参考我将能够做出断言的assumpton。我也欢迎任何关于如何使几个模拟清洁的建议

1 个答案:

答案 0 :(得分:1)

patch对象是补丁而不是模拟。 mock框架有两个主要职责:

  1. 用于录制通话并按照您的剧本播放的模拟对象
  2. 用于通过可用于传感或模拟某些行为的内容替换引用的修补方法和对象
  3. 很多时候我们可以使用补丁来安装模拟......但补丁不是模拟的。

    patch.start()返回用于修补原始引用的新引用(通常是模拟引用)。

    def setUp(self):
                self.params_ret =  {"FOO":"BAR"}
                self.mock_conn = Mock(wraps=conn)
                self.patch_params = patch('task_queues.queue.pika.ConnectionParameters', return_value = self.params_ret)
                self.patch_block = patch('task_queues.queue.pika.BlockingConnection', return_value=self.mock_conn)
                mock_params = self.patch_params.start()
                mock_block = self.patch_block.start()