0

库中的patch函数mock对事物的导入方式很敏感。为什么我不能只使用最初定义函数的完全限定名称，而不管它是如何在其他模块中导入的，有什么深层次的原因吗？

使用“模块导入”工作正常

补丁示例.py：

# WORKS!
from mock import patch
import inner

def outer(x):
    return ("outer", inner.inner(x))

@patch("inner.inner")
def test(mock_inner):
    mock_inner.return_value = "MOCK"
    assert outer(1) == ("outer", "MOCK")
    return "SUCCESS"

if __name__ == "__main__":
    print test()

内部.py：

def inner(x):
    return ("inner.inner", x)

运行python patch_example.py只是输出成功。

但是，更改导入可能会产生非常严重的后果

使用模块别名仍然有效

# WORKS!
from mock import patch
import inner as inner2

def outer(x):
    return ("outer", inner2.inner(x))

@patch("inner.inner")
def test(mock_inner):
    mock_inner.return_value = "MOCK"
    assert outer(1) == ("outer", "MOCK")
    return "SUCCESS"

if __name__ == "__main__":
    print test()

但是，直接导入符号需要您更改完全限定名称。

直接导入，inner.inner作为完全限定名称。

# FAILS!
from mock import patch
from inner import inner

def outer(x):
    return ("outer", inner(x))

@patch("inner.inner")
def test(mock_inner):
    mock_inner.return_value = "MOCK"
    assert outer(1) == ("outer", "MOCK")
    return "SUCCESS"

if __name__ == "__main__":
    print test()

生产

% python patch_example.py
Traceback (most recent call last):
  File "patch_example.py", line 14, in <module>
    print test()
  File "/usr/local/lib/python2.7/site-packages/mock/mock.py", line 1305, in patched
    return func(*args, **keywargs)
  File "patch_example.py", line 10, in test
    assert outer(1) == ("outer", "MOCK")
AssertionError

如果我将完全限定路径更新为patch_example.inner，补丁仍然失败。

# FAILS!
from mock import patch
from inner import inner

def outer(x):
    return ("outer", inner(x))

@patch("patch_example.inner")
def test(mock_inner):
    mock_inner.return_value = "MOCK"
    assert outer(1) == ("outer", "MOCK")
    return "SUCCESS"

if __name__ == "__main__":
    print test()

% python patch_example.py

Traceback (most recent call last):
  File "patch_example.py", line 14, in <module>
    print test()
  File "/usr/local/lib/python2.7/site-packages/mock/mock.py", line 1305, in patched
    return func(*args, **keywargs)
  File "patch_example.py", line 10, in test
    assert outer(1) == ("outer", "MOCK")
AssertionError

不过，使用__main__.inner我的完全限定名称修补正确的东西。

# WORKS!
from mock import patch
from inner import inner

def outer(x):
    return ("outer", inner(x))

@patch("__main__.inner")
def test(mock_inner):
    mock_inner.return_value = "MOCK"
    assert outer(1) == ("outer", "MOCK")
    return "SUCCESS"

if __name__ == "__main__":
    print test()

打印“成功”

那么，为什么当from inner import inner使用原始符号的完全限定名称inner.inner或使用主 python 模块的名称而不是使用 inner 导入时，我不能修补它的值__name__？

在 OS X 上使用 Python 2.7.12 测试。

4

1 回答 1

1

问题是，一旦您直接导入符号，您在模块中使用的绑定与模块中找到的绑定之间就没有任何链接。因此模块不会更改已导入的符号。__main__innerpatch

使用别名导入模块并不重要，因为patch会查找sys.modules字典，它仍然跟踪原始名称，所以这就是为什么这样做（实际上：当调用模拟模块时，模块是新导入的，所以没关系调用时导入的名称patch）

换句话说：您必须修补这两个绑定，因为它们实际上是不相关的。补丁无法知道所有引用的inner.inner结束位置并对其进行修补。

在这种情况下，第二个参数patch可能有助于指定一个现有的模拟对象，该对象可以共享以修补所有绑定。

于 2016-08-11T17:53:19.613 回答