FastAPI自定义错误处理器测试异常响应失败问题

核心原因 #

FastAPI的TestClient默认会重新引发服务器端抛出的所有异常，哪怕你已经注册了全局异常处理器。这就是Postman能正常拿到自定义响应，但测试用例里直接抛出异常、无法获取响应断言的根本原因。

解决方案 #

1. 修改TestClient初始化参数 #

在创建TestClient时添加raise_server_exceptions=False，让客户端返回异常处理器生成的响应而非直接抛出异常：

from fastapi.testclient import TestClient
from app.main import api

class TestBrands:
    @property
    def client(self) -> TestClient:
        # 添加参数禁用服务器异常抛出
        return TestClient(api, base_url=f'http://localhost:8000/brands', raise_server_exceptions=False)

    def test_create_brand_already_exists_exception(self):
        self.client.post('', json={'name': 'schecter', 'founded_in': 1976})
        response = self.client.post('', json={'name': 'schecter'})
        # 现在可以正常断言响应状态码和内容
        assert response.status_code == 422
        assert response.json()['message'] == '品牌已存在'  # 替换为你的异常提示文本

2. 验证自定义异常与处理器逻辑 #

确保你的异常定义和处理器没有逻辑漏洞：

• 确认APIException类的属性定义正确：

  class APIException(Exception):
      def __init__(self, status_code: int, message: str):
          self.status_code = status_code
          self.message = message
  

• 确认路由中是主动raise该异常，而非触发其他未捕获错误：

  @router.post('')
  async def create_brand(name: str, founded_in: int, db: Session = Depends(get_db)):
      if db.query(Brand).filter(Brand.name == name).first():
          raise APIException(status_code=422, message="品牌已存在")
      # 品牌创建逻辑
  

• 确认main.py中导入了所有必要依赖：

  from fastapi import FastAPI, Request, JSONResponse, HTTPStatus
  from your_module import APIException, include_routers
  

3. 验证全局异常处理器覆盖范围 #

可以添加测试用例验证通用500异常的处理逻辑：

def test_internal_server_error(self):
    # 模拟触发未被APIException捕获的异常（比如路由中raise ValueError）
    response = self.client.get('/test-internal-error')
    assert response.status_code == 500
    assert response.json()['message'] == 'Internal server error.'

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