Python mock call_args_list解包参数断言的元组

时间:2016-09-23 21:05:12

标签: python unit-testing python-unittest python-unittest.mock

我在处理Mock.call_args_list返回的嵌套元组时遇到了一些麻烦。

def test_foo(self):
    def foo(fn):
        fn('PASS and some other stuff')

    f = Mock()
    foo(f)
    foo(f)
    foo(f)

    for call in f.call_args_list:
        for args in call:
            for arg in args:
                self.assertTrue(arg.startswith('PASS'))

我想知道是否有更好的方法来解压缩模拟对象上的call_args_list以便进行断言。这个循环有效,但感觉必须有一个更直接的方式。

2 个答案:

答案 0 :(得分:42)

我认为这里的许多困难都包含在对待"呼叫"宾语。它可以被认为是一个有2个成员(args, kwargs)的元组,因此解压缩它通常很好:

args, kwargs = call

一旦它被解压缩,那么你可以单独为args和kwargs做出断言(因为一个是元组而另一个是dict)

def test_foo(self):
    def foo(fn):
        fn('PASS and some other stuff')

    f = Mock()
    foo(f)
    foo(f)
    foo(f)

    for call in f.call_args_list:
        args, kwargs = call
        self.assertTrue(all(a.startswith('PASS') for a in args))

请注意,有时简洁性并不有用(例如,如果有错误):

for call in f.call_args_list:
    args, kwargs = call
    for a in args:
        self.assertTrue(a.startswith('PASS'), msg="%s doesn't start with PASS" % a)

答案 1 :(得分:9)

更好的方法可能是建立预期的调用,然后使用直接断言:

>>> from mock import call, Mock
>>> f = Mock()
>>> f('first call')
<Mock name='mock()' id='31270416'>
>>> f('second call')
<Mock name='mock()' id='31270416'>
>>> expected_calls = [call(s + ' call') for s in ('first', 'second')]
>>> f.assert_has_calls(expected_calls)

请注意,调用必须是顺序的,如果您不希望这样,则将any_order kwarg覆盖到断言。

另请注意,允许在此之前或之后进行额外通话 指定的电话。如果你不想这样,你需要添加另一个断言:

>>> assert f.call_count == len(expected_calls)

解决mgilson的评论,这是创建一个可用于进行通配符相等比较的虚拟对象的示例:

>>> class AnySuffix(object):
...     def __eq__(self, other):
...         try:
...             return other.startswith('PASS')
...         except Exception:
...             return False
...        
>>> f = Mock()
>>> f('PASS and some other stuff')
<Mock name='mock()' id='28717456'>
>>> f('PASS more stuff')
<Mock name='mock()' id='28717456'>
>>> f("PASS blah blah don't care")
<Mock name='mock()' id='28717456'>
>>> expected_calls = [call(AnySuffix())]*3
>>> f.assert_has_calls(expected_calls)

失败模式的一个例子:

>>> Mock().assert_has_calls(expected_calls)
AssertionError: Calls not found.
Expected: [call(<__main__.AnySuffix object at 0x1f6d750>),
 call(<__main__.AnySuffix object at 0x1f6d750>),
 call(<__main__.AnySuffix object at 0x1f6d750>)]
Actual: []