Question

Django新手在这里。我正在尝试为我开发的简单API实现单元测试。下面你可以找到我的测试实现工作正常：

from django.test import TestCase
from my_app.models import MyModel


class TestMyViewSet(TestCase):
    """
    Unit tests for endpoints in MyViewSet.
    """
    fixtures = ['my_app/fixtures/data.yaml']

    def setUp(self):
        # Making setup for the test case here.


    def test_post_my_resource(self):

        # Checking that fixture is loaded correctly.
        self.assertEqual(MyModel.objects.all().count(),1)

        request_body = {
            'my_resource': "input_for_my_resource"
        }

        response = self.client.post('/my_resources/', data=request_body)
        self.assertEqual(response.status_code, 201)
        # self.assertEqual(MyModel.objects.all().count(),2)

但是当我从注释中删除最后一行self.assertEqual(MyModel.objects.all().count(),2)以通过检查实例数来测试在相应模型上成功创建my_resource时，我收到一条错误，说明以下内容：

TransactionManagementError：当前事务中发生错误。在“原子”块结束之前，您无法执行查询。

我在这里缺少什么？

提前致谢！

PS：我遇到了以下问题：TransactionManagementError “You can't execute queries until the end of the 'atomic' block” while using signals, but only during Unit Testing但我不确定在我的情况下发生的情况是一样的。

Answer 1

显然，从django.test.TestCase转移到django.test.TransactionTestCase解决了这个问题。以下是关于django.test.TestCase和django.test.TransactionTestCase之间差异的一些要点：

  除了数据库重置为已知状态的方式以及测试代码测试提交和回滚效果的能力之外，
TransactionTestCase和TestCase是相同的：



TransactionTestCase在测试运行后通过截断所有表来重置数据库。 TransactionTestCase可以调用commit和rollback，并观察这些调用对数据库的影响。



另一方面，TestCase在测试后不截断表。相反，它将测试代码包含在数据库事务中，该事务在测试结束时回滚。这可以保证测试结束时的回滚将数据库恢复到其初始状态。

您可以在此处找到文档TransactionTestCase

的更多详细信息

Django测试：TransactionManagementError：在“原子”块结束之前，您无法执行查询

1 个答案: