Django(PostgreSQL)用户组测试Setup与Teardown异常问题求助

问题根源 #

你手动硬编码了组的ID值（SALES_MANAGER = 1、TEAM = 2），但Django中get_or_create创建的组实例ID并非固定不变：

• 第一次测试时，数据库新建组的ID确实是1和2，所以user.groups.add(TEAM)能正常关联
• 第二次测试时，即便删除了旧组，新创建的组ID会自增（比如变成3、4），此时硬编码的TEAM=2对应的组不存在，调用add不会报错但实际没关联任何组，最终用户的组集合为空。

修复方案 #

不要硬编码ID，直接使用get_or_create返回的组实例或其动态ID：

方案1：使用组实例的动态ID

# Setup环节修改为
sales_manager_group, created = Group.objects.get_or_create(name='Sales Manager')
SALES_MANAGER = sales_manager_group.id  # 取实际创建的组ID
team_group, created = Group.objects.get_or_create(name='Team')
TEAM = team_group.id

# 为用户添加组时保持原有逻辑
user.groups.add(TEAM)

方案2：直接使用组实例（更推荐）

# Setup环节
sales_manager_group, created = Group.objects.get_or_create(name='Sales Manager')
team_group, created = Group.objects.get_or_create(name='Team')

# 为用户添加组时直接传实例
user.groups.add(team_group)

额外建议 #

如果使用Django的TestCase类，测试结束后数据库会自动回滚，不需要在Teardown环节手动删除组，这能避免不必要的数据库操作干扰测试流程。

内容的提问来源于stack exchange，提问作者Anthony Petrillo