在 django 单元测试中使用用户模型的问题

Question

我有以下 django 测试用例，它给了我错误：

class MyTesting(unittest.TestCase):
    def setUp(self):
        self.u1 = User.objects.create(username='user1')
        self.up1 = UserProfile.objects.create(user=self.u1)

    def testA(self):
        ...

    def testB(self):
        ...

当我运行测试时， testA 将成功通过，但在 testB 启动之前，我收到以下错误：

IntegrityError: column username is not unique

很明显它试图在每个测试用例之前创建 self.u1 并发现它已经存在于数据库中。如何在每个测试用例之后正确清理它，以便后续用例正确运行？

Answer 1

setUp 和 tearDown 调用 Unittest 上的方法每个测试用例之前和之后。定义 tearDown 方法删除创建的用户。

class MyTesting(unittest.TestCase):
    def setUp(self):
        self.u1 = User.objects.create(username='user1')
        self.up1 = UserProfile.objects.create(user=self.u1)

    def testA(self):
        ...

    def tearDown(self):
        self.up1.delete()
        self.u1.delete()

我还建议创建用户配置文件 使用 post_save 信号，除非您确实想为每个用户手动创建用户配置文件。

删除评论的后续：

来自 Django 文档：

当 Django 删除一个对象时，它
模拟 SQL 的行为
ON DELETE CASCADE 约束 -- 在
换句话说，任何具有
指向对象的外键
被删除将同时被删除
它。

在您的情况下，用户配置文件指向用户，因此您应该先删除用户，同时删除配置文件。

Answer 2

如果您希望 django 在每次测试运行后自动刷新测试数据库，那么您应该扩展 django.test.TestCase，而不是 django.utils.unittest.TestCase （因为您目前正在做）。

最好在每次测试后转储数据库，这样您就可以额外确保测试是一致的，但请注意，由于这种额外的开销，您的测试将运行得更慢。

请参阅“编写测试”Django 文档中的警告部分。

Answer 3

准确地说，setUp 的存在就是为了在每个测试用例之前运行一次。

相反的方法，即在每个测试用例之后运行一次的方法，名为tearDown：这是您删除self.u1等的地方（大概是通过只需调用 self.u1.delete()，除非除了删除对象之外还有补充的专门清理要求）。