检查函数是否被调用而不改变其行为[重复]

Question

我如何测试 Python 函数是否被调用，完全不改变其行为？

有很多相关的帖子，但我找不到涵盖所有内容的帖子：

• How to test that a function is called within a function with nosetests（改变行为）
• Patching a method without changing how the method works?（不适用于参数和返回值）

基本上，我只需要监视函数就可以知道它是否被调用了。

例如，以下 sn-p 具有极其令人惊讶的行为

from unittest.mock import patch

def func(a,b):
    print('was called')
    return a * b

with patch('test.func') as patched_function:
    print(func(4,5))

print(patched_function.called)

输出：

<MagicMock name='func()' id='1721737249456'>
True
was called
20
False

虽然我只是期待

was called
20
True

Answer 1

你应该像这样为函数创建一个装饰器：

class CheckCall(object):
    called = False

    def __init__(self, func):
        self.func = func

    def __call__(self *args, **kwargs):
        self.called = True
        return self.func(*args, **kwargs)

然后你可以像这样应用你的装饰器

with patch('test.func') as patched_function:
    patched_function = CheckCall(patched_function)
    print(patched_function.called) # Prints False
    patched_function(...)
    print(patched_function.called) # Prints True