Python Unitest模拟一个对象没有属性

Question

如果对象具有某个属性，则我的函数可以运行一些代码，在极少数情况下，如果对象没有属性，则运行不同的代码。我很难在没有测试属性的情况下创建对象。我尝试了del instance.attribute，但出现了错误。该属性实际上是引擎盖下方的@property。

我有一个具有foo属性的对象实例。一个人如何模拟它，以便当一个人尝试访问instance.foo时，如果没有属性，它会像往常一样添加attributeError？

我尝试了mock.mock（side_effect = attributeError（'boom！'）），但它仅适用于方法。

Answer 1

您可以尝试使用属性，通常您可以设置各自的副作用。这是一个简单的工作示例：

from unittest import mock
import pytest

class Foo:
    @property
    def bar(self):
        return "bar"


def test_no_foo():
    bar_mock = mock.PropertyMock()
    with mock.patch(f"{__name__}.Foo.bar", bar_mock):
        bar_mock.side_effect = AttributeError('Boom!')
        foo = Foo()
        with pytest.raises(AttributeError):
            foo.bar

当您修补类中的属性而不是对象中的属性时，您也可以使用patch.Object进行此操作，如果您可以通过访问对象访问对象的访问对象：

def test_no_foo():
    bar_mock = mock.PropertyMock()
    foo = Foo()
    with mock.patch.object(foo.__class__, "bar", bar_mock):
        bar_mock.side_effect = AttributeError('Boom!')
        with pytest.raises(AttributeError):
            foo.bar