如何测试Python/Django视图中捕获异常的结果（含唯一约束场景代码）

我来给你一步步拆解怎么测试这段视图代码里的异常捕获逻辑，用Django自带的测试框架就能轻松搞定：

1. 搭建测试基础环境 #

首先导入测试相关模块，然后在测试类的setUp方法里准备好核心测试数据——关键是先创建一个符合unique_together约束的TheModel实例，这样后续提交相同数据时才会触发IntegrityError。如果你的视图需要登录或者依赖request.org，也要在这一步做好模拟：

from django.test import TestCase, Client
from django.contrib.auth.models import User
from django.contrib.messages import get_messages
from your_app.models import TheModel
from your_app.views import YourCreateView  # 替换成你的实际视图类名

class TestUniqueConstraintHandling(TestCase):
    def setUp(self):
        self.client = Client()
        # 1. 创建测试用户（如果视图需要登录权限）
        self.test_user = User.objects.create_user(username='test_user', password='test_pass123')
        # 2. 创建关联的org实例（确保和视图里的self.request.org逻辑一致）
        self.test_org = ...  # 替换成你的Org模型创建代码，比如 Org.objects.create(name='Test Org')
        # 3. 预先创建满足唯一约束的TheModel实例，为后续重复提交做准备
        TheModel.objects.create(
            org=self.test_org,
            field1='test_value_1',
            field2='test_value_2'
        )
        # 4. 登录测试用户（如果视图要求登录）
        self.client.login(username='test_user', password='test_pass123')

2. 模拟重复提交请求 #

构造和已存在实例完全相同的field1、field2数据，用Client发送POST请求到视图的URL，触发唯一约束异常：

def test_duplicate_submission_triggers_error(self):
    # 构造POST表单数据，和已存在实例的字段值完全一致
    post_data = {
        'field1': 'test_value_1',
        'field2': 'test_value_2',
        # 补充表单里其他必填字段的数据
    }
    # 发送POST请求到视图的实际路由
    response = self.client.post('/your-view-url/', post_data)

3. 验证异常处理结果 #

现在要检查视图是否正确捕获了异常，并完成了预期的处理：

• 验证数据库无新增对象：确保重复提交没有创建新的TheModel实例
• 验证错误消息已生成：检查Django的messages框架是否返回了指定的错误提示
• 验证视图响应符合预期：比如返回表单页面（状态码200）或者重定向（状态码302），根据你的视图逻辑判断

# 验证数据库对象数量未增加（还是之前创建的1个）
    self.assertEqual(TheModel.objects.count(), 1)
    
    # 验证错误消息是否正确添加到messages中
    messages = list(get_messages(response.wsgi_request))
    self.assertEqual(len(messages), 1)
    self.assertEqual(str(messages[0]), 'field1 already exists.')
    
    # 验证视图响应状态码（比如CreateView重复提交后通常回到表单页面，状态码200）
    self.assertEqual(response.status_code, 200)

几个额外的注意点 #

• 你的原代码里用了e.message，但Django 3.1及以上版本这个属性已被弃用，建议改成str(e)获取异常信息，避免版本兼容问题：

  except IntegrityError as e:
      if 'UNIQUE constraint' in str(e):
          messages.error(self.request, _('field1 already exists.'))
  

• 如果是DRF的API视图，错误信息不会在messages框架里，要检查响应的JSON数据字段
• 不要用assertRaises测试这个场景——因为异常已经被视图内部捕获处理，我们要验证的是处理后的结果，而非异常是否抛出

内容的提问来源于stack exchange，提问作者Sam Louis Milligan