如何使用python-mock库

时间:2012-07-12 17:28:58

标签: python mocking

我在python中理解mock库时遇到问题。 这是一个例子: A有两个文件test.py和another.py

test.py

#!
from mock import patch
from another import C


class A(object):
    def method(self):
        return 2


@patch('another.C')
@patch('test.A')
class TestB(object):

    def test1(self, mA, mC):
        print mA, A
        print mC, C

another.py

class C(object):

    def a(self):
        return 3

所以,问题是“为什么A有输出:

<MagicMock name='A' id='13985040'> <MagicMock name='A' id='13985040'>
<MagicMock name='C' id='13993936'> <class 'another.C'>

为什么来自another.py的类无法修补? 在我的情况下,我有很多功能要测试,它从其他文件导入类。 所以我不明白他们是怎么回事。

谢谢

1 个答案:

答案 0 :(得分:1)

使用Python模拟库,您总是在实际使用它的位置修补(模拟)函数或类。

由于您要将'C'导入'test'并在那里使用它,您可以像这样修补它:

@patch('test.C')
@patch('test.A')
class TestB(object):

    def test1(self, mA, mC):
        print mA, A
        print mC, C

这将为您提供如下输出:

<MagicMock name='A' id='3070076204'> <MagicMock name='A' id='3070076204'>
<MagicMock name='C' id='3070084940'> <MagicMock name='C' id='3070084940'>

您可以模拟所需的任何类,方法或函数。但是你需要在实际使用它的位置模拟对象。