如何用Python unittest检测代码是否进入except代码块（无抛异常场景）

我完全懂你的困扰！之前写测试的时候也碰到过这种“吞掉异常只打日志”的情况，assertRaises确实派不上用场——毕竟它的设计就是用来检测代码主动抛出异常的场景，而你的foo()把异常内部消化了，根本没往外抛。

针对这种情况，最直接的思路是验证日志是否被正确记录，或者通过mock来捕获except块内的操作。下面给你两种实用的实现方式：

方法1：Mock日志模块，断言日志调用 #

如果你的foo()在except块里只做了日志记录，那可以通过unittest.mock来mock日志的输出方法，然后断言它是否被调用（以及调用参数是否符合预期），以此证明代码进入了except块。

假设你的foo()代码大概是这样：

import logging

def foo():
    try:
        # 这里是可能触发异常的逻辑
        raise ValueError("模拟异常")
    except ValueError as e:
        logging.error(f"捕获到异常: {e}")

对应的测试用例可以这么写：

import unittest
from unittest.mock import patch
import logging
from your_module import foo  # 替换成你的模块名

class TestFooExceptionHandling(unittest.TestCase):
    def test_foo_enters_except_block(self):
        # Mock logging.error方法
        with patch.object(logging, 'error') as mock_log_error:
            # 调用待测试的foo方法
            foo()
            
            # 断言日志方法被调用过，说明进入了except块
            mock_log_error.assert_called_once()
            
            # 可选：进一步验证日志内容是否正确
            expected_log_msg = "捕获到异常: 模拟异常"
            mock_log_error.assert_called_with(expected_log_msg)

方法2：Mock异常触发源，强制进入except块 #

如果foo()里的异常是由某个依赖方法触发的，你可以mock这个依赖方法，让它主动抛出异常，然后结合日志验证来确认except块被执行。

比如foo()依赖some_dangerous_func()触发异常：

import logging

def some_dangerous_func():
    # 实际可能抛出异常的方法
    return 1 / 0

def foo():
    try:
        some_dangerous_func()
    except ZeroDivisionError as e:
        logging.error(f"处理异常: {e}")

测试用例可以mocksome_dangerous_func()让它抛出异常，再验证日志：

import unittest
from unittest.mock import patch
import logging
from your_module import foo, some_dangerous_func

class TestFooExceptionHandling(unittest.TestCase):
    def test_foo_handles_exception(self):
        with patch.object(logging, 'error') as mock_log_error:
            # Mock依赖方法，强制抛出异常
            with patch('your_module.some_dangerous_func') as mock_dangerous:
                mock_dangerous.side_effect = ZeroDivisionError("被mock的异常")
                
                foo()
                
                # 断言依赖方法被调用，且日志被记录
                mock_dangerous.assert_called_once()
                mock_log_error.assert_called_once_with("处理异常: 被mock的异常")

关键说明 #

• assertRaises之所以无效，是因为它要求被测试代码必须向外抛出异常，而你的foo()在except块内完成了异常处理，没有将异常传递到测试框架中，所以这个方法不适用。
• 通过mock日志或依赖方法，我们相当于“间接证明”了代码进入了except块——毕竟只有进入except块，才会执行日志记录或相关操作。

内容的提问来源于stack exchange，提问作者rand0rn