pytest: Mock返回具有特定属性的mock
pytest: Mock返回具有特定属性的mock
提问于 2016-10-21 22:38:00
我发现我经常在测试中一次又一次地使用相同的模式：
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()有没有办法让这个“双模拟”的东西更短？
回答 2
Stack Overflow用户
发布于 2016-10-21 22:49:32
你可以这样做：
mock_db = mock.Mock(spec=DBClass, return_value=mock.Mock(get_data=lambda:"mocked DB data"))Stack Overflow用户
发布于 2016-10-21 23:08:48
您应该能够通过将字典传递给构造函数来实现这一点，如文档的configure mock部分所述，而不是在创建MagicMock之后对其进行配置。
我相信以下几点会达到预期的效果：
mock_db = mock.Mock(spec=DBClass, return_value=mock.MagicMock(**{"get_data.return_value": "mocked DB data"})页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/40179451
复制相关文章
相似问题
腾讯云开发者
