如何使用unittest.mock在单元测试中模拟任意ConfigParser调用

Question

我正在尝试使用unittest.mock的action / assert模式而不是mox的记录/重放/验证模式。

# foo.py
def op_1(param):
    pass

def op_2(param):
    pass

def do_stuff(param_1, param_2):
    global config
    global log
    try:
        op_1(param_1)
        if config.getboolean('section','option'):
            op_2(param_2)
    except:
         log.error("an error occured")

而且，这是我的unittest文件的示例。

# test_foo.py
class TestFoo(unittest.TestCase):
    def test_do_stuff(self):
        param_1 = None
        param_2 = None
        foo.config = MagicMock()
        foo.config.getboolean('section','option', return_value = True)
        foo.op_1 = MagicMock()
        foo.op_2 = MagicMock()
        do_stuff(param_1, param_2)
        foo.op_1.assert_called_once_with(param_1)
        foo.op_2.assert_called_once_with(param_2)
        foo.config.getboolean.assert_called_once_with('section','option')

此测试是否验证以下项目/我使用模拟权利吗？

1. do_stuff调用返回没有错误
使用param_1 调用
2. op_1 使用param_2
调用
3. op_2
4. 已使用配置解析器对象，但具体调用无关紧要

Answer 1

事实证明我使用了return_value错误。

当我需要mock.Mock或mock.MagicMock对象返回值时，无论传递的参数如何，都需要始终返回该值。但是，基于传递的参数（可能的特征请求）给出不同的行为可能会很好。

我完成这个的方式是：

foo.config.getboolean = mock.MagicMock(return_value = True)

然后我可以这样做：

self.assertGreaterThan(len(foo.config.mock_calls), 0)
self.assertGreaterThan(len(foo.config.getboolean(str(),str())), 0)