是否可以在测试中检查调用参数？

Question

如果我们调用主方法 run 来调用我感兴趣的方法 - self. get_request()。

file.py

class A:
 def run():
    some logic...
    request = self.get_request()
    some logic...
    return response

test.py

from file.py import A
def test():
   """
   Inside this test, I want to check the parameters and the value returned by the
   get_request method, but I don't want to check it separately
   I want to check it by calling the parent method - run
   """
   instance = A()
   response = instance.run()
   assertions logic for instance.get_request..

我知道可以模拟一个方法，然后我们可以访问调用次数、参数等。如果我所要求的内容可以通过模拟以某种方式实现，我只想添加我的模拟必须与它模拟的方法具有相同的逻辑（相同）。

Answer 1

您要求的可能是可在 patch - 这允许您模拟函数，同时它仍然保留以前的（或其他一些）功能（请注意，参数本身在 <一个href="https://docs.python.org/3/library/unittest.mock.html?highlight=patch#unittest.mock.Mock" rel="nofollow noreferrer">模拟）。与任何模拟一样，这确实允许您测试调用和调用参数，但不允许您检查函数的返回值。这必须通过其副作用进行测试（在您的情况下，通过返回的响应，该响应应取决于 get_request 的返回值）。

以下是您的情况的说明：

from unittest import mock

class A:
    def run(self):
        request = self.get_request(21)
        return request

    def get_request(self, foo):
        return foo * 2

def test_run():
    instance = A()
    with mock.patch.object(instance, "get_request", wraps=instance.get_request) as mocked:
        assert instance.run() == 42
        mocked.assert_called_once_with(21)

在这种情况下，模拟调用真正的 get_request 方法并返回其结果，同时记录调用和调用参数。
我向 get_request 添加了一些参数进行演示，并直接在 run 中返回调用结果 - 在您的情况下，这当然会有所不同，但想法应该是相同的。