pytest：模拟返回具有特定属性的模拟

Question

我发现我经常在测试中反复使用相同的模式：

mock_get_data = mock.MagicMock()
mock_get_data.get_data.return_value = "mocked DB data"
mock_db = mock.Mock(spec=DBClass, return_value=mock_get_data)

它将用于测试（使用修补）这种类型的代码：

db_connector = DBClass(settings)
print db_connector.get_data()

有没有办法让那个＆＃39; double mock＆＃39;东西更短？

Answer 1

你可以这样做：

mock_db = mock.Mock(spec=DBClass, return_value=mock.Mock(get_data=lambda:"mocked DB data"))

Answer 2

您应该可以通过将字典传递给构造函数来完成此操作，如文档的configure mock部分所述，而不是在创建MagicMock后配置它。

我相信以下内容会达到预期效果：

mock_db = mock.Mock(spec=DBClass, return_value=mock.MagicMock(**{"get_data.return_value": "mocked DB data"})