如何模拟在单独的命名空间中使用的类？

Question

包装结构:

pqr/pq.py
test.py

pqr/pq.py 有以下结构

其中lmn是全局安装的pip模块

结构 pq.py

from lmn import Lm

class Ab():

    def __init__(self):
        self.lm = Lm()

    def echo(self, msg):
        print msg

test.py 有以下结构

from pqr.pq import Ab

如何Lm()在这里模拟类以便测试Ab类中的所有方法？

Answer 1

Lm来自哪里并不重要.您导入Lm到pqr.pq命名空间为全局有,所以你只需要更换的名字里有和其他地方.那是因为该Ab.__init__方法将在它自己的模块中"本地"查找它.

因此,使用mock库您需要做的就是修补名称pqr.pq.Lm:

import mock
from pqr.pq import Ab

with mock.patch('pqr.pq.Lm') as mock_lm:
    # within this block, `Lm` in the `pqr.pq` namespace has been replaced
    # by a mock object
    test_instance = Ab()
    # assert that the patch succeeded; .return_value is used because
    # Ab.__init__ *calls* Lm()
    assert test_instance.lm is mock_lm.return_value

还看到哪里修补部分的的mock文档.