Django测试中django.test.Client登录失败问题求助

问题现象 #

环境：Python 3.10.7、Django 4.1.1、Django REST Framework 3.13.1
继承django.test.TestCase的测试类中遇到两个核心问题：

1. 首次运行测试后，再次执行会因用户名唯一约束报错，尽管测试框架声称会销毁测试数据库
2. setUp方法中client.login()始终失败，改用force_login()触发DatabaseError，导致需要认证的视图测试因权限不足失败

项目在浏览器中运行正常，无认证视图和登录后受保护视图均工作正常，本地测试用SQLite3，无需认证的测试已通过。复现代码如下：

from django.contrib.auth.models import User
from django.test import Client, TestCase
from django.urls import reverse
from eventsadmin.models import Address


class AddressesViewTest(TestCase):
    username = "jrandomuser"
    password = "qwerty123"
    user = User.objects.filter(username=username).first()
    if user:
        print("User exists")
    else:
        user = User.objects.create(username=username)
        print("User created")
    user.set_password(password)
    user.save()
    client = Client()


    def setUp(self):
        if self.client.login(username=self.username, password=self.password):
            print("Login successful")
        else:
            print("Login failed")
        Address.objects.create(name="White House", address1="1600 Pennsylvania Ave", city="Washington", state="DC", postal_code="37188")


    def test_addresses(self):
        response = self.client.get(reverse("addresses-list"))
        self.assertContains(response, '"name":"White House"')

错误原因分析 #

1. 用户创建位置错误：类级代码会在测试类加载时执行，早于测试数据库的创建/销毁流程。第一次运行时创建的用户会写入测试数据库，但类级代码不会在每次测试方法运行时重新执行，且测试数据库的销毁逻辑不会清理类级变量引用的用户对象，导致后续运行时重复创建用户触发唯一约束报错。
2. 认证操作错误：
   • client实例在类级别初始化，而TestCase本身会自动创建self.client，手动创建的实例可能未正确绑定测试上下文。
   • login失败是因为类级创建的用户对象，密码设置与保存时机不对，且用户对象不在测试数据库的事务上下文内。
   • force_login报错是因为类级创建的用户对象不属于当前测试运行的数据库事务（测试用例每个方法都会开启独立事务），跨事务操作触发数据库错误。

修复方案 #

将用户创建、客户端初始化、认证逻辑全部移到setUp方法中，利用TestCase的事务隔离特性：

from django.contrib.auth.models import User
from django.test import TestCase
from django.urls import reverse
from eventsadmin.models import Address


class AddressesViewTest(TestCase):
    username = "jrandomuser"
    password = "qwerty123"

    def setUp(self):
        # 在测试方法的事务内创建用户，用create_user自动处理密码哈希
        self.user = User.objects.create_user(username=self.username, password=self.password)
        # 使用TestCase自带的self.client，无需手动实例化
        self.client.login(username=self.username, password=self.password)
        # 或者用force_login更可靠，跳过密码验证直接绑定用户
        # self.client.force_login(self.user)
        # 创建测试数据
        Address.objects.create(name="White House", address1="1600 Pennsylvania Ave", city="Washington", state="DC", postal_code="37188")

    def test_addresses(self):
        response = self.client.get(reverse("addresses-list"))
        self.assertContains(response, '"name":"White House"')

关键说明 #

• 使用create_user方法：Django内置方法会自动处理密码哈希，比手动set_password更简洁可靠。
• 依赖TestCase自带的self.client：TestCase会为每个测试方法初始化全新的client实例，确保上下文隔离。
• force_login更适合测试场景：无需验证密码，直接将用户绑定到客户端，避免密码哈希相关的潜在问题。
• setUp方法会在每个测试方法执行前运行，且每个测试方法都在独立的数据库事务中，运行后事务回滚，不会残留测试数据，彻底解决重复创建用户的问题。

内容的提问来源于stack exchange，提问作者Don R