如何创建可订阅的 Mock 对象?
How to create a subscriptable Mock object?
假设我有一个代码片段
foo = SomeClass()
bar = foo[1:999].execute()
为了测试这个,我尝试了一些
foo_mock = Mock()
foo_mock[1:999].execute()
不幸的是,这引发了一个异常,
TypeError: 'Mock' object is not subscriptable
那么,如何创建一个 subscriptable 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'>
它之所以被称为“魔法”,是因为它支持__magic__
方法,例如__getitem__
。
假设我有一个代码片段
foo = SomeClass()
bar = foo[1:999].execute()
为了测试这个,我尝试了一些
foo_mock = Mock()
foo_mock[1:999].execute()
不幸的是,这引发了一个异常,
TypeError: 'Mock' object is not subscriptable
那么,如何创建一个 subscriptable 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'>
它之所以被称为“魔法”,是因为它支持__magic__
方法,例如__getitem__
。