chenhg5andClaude Opus 4.6
e5d328d9d6
Streaming Preview: - Real-time message preview during agent response (like "typing" indicator) - Supported on Telegram, Discord, and Feishu - Configurable interval, min delta chars, and max preview length Rate Limiting: - Per-session sliding-window rate limiter - Prevents users from flooding the bot with messages - Configurable max messages and window size Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package core
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestRateLimiter_AllowWithinLimit(t *testing.T) {
|
|
rl := NewRateLimiter(5, time.Minute)
|
|
for i := 0; i < 5; i++ {
|
|
if !rl.Allow("user1") {
|
|
t.Errorf("request %d should be allowed", i+1)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRateLimiter_BlockExceedingLimit(t *testing.T) {
|
|
rl := NewRateLimiter(3, time.Minute)
|
|
for i := 0; i < 3; i++ {
|
|
rl.Allow("user1")
|
|
}
|
|
if rl.Allow("user1") {
|
|
t.Error("4th request should be blocked")
|
|
}
|
|
}
|
|
|
|
func TestRateLimiter_DifferentKeys(t *testing.T) {
|
|
rl := NewRateLimiter(2, time.Minute)
|
|
rl.Allow("user1")
|
|
rl.Allow("user1")
|
|
|
|
if rl.Allow("user1") {
|
|
t.Error("user1 should be blocked")
|
|
}
|
|
if !rl.Allow("user2") {
|
|
t.Error("user2 should be allowed (independent bucket)")
|
|
}
|
|
}
|
|
|
|
func TestRateLimiter_WindowExpiry(t *testing.T) {
|
|
rl := NewRateLimiter(2, 50*time.Millisecond)
|
|
rl.Allow("user1")
|
|
rl.Allow("user1")
|
|
|
|
if rl.Allow("user1") {
|
|
t.Error("should be blocked immediately")
|
|
}
|
|
|
|
time.Sleep(60 * time.Millisecond)
|
|
|
|
if !rl.Allow("user1") {
|
|
t.Error("should be allowed after window expires")
|
|
}
|
|
}
|
|
|
|
func TestRateLimiter_Disabled(t *testing.T) {
|
|
rl := NewRateLimiter(0, time.Minute)
|
|
for i := 0; i < 100; i++ {
|
|
if !rl.Allow("user1") {
|
|
t.Error("should always allow when disabled")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRateLimiter_Concurrent(t *testing.T) {
|
|
rl := NewRateLimiter(100, time.Minute)
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 200; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
rl.Allow("user1")
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|