如何模拟属性

问题描述

我在询问如何使用Python3模拟单元测试中的类属性。我尝试了以下方法,这对我跟随文档是有意义的,但不起作用:

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

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


解决方案

属性机制依赖于在对象类上定义的属性属性。不能在类的单个实例上创建"Property Like"方法或属性(要更好地理解,请阅读有关Python的descriptor protocol)

因此,您必须将修补程序应用于您的类-您可以使用with语句,以便在测试后正确还原该类:

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

相关文章