你不能执行查询，直到'原子'block"而只在单元测试期间使用信号

小开

最佳答案

我自己也遇到了同样的问题。这是由于新版本的Django中处理事务的方式有一个怪癖，再加上有意触发异常的unittest。

我有一个单元测试，检查确保一个唯一的列约束被强制触发一个IntegrityError异常:

def test_constraint(self):
try:
# Duplicates should be prevented.
models.Question.objects.create(domain=self.domain, slug='barks')
self.fail('Duplicate question allowed.')
except IntegrityError:
pass


do_more_model_stuff()

在Django 1.4中，这工作得很好。然而，在Django 1.5/1.6中，每个测试都被包装在一个事务中，所以如果发生异常，它会中断事务，直到你显式地回滚它。因此，该事务中的任何进一步的ORM操作，例如我的do_more_model_stuff()，将失败并出现django.db.transaction.TransactionManagementError异常。

就像评论中提到的caio一样，解决方案是使用transaction.atomic捕获异常，如下所示:

from django.db import transaction
def test_constraint(self):
try:
# Duplicates should be prevented.
with transaction.atomic():
models.Question.objects.create(domain=self.domain, slug='barks')
self.fail('Duplicate question allowed.')
except IntegrityError:
pass

这将防止故意抛出的异常破坏整个unittest的事务。

小开

由于@mkoistinen从未做出他的评论的答案，我将发布他的建议，这样人们就不必在评论中挖掘。

考虑将测试类声明为TransactionTestCase而不是TestCase。

从Django文档: A TransactionTestCase可以调用commit和rollback，并观察这些调用对数据库的影响。

小开

我也有同样的问题。

在《我的情况》中，我就是这么做的

author.tasks.add(tasks)

那么把它转换成

author.tasks.add(*tasks)

删除了这个错误。

小开

我在使用django 1.9.7的create_test_data函数运行单元测试时得到了这个错误。它可以在早期版本的django中运行。

它是这样的:

cls.localauth,_ = Organisation.objects.get_or_create(organisation_type=cls.orgtypeLA, name='LA for test', email_general='test@test.com', address='test', postcode='test', telephone='test')
cls.chamber,_ = Organisation.objects.get_or_create(organisation_type=cls.orgtypeC, name='chamber for test', email_general='test@test.com', address='test', postcode='test', telephone='test')
cls.lawfirm,_ = Organisation.objects.get_or_create(organisation_type=cls.orgtypeL, name='lawfirm for test', email_general='test@test.com', address='test', postcode='test', telephone='test')


cls.chamber.active = True
cls.chamber.save()


cls.localauth.active = True
cls.localauth.save()    <---- error here


cls.lawfirm.active = True
cls.lawfirm.save()

我的解决方案是使用update_or_create代替:

cls.localauth,_ = Organisation.objects.update_or_create(organisation_type=cls.orgtypeLA, name='LA for test', email_general='test@test.com', address='test', postcode='test', telephone='test', defaults={'active': True})
cls.chamber,_ = Organisation.objects.update_or_create(organisation_type=cls.orgtypeC, name='chamber for test', email_general='test@test.com', address='test', postcode='test', telephone='test', defaults={'active': True})
cls.lawfirm,_ = Organisation.objects.update_or_create(organisation_type=cls.orgtypeL, name='lawfirm for test', email_general='test@test.com', address='test', postcode='test', telephone='test', defaults={'active': True})

小开

我有同样的问题，但with transaction.atomic()和TransactionTestCase不适合我。

python manage.py test -r代替python manage.py test对我来说是可以的，也许执行的顺序是至关重要的

然后我找到了一个关于测试执行的顺序的文档，它提到了哪个测试将首先运行。

所以，我使用TestCase数据库交互，unittest.TestCase为其他简单的测试，它现在工作!

小开

对我来说，提议的修复没有起作用。在我的测试中，我用Popen打开了一些子进程来分析/lint迁移(例如，一个测试检查是否没有模型更改)。

对我来说，从SimpleTestCase而不是TestCase继承子类确实奏效了。

注意SimpleTestCase不允许使用数据库。

虽然这并没有回答最初的问题，但我希望这能帮助到一些人。

小开

@kdazzle的答案是正确的我没有尝试它，因为人们说“Django的TestCase类是TransactionTestCase的一个更常用的子类”，所以我认为这是一个或另一个相同的用途。但是Jahongir Rahmonov博客解释得更好:

TestCase类将测试包装在两个嵌套的原子()块中: 全班一份，每个测试一份。这就是应该使用TransactionTestCase。它不会用原子()块，因此您可以测试您的特殊方法，需要

编辑:它没有工作，我想是的，但不是。

在4年可以固定 .......................................

小开

如果使用pytest-django，你可以将transaction=True传递给django_db装饰器来避免这个错误。

看到https://pytest-django.readthedocs.io/en/latest/database.html#testing-transactions

Django本身有TransactionTestCase，它允许你进行测试事务，并将在测试之间刷新数据库以进行隔离他们。这样做的缺点是，由于需要刷新数据库，这些测试的设置要慢得多。Pytest-django也支持这种风格的测试，你可以使用django_db标记的参数来选择:

@pytest.mark.django_db(transaction=True)
def test_spam():
pass  # test relying on transactions

小开

下面是另一种方法，基于这个问题的答案:

with transaction.atomic():
self.assertRaises(IntegrityError, models.Question.objects.create, **{'domain':self.domain, 'slug':'barks'})

小开

def test_wrong_user_country_db_constraint(self):
"""
Check whether or not DB constraint doesnt allow to save wrong country code in DB.
"""
self.test_user_data['user_country'] = 'XX'
expected_constraint_name = "country_code_within_list_of_countries_check"


with transaction.atomic():
with self.assertRaisesRegex(IntegrityError, expected_constraint_name) as cm:
get_user_model().objects.create_user(**self.test_user_data)


self.assertFalse(
get_user_model().objects.filter(email=self.test_user_data['email']).exists()
)

with transaction.atomic() seems do the job correct

小开

在我的情况下，它是造成的，但没有调用super().tearDownClass()

class TnsFileViewSetTestCase(APITestCase):
@classmethod
def tearDownClass(self):
super().tearDownClass()    # without this line we will get TransactionManagementError
for tnsfile in TnsFile.objects.all():
tnsfile.file.delete()