Python的单元测试unittest中的Mock使用小结

最新推荐文章于 2024-05-28 15:49:19 发布

python012

最新推荐文章于 2024-05-28 15:49:19 发布

阅读量4.9k

点赞数 1

分类专栏： unittest 单元测试 Python

本文链接：https://blog.csdn.net/python012/article/details/79944600

版权

这篇博客总结了Python单元测试中使用unittest库的Mock功能，包括如何处理被测函数内的input自动化输入，验证函数调用，以及讲解mock.patch()与mock.patch.object()的区别及其应用场景。

摘要由CSDN通过智能技术生成

前面一篇博文简单说了使用unittest.mock对无返回值的函数做单元测试。这里是更多一些例子的总结。

被测函数中使用到了input需要自动化输入

#!/usr/bin/env python3

from unittest import TestCase
from unittest.mock import patch
from unittest import main

def func_input():
    name = input("Enter your name: ")
    print('Your name is {}'.format(name))

def test_func_input():
    with patch('builtins.input') as mocked_input:
        mocked_input.side_effect = ('Jo',)  #当input的时候会输入Jo
        with patch(