Monkey 修补 Python 中另一个模块的对象

Question

我在模块 1 中有一个类，比如 foo。

    class foo():
       def __init__(var1, var2):
         self.var1 = var1
         self.var2 = var2
      
      def method1(self):
          pass
      def method2(self):
         pass

foo_ = foo("blah", "blah")

foo_ 对象被广泛用于代码库的各种模块中。我必须为模块 2 中使用 foo_ 对象的方法编写一个测试。

模块 2：

from module1 import foo_
import module3
def blah():
   foo_.method1()
   module3.random_method()
   return blah

模块 3 中的 random_method 还导入并调用了 foo_ 的另一个方法。我的测试模块中有一个虚拟的 foo_ 对象。测试是针对 blah 方法的。这是我到目前为止所做的。

测试模块：

from module1 import foo
from module2 import blah

@pytest.fixture()
def dummy_foo_():
 var1 = 'blah'
 var2 = 'blah'
 return foo(var1,var2)

def test_blah(dummy_foo_):
 assert blah() == 'blah'

有没有一种方法可以让 test_blah 在 blah 方法使用虚拟 foo_ 对象而不是模块 1 中导入的对象的情况下工作？

Answer 1

您可以使用mock.patch 临时修补module1.foo_ 和dummy_foo_，将其用作上下文管理器

from mock import patch
def test_blah(dummy_foo_):
    with patch('module1.foo_', dummy_foo_) as mock:
        assert blah() == 'blah'