如何Mock修补未被外部函数返回的内部函数并捕获其返回值

针对你提出的场景——bar.py中的my_fn调用foo.fn但未返回其结果，要在测试中获取foo.fn的返回值且不修改bar.py，可以通过以下两种方式实现：

情况1：预设返回值并验证类型 #

如果只需验证逻辑对返回值的处理（无需调用真实的foo.fn），直接通过mock对象设置返回值，随后断言即可：

from mock import patch
import foo
import bar

@patch("foo.fn")
def test_bar(mock_foo_fn):
    # 预设foo.fn的返回值
    mock_foo_fn.return_value = {"name": "test"}
    bar.my_fn()
    # 断言返回值符合预期类型
    assert isinstance(mock_foo_fn.return_value, dict)

情况2：捕获真实函数的返回值 #

如果需要保留foo.fn的真实执行逻辑，同时捕获它的返回值，可以用side_effect包装原函数，记录返回结果：

from mock import patch
import foo
import bar

def test_bar():
    captured_result = None
    original_foo_fn = foo.fn

    # 包装原函数，捕获返回值
    def capture_return(*args, **kwargs):
        nonlocal captured_result
        result = original_foo_fn(*args, **kwargs)
        captured_result = result
        return result

    with patch("foo.fn", side_effect=capture_return):
        bar.my_fn()
        # 断言真实返回值的类型
        assert isinstance(captured_result, dict)

内容的提问来源于stack exchange，提问作者aweeeezy