摘要.
使用 mock 对象替换系统的一部分并且能获取它们的使用情况。
具体的说,你可以获取方法 / 属性的使用情况以及它们的调用参数。也可以指定返回值和设置属性。
思路是将对象设置为 mock 对象,然后根据需要配置 mock 对象,比如返回值、异常等。
# 设置类方法的返回值并验证被调用过
from unittest.mock import MagicMock
class ProductionClass:
def method(self, arg1, arg2, key):
pass
thing = ProductionClass()
thing.method = MagicMock(return_value=3)
print(thing.method(3, 4, key='value'))
thing.method.assert_called_with(3, 4, key='value')
# 设置调用 mock 后的副作用
side_effect 可以是一个函数、iterable 或 exception(类或实例)。
详情见:https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.side_effect
# 使用装饰器和上下文管理器
上面『设置类方法的返回值并验证被调用过』比较啰嗦且不利用复用,所以改写为装饰器或者
上下文管理器。
装饰器:
from unittest.mock import patch
from strategy.module import ProductionClass
@patch('strategy.module.ProductionClass.method', return_value=3)
def test(method):
print(ProductionClass().method(3, 4, key='value'))
method.assert_called_with(3, 4, key='value')
test()
上下文管理器:
from unittest.mock import patch
from strategy.module import ProductionClass
def test():
with patch.object(ProductionClass, 'method', return_value=3) as mock_method:
print(ProductionClass().method(3, 4, key='value'))
mock_method.assert_called_with(3, 4, key='value')
test()
对于其他对象,比如函数,只需要将 patch
中的路径变为函数的路径即可。
https://www.cnblogs.com/jay54520/p/9081873.html https://www.cnblogs.com/jay54520/p/9081873.html