SakurasanandClaude
4846db9293
- ratelimit(内存计数): 密钥级每日请求数/Token 配额、用户级每秒速率 (OT_RATELIMIT_USER_RPS), 超限返回 429 - 网关 Auth 前置配额/限流检查, finishUsage 累计密钥 token 用量 - 前端 Skeleton 组件 + Dashboard/管理总览加载态 - Go 服务托管 web/dist 静态资源(SPA 回退), 单端口即可访问前后端 Co-Authored-By: Claude <noreply@anthropic.com>
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package ratelimit
|
|
|
|
import "testing"
|
|
|
|
func TestAllowRequestDaily(t *testing.T) {
|
|
l := New()
|
|
if !l.AllowRequestDaily(1, 2) {
|
|
t.Fatal("first request should be allowed")
|
|
}
|
|
if !l.AllowRequestDaily(1, 2) {
|
|
t.Fatal("second request should be allowed")
|
|
}
|
|
if l.AllowRequestDaily(1, 2) {
|
|
t.Fatal("third request should be blocked")
|
|
}
|
|
// 另一个 key 不受影响
|
|
if !l.AllowRequestDaily(2, 2) {
|
|
t.Fatal("other key should be allowed")
|
|
}
|
|
}
|
|
|
|
func TestTokensDaily(t *testing.T) {
|
|
l := New()
|
|
l.AddTokens(1, 100)
|
|
l.AddTokens(1, 50)
|
|
if got := l.TokensUsed(1); got != 150 {
|
|
t.Fatalf("tokens = %d, want 150", got)
|
|
}
|
|
if got := l.TokensUsed(2); got != 0 {
|
|
t.Fatalf("other key tokens = %d, want 0", got)
|
|
}
|
|
// 配额检查语义：已达 150 再设配额 150 应拒绝
|
|
l.AddTokens(3, 150)
|
|
if l.TokensUsed(3) >= 150 {
|
|
// 注意：>= 表示已达上限，Auth 层据此拒绝
|
|
}
|
|
}
|
|
|
|
func TestAllowUserRate(t *testing.T) {
|
|
l := New()
|
|
// limit=1：立即第二次应被拒绝
|
|
if !l.AllowUserRate(9, 1) {
|
|
t.Fatal("first should be allowed")
|
|
}
|
|
if l.AllowUserRate(9, 1) {
|
|
t.Fatal("second in same second should be blocked")
|
|
}
|
|
// limit<=0 不限制
|
|
if !l.AllowUserRate(9, 0) {
|
|
t.Fatal("limit 0 should always allow")
|
|
}
|
|
}
|