huang
1b9080e3ab
本次提交完成了角色权限体系的重构，主要包括： 1. 数据库迁移 - 添加 tb_permission.platform 字段（all/web/h5） - 更新 tb_role.role_type 注释（1=平台角色，2=客户角色） 2. GORM 模型更新 - Permission 模型添加 Platform 字段 - Role 模型更新 RoleType 注释 3. 常量定义 - 新增角色类型常量（RoleTypePlatform, RoleTypeCustomer） - 新增权限端口常量（PlatformAll, PlatformWeb, PlatformH5） - 添加角色类型与用户类型匹配规则函数 4. Store 层实现 - Permission Store 支持按 platform 过滤 - Account Role Store 添加 CountByAccountID 方法 5. Service 层实现 - 角色分配支持类型匹配校验 - 角色分配支持数量限制（超级管理员0个，平台用户无限制，代理/企业1个） - Permission Service 支持 platform 过滤 6. 权限校验中间件 - 实现 RequirePermission、RequireAnyPermission、RequireAllPermissions - 支持 platform 字段过滤 - 支持跳过超级管理员检查 7. 测试用例 - 角色类型匹配规则单元测试 - 角色分配数量限制单元测试 - 权限 platform 过滤单元测试 - 权限校验中间件集成测试（占位） 8. 代码清理 - 删除过时的 subordinate 测试文件 - 移除 Account.ParentID 相关引用 - 更新 DTO 验证规则 Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
131 lines
4.4 KiB
Go
131 lines
4.4 KiB
Go
package integration
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
|
||
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
||
)
|
||
|
||
// MockPermissionChecker 模拟权限检查器
|
||
type MockPermissionChecker struct {
|
||
permissions map[uint]map[string]bool // userID -> permCode -> hasPermission
|
||
}
|
||
|
||
func NewMockPermissionChecker() *MockPermissionChecker {
|
||
return &MockPermissionChecker{
|
||
permissions: make(map[uint]map[string]bool),
|
||
}
|
||
}
|
||
|
||
func (m *MockPermissionChecker) GrantPermission(userID uint, permCode string) {
|
||
if m.permissions[userID] == nil {
|
||
m.permissions[userID] = make(map[string]bool)
|
||
}
|
||
m.permissions[userID][permCode] = true
|
||
}
|
||
|
||
func (m *MockPermissionChecker) CheckPermission(ctx context.Context, userID uint, permCode string, platform string) (bool, error) {
|
||
if m.permissions[userID] == nil {
|
||
return false, nil
|
||
}
|
||
return m.permissions[userID][permCode], nil
|
||
}
|
||
|
||
// TestPermissionMiddleware_RequirePermission 测试权限校验中间件（单个权限）
|
||
// TODO: 完整实现需要启动 Fiber 应用并模拟 HTTP 请求
|
||
func TestPermissionMiddleware_RequirePermission(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
|
||
// 占位测试：验证 PermissionChecker 接口可以被 mock
|
||
checker := NewMockPermissionChecker()
|
||
checker.GrantPermission(1, "user:read")
|
||
|
||
ctx := context.Background()
|
||
hasPermission, err := checker.CheckPermission(ctx, 1, "user:read", constants.PlatformAll)
|
||
assert.NoError(t, err)
|
||
assert.True(t, hasPermission)
|
||
|
||
hasPermission, err = checker.CheckPermission(ctx, 1, "user:write", constants.PlatformAll)
|
||
assert.NoError(t, err)
|
||
assert.False(t, hasPermission)
|
||
}
|
||
|
||
// TestPermissionMiddleware_RequireAnyPermission 测试权限校验中间件（多个权限任一）
|
||
func TestPermissionMiddleware_RequireAnyPermission(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
}
|
||
|
||
// TestPermissionMiddleware_RequireAllPermissions 测试权限校验中间件（多个权限全部）
|
||
func TestPermissionMiddleware_RequireAllPermissions(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
}
|
||
|
||
// TestPermissionMiddleware_SkipSuperAdmin 测试超级管理员跳过权限检查
|
||
func TestPermissionMiddleware_SkipSuperAdmin(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
}
|
||
|
||
// TestPermissionMiddleware_PlatformFiltering 测试按 platform 过滤权限
|
||
func TestPermissionMiddleware_PlatformFiltering(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
|
||
// 测试场景：
|
||
// 1. Web 端请求需要 Web 权限
|
||
// 2. H5 端请求需要 H5 权限
|
||
// 3. all 权限在所有端口都有效
|
||
}
|
||
|
||
// TestPermissionMiddleware_Unauthorized 测试未认证用户访问受保护路由
|
||
func TestPermissionMiddleware_Unauthorized(t *testing.T) {
|
||
t.Skip("TODO: 需要完整的 Fiber 集成测试环境")
|
||
}
|
||
|
||
// 集成测试实现指南:
|
||
//
|
||
// 完整的集成测试应该：
|
||
// 1. 启动 Fiber 应用
|
||
// 2. 注册受权限保护的路由：
|
||
// - 使用 middleware.RequirePermission("user:read", config)
|
||
// - 使用 middleware.RequireAnyPermission([]string{"user:read", "user:write"}, config)
|
||
// - 使用 middleware.RequireAllPermissions([]string{"user:read", "user:write"}, config)
|
||
// 3. 模拟不同用户的 HTTP 请求
|
||
// 4. 验证权限检查结果（200 OK 或 403 Forbidden）
|
||
//
|
||
// 示例代码结构：
|
||
//
|
||
// func TestPermissionMiddleware_Integration(t *testing.T) {
|
||
// // 1. 初始化数据库和 Redis
|
||
// db, redisClient := testutils.SetupTestDB(t)
|
||
// defer testutils.TeardownTestDB(t, db, redisClient)
|
||
//
|
||
// // 2. 创建测试数据（用户、角色、权限）
|
||
// // ...
|
||
//
|
||
// // 3. 初始化 Service 和 Middleware
|
||
// permissionService := permission.New(permissionStore)
|
||
// config := middleware.PermissionConfig{
|
||
// PermissionChecker: permissionService,
|
||
// Platform: constants.PlatformWeb,
|
||
// SkipSuperAdmin: true,
|
||
// }
|
||
//
|
||
// // 4. 创建 Fiber 应用并注册路由
|
||
// app := fiber.New()
|
||
// app.Get("/protected",
|
||
// middleware.RequirePermission("user:read", config),
|
||
// func(c *fiber.Ctx) error {
|
||
// return c.JSON(fiber.Map{"message": "success"})
|
||
// },
|
||
// )
|
||
//
|
||
// // 5. 模拟请求并验证响应
|
||
// req := httptest.NewRequest("GET", "/protected", nil)
|
||
// // 设置认证信息...
|
||
// resp, err := app.Test(req)
|
||
// require.NoError(t, err)
|
||
// assert.Equal(t, fiber.StatusOK, resp.StatusCode)
|
||
// }
|