Python 内部实体模拟

Question

我想测试一个方法，是否调用临时内部对象的特定方法。 (ConfigParser.read)

所以该对象是在内部创建的，方法退出后无法从外部访问。

使用 python 2.7

在foobar.py

 import ConfigParser

 class FooBar:
     def method(self, filename):
         config=ConfigParser.ConfigParser()
         config.read(filename)
         do_some_stuff()

我想测试一下是否调用了config.read。

据我所知，补丁装饰器是为此制作的，但不幸的是，测试用例接收到的 MagicMock 对象与内部创建的对象不同，我无法靠近方法内部的对象。

我这样试过：

class TestFooBar(TestCase):

    def setUp(self):
         self.myfoobar = FooBar()

    @mock.patch('foobar.ConfigParser')
    def test_read(self,mock_foobar):
        self.myfoobar.method("configuration.ini")
        assert mock_foobar.called # THIS IS OKAY
        assert mock_foobar.read.called # THIS FAILS
        mock_foobar.read.assert_called_with("configuration.ini") # FAILS TOO

问题是： - mock_foobar 在 self.myfoobar.method 内部创建 ConfigReader 之前创建。 - 调试时 mock_foobar 有关于先前调用的内部数据，但没有 "read" 属性（用于模拟读取方法的内部 MagicMock）

当然，一种解决方法是重构并为 .read() 或 init() 提供一个 ConfigReader 对象，但并不总是可以更改代码，而且我希望在不接触被测模块的情况下掌握方法的内部对象。

Answer 1

你太接近了！问题是您正在模拟 class，但是随后您的测试检查是否在该模拟 class 上调用了 read() - 但您实际上希望在返回的实例上调用 read()当您调用 class 时。以下作品 - 我发现第二个测试比第一个更具可读性，但它们都有效：

import ConfigParser
from unittest import TestCase

from mock import create_autospec, patch, Mock 


class FooBar(object):
    def method(self, filename):
        config=ConfigParser.ConfigParser()
        config.read(filename)


class TestFooBar(TestCase):

    def setUp(self):
         self.myfoobar = FooBar()

    @patch('ConfigParser.ConfigParser')
    def test_method(self, config_parser_class_mock):
        config_parser_mock = config_parser_class_mock.return_value

        self.myfoobar.method("configuration.ini")

        config_parser_class_mock.assert_called_once_with()
        config_parser_mock.read.assert_called_once_with("configuration.ini")

    def test_method_better(self):
        config_parser_mock = create_autospec(ConfigParser.ConfigParser, instance=True)
        config_parser_class_mock = Mock(return_value=config_parser_mock)

        with patch('ConfigParser.ConfigParser', config_parser_class_mock):
            self.myfoobar.method("configuration.ini")

        config_parser_class_mock.assert_called_once_with()
        config_parser_mock.read.assert_called_once_with("configuration.ini")

Python 内部实体模拟

Python internal entity mocking

python

unit-testing

mocking

patch

magicmock