unittest.mock：在指定的模拟 object 上设置自定义属性（变量）

Question

我正在尝试从另一个库的 class mock现有的 object ，以使用pytest进行单元测试。 但是，来自其他库的属性（不是方法）大多是在运行时设置的。

我想要达到的目标

• 获得 mocking 和 object spec的所有好处
• 设置我的单元测试所需的（嵌套）属性（不是方法），以模拟它们是在 object 创建期间设置的

from unittest.mock import Mock
from otherlib import ClassName

def test_stuff():
    mock_object = Mock(spec=ClassName)
    mock_object.data.outward.key = 12345

    assert mock_object.data.outward.key == 12345  # AttributeError: Mock object has no attribute 'data'

我在代码更改中进行了这些尝试，但没有成功

...
def test_stuff():
    mock_object = Mock(spec=ClassName, **{'data.outward.key': 12345})

    assert mock_object.data.outward.key == 12345

...
def test_stuff():
    mock_object = Mock(spec=ClassName)
    attrs = {'data.outward.key': 12345}
    mock_object.configure_mock(**attrs)

    assert mock_object.data.outward.key == 12345

Answer 1

我想出的最好的方法是使用另一个Mock object来设置main mock object's属性。 它有效，但我想有一个更好的解决方案......？

from unittest.mock import Mock
from otherlib import ClassName

def test_stuff():
    mock_data = Mock(spec=["outward"], key=12345)
    mock_object = Mock(spec=ClassName, data=mock_data)
    mock_object.data.outward.key = 12345

    assert mock_object.data.outward.key == 12345  # tests.py::test_stuff PASSED [100%]