dc5d42addc
P0: - rpm_override 嵌入 Auth Cache Snapshot，消除每请求 DB 查询 (snapshot v6→v7) - 429 RPM 响应返回 Retry-After 头（当前分钟剩余秒数） P1: - ClearAll 按钮直连 DELETE API，带 loading 防重复 - 新增 GET /admin/users/:id/rpm-status 管理员 RPM 用量查询端点 优化: - checkRPM 从级联互斥改为并行取最严，user.rpm_limit 作为全局硬上限始终生效 - Override/Group 变更后自动失效 auth cache - fail-open 语义不变，Redis 故障不阻塞业务
55 lines
2.0 KiB
Go
55 lines
2.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBillingErrorDetails_MapsGroupRPMExceededToTooManyRequests(t *testing.T) {
|
|
status, code, msg, retryAfter := billingErrorDetails(service.ErrGroupRPMExceeded)
|
|
require.Equal(t, http.StatusTooManyRequests, status)
|
|
require.Equal(t, "rate_limit_exceeded", code)
|
|
require.NotEmpty(t, msg)
|
|
require.Greater(t, retryAfter, 0, "RPM exceeded should return positive Retry-After")
|
|
require.LessOrEqual(t, retryAfter, 60)
|
|
}
|
|
|
|
func TestBillingErrorDetails_MapsUserRPMExceededToTooManyRequests(t *testing.T) {
|
|
status, code, msg, retryAfter := billingErrorDetails(service.ErrUserRPMExceeded)
|
|
require.Equal(t, http.StatusTooManyRequests, status)
|
|
require.Equal(t, "rate_limit_exceeded", code)
|
|
require.NotEmpty(t, msg)
|
|
require.Greater(t, retryAfter, 0, "RPM exceeded should return positive Retry-After")
|
|
require.LessOrEqual(t, retryAfter, 60)
|
|
}
|
|
|
|
func TestBillingErrorDetails_APIKeyRateLimitStillMaps(t *testing.T) {
|
|
// 回归保护：加 RPM 分支后不应影响已有 APIKey rate limit 的映射。
|
|
for _, err := range []error{
|
|
service.ErrAPIKeyRateLimit5hExceeded,
|
|
service.ErrAPIKeyRateLimit1dExceeded,
|
|
service.ErrAPIKeyRateLimit7dExceeded,
|
|
} {
|
|
status, code, _, _ := billingErrorDetails(err)
|
|
require.Equal(t, http.StatusTooManyRequests, status, "status for %v", err)
|
|
require.Equal(t, "rate_limit_exceeded", code)
|
|
}
|
|
}
|
|
|
|
func TestBillingErrorDetails_BillingServiceUnavailableMapsTo503(t *testing.T) {
|
|
status, code, _, retryAfter := billingErrorDetails(service.ErrBillingServiceUnavailable)
|
|
require.Equal(t, http.StatusServiceUnavailable, status)
|
|
require.Equal(t, "billing_service_error", code)
|
|
require.Equal(t, 0, retryAfter, "non-RPM errors should not set Retry-After")
|
|
}
|
|
|
|
func TestBillingErrorDetails_UnknownErrorFallsBackTo403(t *testing.T) {
|
|
status, code, msg, _ := billingErrorDetails(service.ErrInsufficientBalance)
|
|
require.Equal(t, http.StatusForbidden, status)
|
|
require.Equal(t, "billing_error", code)
|
|
require.NotEmpty(t, msg)
|
|
}
|