python - 来自另一个文件/模块的 Python 模拟补丁

Question

我有一个按预期工作的模拟。

from mock import patch

def second(arg):
    return 3


def first():
    return second('arg')


@patch('test.second')
def test_test(second_mock):
    second_mock.return_value = 47  # We decide this

    call_it = first()

    second_mock.assert_called_once()
    second_mock.assert_called_with('arg')

    assert call_it == 47

但如果我将我的 second() 方法移动到另一个文件，则不会......

from mock import patch
from test_help import second


def first():
    return second('arg')


@patch('test_help.second')
def test_test(second_mock):
    second_mock.return_value = 47  # We decide this

    call_it = first()

    second_mock.assert_called_once()
    second_mock.assert_called_with('arg')

    assert call_it == 47

我得到了同样的错误： AssertionError: Expected 'second' to have been called once。调用 0 次。

我在这里想念什么？

我尝试了几种不同的格式化方式，但似乎都没有。在这种情况下，这甚至是单元测试的最佳实践/包吗？

Answer 1

别担心，你走在正确的道路上，这就是模拟函数的方式。

关于您的探针，请记住您根据调用模拟函数的函数修补命名空间。

因此，当您在模块中module_where_first_is_located进行导入时from test_help import second，第二个被识别为module_where_first_is_located.second.

所以而不是@patch('test_help.second')do @patch('module_of_first.second')。