Accountant: Inventar, where, Checkmk, Abrechnung

pkg/ratelimit/ratelimit_test.go

@@ -0,0 +1,56 @@
+package server
+
+import (
+	"testing"
+	"time"
+)
+
+func TestRateLimiter(t *testing.T) {
+	now := time.Date(2026, 10, 2, 12, 0, 0, 0, time.UTC)
+	l := &rateLimiter{limit: 3, window: time.Minute, now: func() time.Time { return now }}
+
+	for i := range 3 {
+		if ok, _ := l.allow("1.2.3.4"); !ok {
+			t.Fatalf("Versuch %d abgelehnt", i+1)
+		}
+	}
+	ok, retry := l.allow("1.2.3.4")
+	if ok || retry != time.Minute {
+		t.Errorf("4. Versuch: ok=%v retry=%v, erwartet abgelehnt mit 1 min", ok, retry)
+	}
+	if ok, _ := l.allow("5.6.7.8"); !ok {
+		t.Error("andere IP mitbetroffen")
+	}
+
+	now = now.Add(40 * time.Second)
+	if _, retry := l.allow("1.2.3.4"); retry != 20*time.Second {
+		t.Errorf("Retry-After nach 40 s: %v, erwartet 20 s", retry)
+	}
+	now = now.Add(20 * time.Second)
+	if ok, _ := l.allow("1.2.3.4"); !ok {
+		t.Error("nach Ablauf des Fensters weiter gesperrt")
+	}
+}
+
+// Die Zählerliste darf bei vielen IPs nicht unbegrenzt wachsen.
+func TestRateLimiter_ForgetsOldEntries(t *testing.T) {
+	now := time.Date(2026, 10, 2, 12, 0, 0, 0, time.UTC)
+	l := &rateLimiter{limit: 1, window: time.Minute, now: func() time.Time { return now }}
+	for i := range 1000 {
+		l.allow(string(rune('a'+i%26)) + time.Duration(i).String())
+	}
+	now = now.Add(2 * time.Minute)
+	l.allow("neu")
+	if n := l.size(); n != 1 {
+		t.Errorf("%d Einträge nach Ablauf, erwartet 1", n)
+	}
+}
+
+func TestRateLimiter_Disabled(t *testing.T) {
+	l := &rateLimiter{limit: -1, window: time.Minute}
+	for range 100 {
+		if ok, _ := l.allow("x"); !ok {
+			t.Fatal("Limit -1 muss unbegrenzt sein")
+		}
+	}
+}