pytest 在类的 __init__ 中模拟函数调用

Question

我有这样的课，

from external_package.module.sub_module import fun

class MyClass:
    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.z = fun()
        
    def printer(self):
        print(f"X : {self.x}, y : {self.y}, z: {self.z}")

我想为这个类编写一个单元测试并模拟 init 方法中发生的函数调用。

我试过这样打补丁，

from external_package.module import sub_module
    
def test_my_class(mocker):

    mocker.patch.object(sub_module, 'fun', return_value="this is mocked")
    obj = MyClass(10, 20)
    
    assert obj.z == "this is mocked"

但我可以看到调用没有打补丁，并且在 init 方法中发生了函数调用。

如何使用 pytest-mock 或任何其他模拟包修补调用？

Answer 1

您应该使用使用它的路径而不是原始路径或位置来模拟您的函数。在下面的示例中，定义了对象 MyClass 中的 fun 方法将被模拟，返回值将是“this is mocked”。

from external_package.module import sub_module
from my_class import MyClass

def test_my_class(mocker):

    mocker.patch.object(MyClass.submodule, 'fun', return_value="this is mocked")
    obj = MyClass(10, 20)
    
    assert obj.z == "this is mocked"

查看link 了解如何在 Python 中模拟对象的一些很好的示例。

Answer 2

我可以通过稍微更改源代码中的导入语句来解决这个问题，

from external_package.module import sub_module 

class MyClass:
    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.z = sub_module.fun()
        
    def printer(self):
        print(f"X : {self.x}, y : {self.y}, z: {self.z}")

请注意，我正在导入子模块并从导入的子模块中调用方法。

现在，在测试中，我像这样使用了pytest-mock 夹具，

import external_package
    
def test_my_class(mocker):

    method = mocker.patch("external_package.module.sub_module.fun")
    method.return_value = "this is mocked"

    obj = MyClass(10, 20)
    
    assert obj.z == "this is mocked"

我不确定为什么这有效，而不是早期的方法。希望有人能解决这个问题。