python

Python mock 的使用 - Jay54520 - 博客园

文章暂存

systemime
2021-01-05
2 min

摘要.

使用 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

上次编辑于: 2021/5/20 下午3:26:49