python - mock.patch() 不修补在函数调用中调用了几个级别的类

Question

我有一个任务

def task():
   a = worker()
   a.do_some_work()

Worker 本身是独立模块中的独立类，他们使用这样的 Driver 类，

class Worker(object):
   def __init__(self):
     self.driver = Driver(args)
     ...

Driver 是单独模块中的单独类

所以当我尝试类似的东西时

with patch('package.module.Driver', new=Mock(return_value=999)):
   task()

在任务中仍然有一个 Driver 类实例，但不是模拟。那是错误的。如何解决？

UPD1:Driver 和 Worker 存在于不同的模块中，Worker import Driver

Answer 1

您的 MyProject.Workers.base 模块将 Driver 作为全局导入:

from MyProject.utils.drivers import Driver

这是对 Driver 类的新的、单独的引用。如果您现在修补 MyProject.utils.drivers.Driver，Worker 类将看不到修补的对象；因为它将使用它自己的全局引用。

修补全局 Worker 实际使用:

with patch('MyProject.Workers.base.Driver', new=Mock(return_value=999)):
    task()

参见 Where to patch section mock 文档。