如何模拟属性

时间:2017-03-09 20:08:40

标签: python python-3.x pytest python-unittest python-mock

我正在问如何使用Python 3在单元测试中模拟一个类属性。我尝试了以下内容,这对我来说对文档很有意义,但它不起作用:

foo.py:

class Foo():
    @property
    def bar(self):
        return 'foobar'


def test_foo_bar(mocker):
    foo = Foo()
    mocker.patch.object(foo, 'bar', new_callable=mocker.PropertyMock)
    print(foo.bar)

我已经安装了pytestpytest_mock并按照以下方式运行测试:

pytest foo.py

我收到以下错误:

>       setattr(self.target, self.attribute, new_attr)
E       AttributeError: can't set attribute

/usr/lib/python3.5/unittest/mock.py:1312: AttributeError

我的期望是测试运行没有错误。

1 个答案:

答案 0 :(得分:9)

属性机制依赖于在对象类上定义的属性属性。您无法创建"属性,例如"单个类实例上的方法或属性(为了更好地理解,请阅读Python' s descriptor protocol

因此,您必须将补丁应用于您的类 - 您可以使用with语句,以便在测试后正确恢复该类:

def test_foo_bar(mock):
    foo = Foo()
    with mock.patch(__name__ + "Foo.bar", new=mocker.PropertyMock)
        print(foo.bar)