fix: 系统性修复安全问题、性能问题和错误处理
安全问题修复: - X-Forwarded-For越界检查(auth.go) - checkTokenStatus Context参数传递(auth.go) - Type Assertion安全检查(auth.go) 性能问题修复: - TokenCache过期清理机制 - BruteForceProtection过期清理 - InMemoryIdempotencyStore过期清理 错误处理修复: - AuditStore.Emit返回error - domain层emitAudit辅助方法 - List方法返回空slice而非nil - 金额/价格负数验证 架构一致性: - 统一使用model.RoleHierarchyLevels 新增功能: - Alert API完整实现(CRUD+Resolve) - pkg/error错误码集中管理
This commit is contained in:
95
supply-api/pkg/error/errors_test.go
Normal file
95
supply-api/pkg/error/errors_test.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package error
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestNewCodeError(t *testing.T) {
|
||||
err := NewCodeError("TEST_4001", "test error message")
|
||||
assert.Equal(t, "TEST_4001", err.Code)
|
||||
assert.Equal(t, "test error message", err.Message)
|
||||
assert.Nil(t, err.Err)
|
||||
assert.Equal(t, "TEST_4001: test error message", err.Error())
|
||||
}
|
||||
|
||||
func TestWrapCodeError(t *testing.T) {
|
||||
originalErr := errors.New("original error")
|
||||
err := WrapCodeError(originalErr, "TEST_4001", "wrapped error")
|
||||
assert.Equal(t, "TEST_4001", err.Code)
|
||||
assert.Equal(t, "wrapped error", err.Message)
|
||||
assert.Equal(t, originalErr, err.Err)
|
||||
assert.Contains(t, err.Error(), "caused by: original error")
|
||||
}
|
||||
|
||||
func TestIsCodeError(t *testing.T) {
|
||||
codeErr := NewCodeError("TEST_4001", "test")
|
||||
assert.True(t, IsCodeError(codeErr))
|
||||
|
||||
stdErr := errors.New("standard error")
|
||||
assert.False(t, IsCodeError(stdErr))
|
||||
}
|
||||
|
||||
func TestGetErrorCode(t *testing.T) {
|
||||
codeErr := NewCodeError("SUP_ACC_4001", "test")
|
||||
assert.Equal(t, "SUP_ACC_4001", GetErrorCode(codeErr))
|
||||
|
||||
stdErr := errors.New("standard error")
|
||||
assert.Equal(t, "", GetErrorCode(stdErr))
|
||||
}
|
||||
|
||||
func TestUnwrap(t *testing.T) {
|
||||
originalErr := errors.New("original")
|
||||
wrapped := WrapCodeError(originalErr, "TEST_4001", "wrapped")
|
||||
|
||||
// 通过 Unwrap 获取原始错误
|
||||
unwrapped := wrapped.Unwrap()
|
||||
assert.Equal(t, originalErr, unwrapped)
|
||||
}
|
||||
|
||||
func TestValidateErrorCode(t *testing.T) {
|
||||
tests := []struct {
|
||||
code string
|
||||
expected bool
|
||||
}{
|
||||
{"SUP_ACC_4001", true},
|
||||
{"IAM_ROLE_4040", true},
|
||||
{"AUDIT_EVT_5000", true},
|
||||
{"REPO_NOT_FOUND", true},
|
||||
{"SYS_5000", true},
|
||||
{"INVALID", false}, // 没有下划线分隔
|
||||
{"BAD_CODE", false}, // 前缀不在白名单
|
||||
{"X_4001", false}, // 前缀不在白名单
|
||||
{"", false}, // 空字符串
|
||||
{"TOOLONG_4001", false}, // 前缀太长
|
||||
}
|
||||
|
||||
for _, tc := range tests {
|
||||
t.Run(tc.code, func(t *testing.T) {
|
||||
result := ValidateErrorCode(tc.code)
|
||||
assert.Equal(t, tc.expected, result, "code: %s", tc.code)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCommonErrors(t *testing.T) {
|
||||
assert.Equal(t, "SYS_4040", ErrNotFound.Code)
|
||||
assert.Equal(t, "resource not found", ErrNotFound.Message)
|
||||
|
||||
assert.Equal(t, "SYS_4000", ErrInvalidInput.Code)
|
||||
assert.Equal(t, "invalid input parameter", ErrInvalidInput.Message)
|
||||
|
||||
assert.Equal(t, "SYS_4010", ErrUnauthorized.Code)
|
||||
assert.Equal(t, "unauthorized", ErrUnauthorized.Message)
|
||||
|
||||
assert.Equal(t, "SYS_4030", ErrForbidden.Code)
|
||||
assert.Equal(t, "forbidden", ErrForbidden.Message)
|
||||
|
||||
assert.Equal(t, "SYS_5000", ErrInternalServer.Code)
|
||||
assert.Equal(t, "internal server error", ErrInternalServer.Message)
|
||||
|
||||
assert.Equal(t, "SYS_4090", ErrConcurrencyConflict.Code)
|
||||
assert.Equal(t, "concurrency conflict", ErrConcurrencyConflict.Message)
|
||||
}
|
||||
Reference in New Issue
Block a user