检查多个模拟的呼叫顺序

Question

我要尝试测试其呼叫顺序的三个功能。

假设在模块module.py中，我有以下内容

# module.py

def a(*args):
    # do the first thing

def b(*args):
    # do a second thing

def c(*args):
    # do a third thing


def main_routine():
    a_args = ('a')
    b_args = ('b')
    c_args = ('c')

    a(*a_args)
    b(*b_args)
    c(*c_args)

我想检查b在a之后和c之前被调用。因此，对a，b和c中的每一个进行模拟都是容易的：

# tests.py

@mock.patch('module.a')
@mock.patch('module.b')
@mock.patch('module.c')
def test_main_routine(c_mock, b_mock, a_mock):
    # test all the things here

检查每个单独的模拟都被调用也很容易。如何检查呼叫相对于彼此的顺序？

call_args_list 将无法使用，因为每个模拟都单独维护。

我尝试使用副作用来记录每个调用：

calls = []
def register_call(*args):
    calls.append(mock.call(*args))
    return mock.DEFAULT

a_mock.side_effect = register_call
b_mock.side_effect = register_call
c_mock.side_effect = register_call

但这仅给我提供了与模拟一起调用的args，而没有给实际调用的模拟。我可以添加更多逻辑：

# tests.py
from functools import partial

def register_call(*args, **kwargs):
    calls.append(kwargs.pop('caller', None), mock.call(*args, **kwargs))
    return mock.DEFAULT

a_mock.side_effect = partial(register_call, caller='a')
b_mock.side_effect = partial(register_call, caller='b')
c_mock.side_effect = partial(register_call, caller='c')

似乎可以完成工作了……是否有更好的方法？感觉API中应该已经有某些东西可以做到这一点，而我所缺少的。

Answer 1

定义一个Mock管理器，并通过附加模拟attach_mock()。然后检查mock_calls：

@patch('module.a')
@patch('module.b')
@patch('module.c')
def test_main_routine(c, b, a):
    manager = Mock()
    manager.attach_mock(a, 'a')
    manager.attach_mock(b, 'b')
    manager.attach_mock(c, 'c')

    module.main_routine()

    expected_calls = [call.a('a'), call.b('b'), call.c('c')]
    assert manager.mock_calls == expected_calls

只是为了测试它是否有效，main_routine()请在函数add中更改函数调用的顺序，以查看它是否抛出了AssertionError。

请参阅跟踪呼叫顺序和较少冗长的呼叫断言中的更多示例

希望能有所帮助。