Django单元测试校验slug唯一约束时触发TransactionManagementError如何处理

错误根因 #

你遇到的TransactionManagementError是PostgreSQL事务机制与Django测试封装逻辑共同作用的结果：

• Django的TestCase会把整个测试方法的执行流程包裹在一个外层atomic事务块中，测试结束后自动回滚所有操作，避免污染测试数据库。
• 当事务块内的写入操作违反唯一约束抛出IntegrityError时，PostgreSQL会立即将当前事务标记为故障态。此状态下事务未回滚，所有后续SQL查询都会直接被数据库拒绝，直到事务执行回滚到最近的有效保存点，才能继续执行操作。
• 原有代码在同一个外层测试事务块中，捕获到唯一约束异常后没有触发回滚流程就直接执行接口查询，自然会触发事务状态错误。

具体修复步骤 #

1. 在会抛出IntegrityError的写入操作外层，嵌套一层独立的transaction.atomic块。这个块会自动设置事务保存点，当块内抛出约束异常时，会自动回滚到保存点，不会让外层测试事务进入故障态。
2. 调整异常校验逻辑，避免直接对比异常类的内存引用，改用实例判断提高兼容性。
3. 梳理测试数据流，避免重复入库导致的误触发约束。

修正后的完整测试代码 #

from django.test import TestCase
from django.urls import reverse
from django.db import transaction
from django.db.utils import IntegrityError
from rest_framework.test import APIClient

from .factories import ProductFactory
from .models import Product
from .serializer import ProductSerializer

PRODUCTS_URL = reverse('product-list')

def create_product(**params):
    """ Helper function to create a new product """
    return Product.objects.create(**params)

class PrivateProductsTests(TestCase):
    def setUp(self):
        self.client = APIClient()
    
    def test_an_error_occurs_when_forcing_the_same_slug_for_different_products(self):
        # 生成第一个合法商品的测试数据，直接入库获取合法slug
        product1_mock = ProductFactory.build(slug='')
        product1 = create_product(**ProductSerializer(product1_mock).data)

        # 生成和第一个商品同slug的重复测试数据
        product2_mock = ProductFactory.build(slug=product1.slug)
        product2_data = ProductSerializer(product2_mock).data

        # 在独立原子块内执行触发唯一约束的写入操作
        with self.assertRaises(IntegrityError) as raised:
            with transaction.atomic():
                create_product(**product2_data)

        # 校验异常类型，兼容不同环境下的异常类封装
        self.assertIsInstance(raised.exception, IntegrityError)

        # 事务已经回滚到安全状态，可以正常查询校验冗余数据
        product_count = Product.objects.filter(slug=product1.slug).count()
        self.assertEqual(product_count, 1)

        # 可以正常调用接口校验返回结果
        res = self.client.get(PRODUCTS_URL)
        self.assertEqual(len(res.data), 1)

常见踩坑提示 #

• 不要在assertRaises的回调形式下直接执行会抛IntegrityError的数据库操作，这种写法不会自动触发事务保存点回滚，依然会让外层测试事务进入故障态。
• 不要使用ProductFactory.create()生成测试数据后再调用create_product()二次入库：前者会直接将mock数据写入数据库，二次创建很可能提前触发唯一约束报错，建议用build()方法生成未入库的mock对象。
• 直接对比两个IntegrityError的类引用是否相等的逻辑可靠性很低：不同Python环境、Django版本下的数据库异常类引用可能存在差异，用isinstance做实例判断更稳定。

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