Django 测试 - 内部错误：当前事务被中止，命令被忽略直到事务块结束

Question

在我的测试中，我不仅测试完美的情况，而且特别测试边缘情况和错误条件。所以我想确保一些唯一性约束起作用。

虽然我的测试和测试装置非常复杂，但我能够将问题追溯到以下示例，该示例不使用任何自定义模型。要重现该行为，只需将代码保存到tests.py 中并运行django 测试运行程序。

from django.contrib.auth.models import User
from django.db import IntegrityError
from django.test import TransactionTestCase

class TransProblemTest(TransactionTestCase):
    def test_uniqueness1(self):
        User.objects.create_user(username='user1', email='[email protected]', password='secret')
        self.assertRaises(IntegrityError, lambda :
            User.objects.create_user(username='user1', email='[email protected]', password='secret'))

    def test_uniqueness2(self):
        User.objects.create_user(username='user1', email='[email protected]', password='secret')
        self.assertRaises(IntegrityError, lambda :
            User.objects.create_user(username='user1', email='[email protected]', password='secret'))

具有单个测试方法的测试类可以工作，但如果有两个相同的方法实现，则会失败。 第一个测试抛出异常破坏了 Django 测试环境，并使以下所有测试失败。

我正在使用 Django 1.1 和 Ubuntu 10.04、Postgres 8.4 和 psycopg2。

Django 1.2 中还存在这个问题吗？

这是一个已知的错误还是我遗漏了什么？

Answer 1

Django 有两种类型的 TestCase：“plain”TestCase 和 TransactionTestCase。 文档有以下关于差异的说明他们两人之间：

TransactionTestCase 和 TestCase 完全相同，只是数据库重置为已知状态的方式以及测试代码测试提交和回滚效果的能力不同。 TransactionTestCase 在测试运行之前通过截断所有表并重新加载初始数据来重置数据库。 TransactionTestCase 可以调用提交和回滚，并观察这些调用对数据库的影响。

另一方面，TestCase 不会在测试开始时截断表并重新加载初始数据。相反，它将测试代码包含在数据库事务中，该事务在测试结束时回滚。

您正在使用 TransactionTestCase 来执行这些测试。切换到普通的TestCase，如果您维护现有的测试代码，您将看到问题消失。

为什么会出现这种情况？ TestCase 在事务块内执行测试方法。这意味着测试类中的每个测试方法将在单独的事务而不是同一事务中运行。当断言（或者更确切地说是内部的 lambda ）引发错误时，它会随事务一起终止。下一个测试方法是在新事务中执行的，因此您不会看到收到的错误。

但是，如果您要在相同测试方法中添加另一个相同的断言，您将再次看到错误：

class TransProblemTest(django.test.TestCase):
    def test_uniqueness1(self):
        User.objects.create_user(username='user1', email='[email protected]', password='secret')
        self.assertRaises(IntegrityError, lambda :
            User.objects.create_user(username='user1', email='[email protected]', password='secret'))
        # Repeat the test condition.
        self.assertRaises(IntegrityError, lambda :
            User.objects.create_user(username='user1', email='[email protected]', password='secret'))

这是因为第一个断言将被触发创建导致事务中止的错误。因此第二个无法执行。由于两个断言都发生在同一测试方法内，因此与之前的情况不同，尚未启动新事务。

希望这有帮助。

Answer 2

我假设当您说“单个测试方法有效”时，您的意思是它失败，引发异常，但不会破坏测试环境。

也就是说，您正在关闭 AutoCommit 的情况下运行。在该模式下，默认情况下共享数据库连接上的所有内容都是单个事务，出现故障时需要通过 ROLLBACK 中止事务，然后才能启动新事务。我建议如果可能的话，打开自动提交——除非您需要将多个写入操作包装到一个单元中，否则关闭它就太过分了。