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

最新推荐文章于 2025-10-15 17:32:34 发布

原创

最新推荐文章于 2025-10-15 17:32:34 发布 · 5k 阅读

3 ·

CC 4.0 BY-SA版权

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

前面一篇博文简单说了使用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