如何在使用patch.object(autospec=True)时获取Mock而非MagicMock对象
解决方法
要让patch.object生成Mock或NonCallableMock而非MagicMock子类,有两种常用实现方式:
方式一:保留autospec特性并指定mock类型
如果需要保留autospec=True带来的类型校验特性,同时替换生成的mock类,可通过new_callable参数指定要使用的mock类型:
from unittest.mock import patch, NonCallableMock class X: x=1 with patch.object(X, 'x', autospec=True, new_callable=NonCallableMock) as my_mock: print(type(my_mock)) # 输出 <class 'unittest.mock.NonCallableMock'>
因为X.x是不可调用的int类型,这里用NonCallableMock更匹配;如果原对象是可调用的(比如函数、方法),则将new_callable的值替换为Mock即可。
方式二:直接传入自定义mock实例
如果不需要autospec的特性,可直接通过new参数传入预先创建的Mock或NonCallableMock实例:
from unittest.mock import patch, NonCallableMock class X: x=1 with patch.object(X, 'x', new=NonCallableMock()) as my_mock: print(type(my_mock)) # 输出 <class 'unittest.mock.NonCallableMock'>
内容的提问来源于stack exchange,提问作者salius
相关产品推荐
相关产品推荐

