Algis DumbrisandClaude Opus 4.6
42775df65f
- Webhook registration via MCP (register_webhook, list_webhooks, delete_webhook) - HMAC-SHA256 payload signing, SSRF-safe HTTP client, loop detection (depth 5) - 8-worker goroutine delivery pool with exponential backoff retry (1s/5s/30s) - Dead letter queue with auto-purge, auto-disable after 50 consecutive failures - Per-agent rate limiting (60 deliveries/min) - K8s Job runner (register_k8s_handler, list_k8s_handlers, delete_k8s_handler) - Auto-detect in-cluster via InClusterConfig, NoopRunner fallback - REST API for webhook deliveries, dead letters, K8s job runs and logs - Web UI: webhook management, K8s handler pages, dead letters view - MultiDispatcher fan-out pattern for webhook + K8s event dispatch - SQLite migration 009: webhooks, webhook_deliveries, k8s_handlers, k8s_job_runs - 51 tests across 9 test packages, all passing Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package webhooks
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestRateLimiterAllow(t *testing.T) {
|
|
// Create a limiter with 60 per minute (burst of 60)
|
|
rl := NewAgentRateLimiter(60)
|
|
|
|
// First call should be allowed (burst available)
|
|
if !rl.Allow("agent-1") {
|
|
t.Error("first Allow() should return true (burst available)")
|
|
}
|
|
|
|
// Multiple rapid calls should be allowed up to burst capacity
|
|
allowed := 0
|
|
for i := 0; i < 100; i++ {
|
|
if rl.Allow("agent-2") {
|
|
allowed++
|
|
}
|
|
}
|
|
// The burst is 60, so we should get at most 60 allowed
|
|
if allowed > 60 {
|
|
t.Errorf("allowed %d calls, expected at most 60 (burst)", allowed)
|
|
}
|
|
if allowed < 50 {
|
|
t.Errorf("allowed %d calls, expected around 60 (burst)", allowed)
|
|
}
|
|
|
|
// Different agents have independent limits
|
|
if !rl.Allow("agent-3") {
|
|
t.Error("new agent should have full burst available")
|
|
}
|
|
}
|
|
|
|
func TestRateLimiterRemove(t *testing.T) {
|
|
rl := NewAgentRateLimiter(60)
|
|
|
|
// Exhaust burst for agent-1
|
|
for i := 0; i < 100; i++ {
|
|
rl.Allow("agent-remove")
|
|
}
|
|
|
|
// Remove the limiter
|
|
rl.Remove("agent-remove")
|
|
|
|
// After removal, a new limiter is created with full burst
|
|
if !rl.Allow("agent-remove") {
|
|
t.Error("after Remove(), agent should get a fresh limiter with full burst")
|
|
}
|
|
}
|