python mock 打桩抛出异常

最新推荐文章于 2025-06-11 14:51:15 发布

原创最新推荐文章于 2025-06-11 14:51:15 发布 · 5.6k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #mock #打桩 #异常

python 专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种使用Python的mock库中的side_effect参数来模拟异常的方法，这种方法有助于提高单元测试的覆盖率。通过示例展示了如何对自定义函数及Python内置库函数进行异常模拟。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

总述：

在python的LLT测试中，需要保证覆盖率，但是有一个问题就是，python代码本来就很短，一个函数可能就几行，而且工程代码为了方便定位错误，会在函数的前后包围try_except_else块，这一个块就占了几行，如果没一个函数都无法覆盖这个异常处理的块，那么覆盖率将会受到很大的影响，这篇博客演示如何用python模拟异常。

做法：

其实只需要知道mock有一个side_effect的参数就很容易了，这个参数就是返回打桩函数的异常情况的。

示例：

import os

class classA():

    def getnum(self):
        return 0


    def self_function(self):
        try:
            if  self.getnum()==0:
                print("test_self_function you are very good")
        except:
            print("test_self_function the except module")
        else:
            print("test_self_function the else module")

定义测试函数：

    @patch("aa.classA.getnum")
    def test_self_function(self,mock_getnum):
        mock_getnum.side_effect = IOError()
        classA().self_function()

这样运行这个测试函数就会被捕获到异常。

当然这么强大的函数不仅仅可以打桩自己写的函数，甚至可以捕获python的内置库函数。

加上一个函数：

    def python_function(self,path):
        try:
            if  os.path.exists(path):
                print(" test_python_function you are very good")
        except:
            print("test_python_function the except module")
        else:
            print("test_python_function the else module")

其测试函数：

    @patch("os.path.exists")
    def test_python_function(self,mock_exists):
        mock_exists.side_effect = IOError
        classA().python_function("/you")

可以看到，这个地方也成功的捕获到了异常。