feat(rbac): 落地 Wagtail Group + GroupPagePermission/GroupCollectionPermission/Snippet 权限体系

新增 setup_rbac_groups 幂等管理命令，按设计方案 §2.8 矩阵创建 5 个 Group（站点管理员/编辑/审核员/作者/查看者），配置页面/图片文档/Snippet/用户管理权限；Author 仅编辑自己页面利用 Wagtail 内置 OwnershipPermissionPolicy 实现，无需自定义 hook；新增 7 个单元测试，pytest 33→40 全部通过。
2026-08-07 17:07:39 +08:00
parent 3944e11330
commit f37571f395
3 changed files with 360 additions and 1 deletions
@@ -1,5 +1,6 @@
 """apps.core 基础单元测试：SEOablePage 抽象基类字段与公共 Block。"""
 import pytest
+from django.contrib.auth import get_user_model
 from rest_framework.test import APIClient

 from apps.core.blocks import FormBlock, TeamMemberChooserBlock
@@ -215,3 +216,150 @@ def test_site_settings_api_returns_defaults():

    assert response.status_code == 200
    assert response.data["icp_url"] == "https://beian.miit.gov.cn/"
+
+
+# ---------------------------------------------------------------------------
+# RBAC 权限组（setup_rbac_groups 管理命令）
+# ---------------------------------------------------------------------------
+
+
+def test_setup_rbac_groups_creates_all_expected_groups():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+
+    call_command("setup_rbac_groups")
+
+    expected_names = {"站点管理员", "编辑", "审核员", "作者", "查看者"}
+    assert expected_names <= set(Group.objects.values_list("name", flat=True))
+
+
+def test_setup_rbac_groups_grants_access_admin_to_all_groups():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+
+    call_command("setup_rbac_groups")
+
+    for name in ["站点管理员", "编辑", "审核员", "作者", "查看者"]:
+        group = Group.objects.get(name=name)
+        assert group.permissions.filter(codename="access_admin").exists()
+
+
+def test_setup_rbac_groups_page_permissions_match_role_matrix():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+    from wagtail.models import GroupPagePermission, Page
+
+    call_command("setup_rbac_groups")
+
+    root_page = Page.objects.get(depth=1)
+
+    def page_codenames(group_name):
+        group = Group.objects.get(name=group_name)
+        return set(
+            GroupPagePermission.objects.filter(group=group, page=root_page).values_list(
+                "permission__codename", flat=True
+            )
+        )
+
+    # 站点管理员拥有全部页面权限
+    assert page_codenames("站点管理员") == {
+        "add_page", "change_page", "publish_page", "delete_page",
+        "bulk_delete_page", "lock_page", "unlock_page", "view_page",
+    }
+    # 编辑可以创建/编辑但不能发布
+    editor_perms = page_codenames("编辑")
+    assert editor_perms == {"add_page", "change_page", "view_page"}
+    assert "publish_page" not in editor_perms
+    # 审核员可以编辑并发布（用于审批发布场景）
+    assert page_codenames("审核员") == {"change_page", "publish_page", "view_page"}
+    # 作者只有 add + view：只能创建页面，仅能编辑自己拥有的页面（Wagtail 内置 ownership 策略）
+    assert page_codenames("作者") == {"add_page", "view_page"}
+    assert "change_page" not in page_codenames("作者")
+    # 查看者仅能查看
+    assert page_codenames("查看者") == {"view_page"}
+
+
+def test_setup_rbac_groups_author_can_only_edit_own_pages(home_page):
+    """验证“Author 仅能编辑自己的页面”这一矩阵要求，由 Wagtail 内置的
+    OwnershipPermissionPolicy（仅 add 权限）自然实现，无需自定义 hook。"""
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+
+    call_command("setup_rbac_groups")
+
+    User = get_user_model()
+    author = User.objects.create_user(username="author-user", password="pass1234")
+    other_author = User.objects.create_user(username="other-author", password="pass1234")
+    author.groups.add(Group.objects.get(name="作者"))
+    other_author.groups.add(Group.objects.get(name="作者"))
+
+    own_page = SimpleContentPage(title="自己的页面", slug="own-page", owner=author)
+    home_page.add_child(instance=own_page)
+
+    others_page = SimpleContentPage(title="他人的页面", slug="others-page", owner=other_author)
+    home_page.add_child(instance=others_page)
+
+    assert own_page.permissions_for_user(author).can_edit() is True
+    assert others_page.permissions_for_user(author).can_edit() is False
+
+
+def test_setup_rbac_groups_snippet_permissions_restrict_lead_to_admin_and_reviewer():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+
+    call_command("setup_rbac_groups")
+
+    def lead_codenames(group_name):
+        group = Group.objects.get(name=group_name)
+        return set(
+            group.permissions.filter(content_type__app_label="forms", codename__endswith="lead")
+            .values_list("codename", flat=True)
+        )
+
+    assert lead_codenames("站点管理员") == {"add_lead", "change_lead", "delete_lead", "view_lead"}
+    assert lead_codenames("审核员") == {"view_lead"}
+    assert lead_codenames("编辑") == set()
+    assert lead_codenames("作者") == set()
+    assert lead_codenames("查看者") == set()
+
+
+def test_setup_rbac_groups_only_site_admin_has_user_management_permissions():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+
+    call_command("setup_rbac_groups")
+
+    User = get_user_model()
+    user_ct_app_label = User._meta.app_label
+
+    for name in ["编辑", "审核员", "作者", "查看者"]:
+        group = Group.objects.get(name=name)
+        assert not group.permissions.filter(
+            content_type__app_label=user_ct_app_label, content_type__model="user"
+        ).exists()
+
+    site_admin = Group.objects.get(name="站点管理员")
+    site_admin_codenames = set(
+        site_admin.permissions.filter(
+            content_type__app_label=user_ct_app_label, content_type__model="user"
+        ).values_list("codename", flat=True)
+    )
+    assert site_admin_codenames == {"add_user", "change_user", "delete_user", "view_user"}
+
+
+def test_setup_rbac_groups_is_idempotent():
+    from django.contrib.auth.models import Group
+    from django.core.management import call_command
+    from wagtail.models import GroupPagePermission, Page
+
+    call_command("setup_rbac_groups")
+    call_command("setup_rbac_groups")
+
+    assert Group.objects.filter(name="站点管理员").count() == 1
+    group = Group.objects.get(name="站点管理员")
+    root_page = Page.objects.get(depth=1)
+    # 重复运行不应产生重复的页面权限记录
+    assert GroupPagePermission.objects.filter(
+        group=group, page=root_page, permission__codename="add_page"
+    ).count() == 1
+    assert group.permissions.filter(codename="access_admin").count() == 1