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

Question

我在处理 Mock.call_args_list returns.

的嵌套元组时遇到了一些问题

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 以做出我的断言。这个循环有效，但感觉必须有更直接的方法。

Answer 1

更好的方法可能是自己建立预期调用，然后使用直接断言：

>>> 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: []

Answer 2

我认为这里的许多困难都集中在 "call" 对象的处理上。它可以被认为是一个有 2 个成员 (args, kwargs) 的元组，所以解压它通常很好：

args, kwargs = call

解压后，您可以分别对 args 和 kwargs 进行断言（因为一个是元组，另一个是字典）

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)

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

Python mock call_args_list unpacking tuples for assertion on arguments

python

unit-testing

python-unittest

python-unittest.mock