python3 mock 不适用于所有路径

Question

生产文件 (production_file.py) 是：

class MyError(Exception):
    pass

class MyClass:
    def __init__(self):
        self.value = None

    def set_value(self, value):
        self.value = value

    def foo(self):
        raise RuntimeError("error!")


class Caller:
    def bar(self, smth):
        obj = MyClass()
        obj.set_value(smth)

        try:
            obj.foo()
        except MyError:
            pass

        obj.set_value("str2")
        obj.foo()

测试文件（test.py）：

import unittest

from unittest.mock import patch
from unittest.mock import call
from production_file import MyClass, Caller

class MyTest(unittest.TestCase):
    def test_caller(self):
        with patch('production_file.MyClass', autospec=MyClass) as MyClassMock:
            my_class_mock_obj = MyClassMock.return_value
            my_class_mock_obj.foo.side_effect = [MyError("msg"), "text"]

            caller = Caller()
            caller.bar("str1")

            calls = [call("str1"), call("str2")]

            my_class_mock_obj.set_value.assert_has_calls(calls)

if __name__ == '__main__':
    unittest.main()

以上方法有效。但是如果我将生产类（MyError、MyClass、Caller）移动到测试文件中，并将补丁更新为：

with patch('test.MyClass', autospec=MyClass) as MyClassMock:

然后不再模拟实例方法“foo”。

有人知道这是为什么吗？

我在一些更复杂的代码中也遇到过类似的问题，其中生产代码在 my_package/src/production_file.py 中，而测试在 my_package/tests/test_file.py 中。 Python 不会产生路径错误，路径是正确的，但 mock 仍然不起作用。

Answer 1

如果您将test.py 运行为__main__，则它不是test.MyClass，而是__main__.MyClass，或者在这两种情况下都是__name__+".MyClass"。

通过添加打印语句，我能够确定使用的类和修补的类不同：

class Caller:
    def bar(self, smth):
        print(MyClass) #lets see what we are actually making an instance of...
        obj = MyClass()
        ...

当补丁应用到它正在使用的类时，您会看到如下内容：

<MagicMock name='MyClass' spec='MyClass' id='4387629656'>

但是当类移到test.py 时，你会看到类似：

<class '__main__.MyClass'>

这表明：

1. MyClass（至少是用于测试的那个）没有应用补丁。
2. 需要修补的类名是__main__.MyClass

但是，由于这样的设置，您的“更...复杂的情况”很可能无法正常工作：

from production_file import MyClass

class MyError(Exception):
    pass


class Caller:
    def bar(self, smth):
        print(MyClass)
        obj = MyClass()
        ...

class MyTest(unittest.TestCase):
    def test_caller(self):
        with patch('production_file.MyClass', autospec=MyClass) as MyClassMock:
            ...

在这种情况下，production_file.MyClass 正在被修补，MyClass 正在从 production_file 导入，因此正在修补正确的类，但输出仍然是：

<class 'production_file.MyClass'>

这是因为Class是直接导入到本地命名空间的，所以当补丁应用到production_file时本地命名空间仍然不受影响，我们可以检查补丁是否实际应用了：

...
def bar(self, smth):
    print(MyClass)
    from production_file import MyClass as pf_MyClass
    print(pf_MyClass)
...


#output:
<class 'production_file.MyClass'>
<MagicMock name='MyClass' spec='MyClass' id='4387847136'>

如果是这种情况，您只需要导入模块，而不是直接导入类。然后，一旦应用了补丁，您将直接从文件中使用它：

import production_file

...
class Caller:
    def bar(self, smth):
        print(production_file.MyClass)
        obj = production_file.MyClass()
        ...

class MyTest(unittest.TestCase):
    def test_caller(self):
        with patch('production_file.MyClass', autospec=MyClass) as MyClassMock:
            ...