huang
eaa70ac255
主要功能: - 实现完整的 RBAC 权限系统（账号、角色、权限的多对多关联） - 基于 owner_id + shop_id 的自动数据权限过滤 - 使用 PostgreSQL WITH RECURSIVE 查询下级账号 - Redis 缓存优化下级账号查询性能（30分钟过期） - 支持多租户数据隔离和层级权限管理 技术实现: - 新增 Account、Role、Permission 模型及关联关系表 - 实现 GORM Scopes 自动应用数据权限过滤 - 添加数据库迁移脚本（000002_rbac_data_permission、000003_add_owner_id_shop_id） - 完善错误码定义（1010-1027 为 RBAC 相关错误） - 重构 main.go 采用函数拆分提高可读性 测试覆盖: - 添加 Account、Role、Permission 的集成测试 - 添加数据权限过滤的单元测试和集成测试 - 添加下级账号查询和缓存的单元测试 - 添加 API 回归测试确保向后兼容 文档更新: - 更新 README.md 添加 RBAC 功能说明 - 更新 CLAUDE.md 添加技术栈和开发原则 - 添加 docs/004-rbac-data-permission/ 功能总结和使用指南 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"io"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/requestid"
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
"github.com/break/junhong_cmp_fiber/pkg/logger"
|
|
)
|
|
|
|
// TestRecover_PanicCapture 测试 panic 捕获功能
|
|
func TestRecover_PanicCapture(t *testing.T) {
|
|
// 初始化日志器
|
|
_ = logger.InitLoggers(
|
|
"debug",
|
|
true,
|
|
logger.LogRotationConfig{
|
|
Filename: "../../tests/integration/logs/recover_test.log",
|
|
MaxSize: 10,
|
|
MaxBackups: 3,
|
|
MaxAge: 7,
|
|
Compress: false,
|
|
},
|
|
logger.LogRotationConfig{
|
|
Filename: "../../tests/integration/logs/access_test.log",
|
|
MaxSize: 10,
|
|
MaxBackups: 3,
|
|
MaxAge: 7,
|
|
Compress: false,
|
|
},
|
|
)
|
|
|
|
appLogger := logger.GetAppLogger()
|
|
|
|
app := fiber.New(fiber.Config{
|
|
ErrorHandler: errors.SafeErrorHandler(appLogger),
|
|
})
|
|
|
|
// 注册 recover 中间件
|
|
app.Use(Recover(appLogger))
|
|
app.Use(requestid.New(requestid.Config{
|
|
Generator: func() string {
|
|
return uuid.NewString()
|
|
},
|
|
}))
|
|
|
|
// 创建会触发 panic 的路由
|
|
app.Get("/panic", func(c *fiber.Ctx) error {
|
|
panic("测试 panic")
|
|
})
|
|
|
|
// 发起请求
|
|
req := httptest.NewRequest("GET", "/panic", nil)
|
|
resp, err := app.Test(req, -1)
|
|
require.NoError(t, err)
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
// 验证响应状态码为 500 (内部错误)
|
|
assert.Equal(t, 500, resp.StatusCode, "panic 应转换为 500 错误")
|
|
|
|
// 验证响应体不为空
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, body, "panic 响应体不应为空")
|
|
|
|
t.Log("✓ Panic 捕获测试通过")
|
|
}
|
|
|
|
// TestRecover_NilPointerPanic 测试空指针 panic
|
|
func TestRecover_NilPointerPanic(t *testing.T) {
|
|
appLogger := logger.GetAppLogger()
|
|
|
|
app := fiber.New(fiber.Config{
|
|
ErrorHandler: errors.SafeErrorHandler(appLogger),
|
|
})
|
|
|
|
app.Use(Recover(appLogger))
|
|
app.Use(requestid.New(requestid.Config{
|
|
Generator: func() string {
|
|
return uuid.NewString()
|
|
},
|
|
}))
|
|
|
|
// 创建会触发空指针 panic 的路由
|
|
app.Get("/nil-panic", func(c *fiber.Ctx) error {
|
|
var ptr *string
|
|
_ = *ptr // 空指针引用会导致 panic
|
|
return nil
|
|
})
|
|
|
|
req := httptest.NewRequest("GET", "/nil-panic", nil)
|
|
resp, err := app.Test(req, -1)
|
|
require.NoError(t, err)
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
assert.Equal(t, 500, resp.StatusCode, "空指针 panic 应转换为 500 错误")
|
|
|
|
t.Log("✓ 空指针 Panic 捕获测试通过")
|
|
}
|
|
|
|
// TestRecover_NormalRequest 测试正常请求不受影响
|
|
func TestRecover_NormalRequest(t *testing.T) {
|
|
appLogger := logger.GetAppLogger()
|
|
|
|
app := fiber.New(fiber.Config{
|
|
ErrorHandler: errors.SafeErrorHandler(appLogger),
|
|
})
|
|
|
|
app.Use(Recover(appLogger))
|
|
|
|
// 创建正常的路由
|
|
app.Get("/normal", func(c *fiber.Ctx) error {
|
|
return c.JSON(fiber.Map{"status": "ok"})
|
|
})
|
|
|
|
req := httptest.NewRequest("GET", "/normal", nil)
|
|
resp, err := app.Test(req, -1)
|
|
require.NoError(t, err)
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
assert.Equal(t, 200, resp.StatusCode, "正常请求应返回 200")
|
|
|
|
t.Log("✓ 正常请求测试通过")
|
|
}
|