Django TestCase 在测试完成后不破坏测试数据 运行

Django TestCase not destroying test data after tests finish running

我目前正在尝试为应用程序编写我的第一个 Django 规范。但是,每次我 运行 测试都会收到 IntegrityError: UNIQUE constraint failed: auth_user.username,因为我在之前的 运行 测试中创建的用户现在存在于数据库中。

我的印象是 TestCase 将测试代码包含在数据库事务中,以便在测试期间创建的所有数据在完成时回滚?我在这里做错了什么?

from django.test import TestCase

from django.contrib.auth.models import User
from basic.models import SavedModels

# Create your tests here.
class SavedModelsTestCase(TestCase):
    User.objects.create(username="user_001",
                        password="useruser",
                        first_name="Test",
                        last_name="User",
                        email="test_user@test.com"
                        )

    user = User.objects.filter(username="user_001")
    saved_model = SavedModels.objects.create(name="Test Model",
                                             description="A model",
                                             user=user
                                            )

    def test_saved_models(self):
        saved_model = create_saved_model()
        self.assertTrue(isinstance(saved_model, SavedModels))
        self.assertEqual(saved_model.name, "Test Model")
        self.assertEqual(saved_model.description, "A model")
        self.assertEqual(saved_model.user, user)

它不是那样工作的,你需要两个名为 setup 的函数来设置所需的数据(它在测试开始时运行)你需要在这里创建你的用户和 tearDown(它在测试结束时运行) 在 运行 之后进行清理 添加这两个测试后你的代码将变成这样

class SavedModelsTestCase(TestCase):

   def setUp(self):
       #create your user here
    def tearDown(self):
       # for any clenaup required none in your case

    def test_saved_models(self):
        saved_model = create_saved_model()
        self.assertTrue(isinstance(saved_model, SavedModels))
        self.assertEqual(saved_model.name, "Test Model")
        self.assertEqual(saved_model.description, "A model")
        self.assertEqual(saved_model.user, user)