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

问题描述

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

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__方法。