python - 如何模拟缺失的属性

Question

我有一行代码是:

if not hasattr(class.a, u'c'):
    return

如何模拟类以便 class.a.c 为 hasattr 返回 False？

如果我这样做:

>>> from mock import MagicMock
>>> mock_class = MagicMock(spec=[u'a'])
>>> hasattr(mock_class, u'a')
True
>>> hasattr(mock_class, u'b')
False
>>> hasattr(mock_class.a, u'c')
True

虽然我没有指定 class.a.c，但它被 mock 了!!!

Answer 1

实际上 mock_class.a 将创建另一个没有规范的 MagicMock。我能想到的唯一方法是将 mock_class 的属性 a 分配给另一个具有规范的 MagicMock，如下所示:

mock_class = MagicMock(spec=[u'a'])
mock_class.a = MagicMock(spec=[u'a'])
hasattr(mock_class.a, u'c')  # returns False

此外，如果您有一些想要模拟的真实对象，可以做一些recursive autospecing .