如何创建可订阅的模拟对象？

How to create a subscriptable Mock object?(如何创建可订阅的模拟对象？)

问题描述

假设我有一个代码片段，如下所示

foo = SomeClass()
bar = foo[1:999].execute()

为了测试这一点，我尝试了

foo_mock = Mock()
foo_mock[1:999].execute()

遗憾的是，这引发了异常

TypeError：‘mock’对象不可订阅

那么，我如何创建可订阅Mock对象？

推荐答案

请改用MagicMock。

>>> from unittest.mock import Mock, MagicMock
>>> Mock()[1:999]
TypeError: 'Mock' object is not subscriptable
>>> MagicMock()[1:999]
<MagicMock name='mock.__getitem__()' id='140737078563504'>

之所以称为魔术，是因为它支持__getitem__等__magic__方法。

这篇关于如何创建可订阅的模拟对象？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！