如何在Django+pytest测试中集成授权组？遇实现难题求助

核心问题梳理 #

你的代码存在以下关键错误：

• Group创建时未指定name字段，参数传递错误
• 授权用户Fixture未将用户关联到目标组
• 测试方法未正确注入Fixture依赖，无法访问预定义的组和用户
• has_post_permissions方法实现无效（递归调用无实际逻辑）
• 测试类方法缺少必要的self参数，不符合类实例方法规范

分步修正方案 #

1. 修正Group与授权用户Fixture #

import pytest
import factory
from django.contrib.auth import get_user_model
from posts.models import Post
from members.models import Member
from factories import MemberFactory, PostFactory
from faker import Faker
from django.contrib.auth.models import Group

User = get_user_model()
fake = Faker()

# Fixture：创建Contributor授权组
@pytest.fixture(scope="session")
def contributor_group(db):
    # 修复：创建Group必须显式指定name参数
    return Group.objects.create(name="Contributor")

# Fixture：创建已关联到Contributor组的授权用户
@pytest.fixture(scope="session")
def authorized_user(db, contributor_group):
    user = MemberFactory()
    # 将用户添加到目标组
    user.groups.add(contributor_group)
    return user

2. 修复MemberFactory中的权限判断逻辑 #

原工厂类的has_post_permissions是无效递归，替换为实际的组校验逻辑：

import factory
from faker import Faker
from django.contrib.auth import get_user_model
from posts.models import Post
from django.contrib.auth.models import Group

fake = Faker()
User = get_user_model()

class MemberFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = User

    username = factory.Faker('user_name')
    email = factory.Faker('email')
    # 优化：使用PostGenerationMethodCall确保密码正确加密
    password = factory.PostGenerationMethodCall('set_password', 'password')
    is_active = True
    is_superuser = False

    # 实现正确的权限判断逻辑
    def has_post_permissions(self):
        # 校验用户是否属于Contributor组
        return self.groups.filter(name="Contributor").exists()

class PostFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Post

    # 优化：使用factory内置Faker调用，避免初始化时生成固定值
    title = factory.Faker('text', max_nb_chars=150)
    author = factory.SubFactory(MemberFactory)
    content = factory.Faker('text')

3. 修正测试类方法 #

测试类方法需添加self参数，并正确注入Fixture依赖：

# Post权限测试类
class TestPosts:
    # 测试非Contributor用户无发布权限
    @pytest.mark.django_db
    def test_non_contributor_cannot_post(self):
        reg = MemberFactory()
        assert reg.has_post_permissions() is False

    # 测试Contributor用户拥有发布权限
    @pytest.mark.django_db
    def test_contributor_can_post(self, authorized_user):
        assert authorized_user.has_post_permissions() is True

说明：

• 测试方法必须包含self，符合类实例方法的要求
  通过参数直接注入authorized_user Fixture，无需手动处理组关联
• 方法名采用规范命名，提升可读性

额外注意事项 #

• 测试环境数据库独立于开发环境，后台创建的Group不会在测试中生效，必须通过Fixture/Factory创建
• 如果权限控制基于具体权限而非组，可修改has_post_permissions为校验用户权限：

  def has_post_permissions(self):
      return self.has_perm('posts.add_post')
  

  同时需在Contributor组中添加对应的posts.add_post权限。

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