package middleware

import (
	"fmt"
	"testing"
)

// jiu 反馈波安全 review item 2:攻击者可用无限伪造 IP(哪怕已按 SetTrustedProxies(nil)
// 锁死了 XFF 伪造这条路,真实的分布式来源/僵尸网络仍能刷出海量不同 socket 源 IP)撑爆
// buckets map。这条白盒测试直接灌入超过 maxBuckets 的不同 IP,断言 map 有硬上限、
// 不会无限增长,且超限后的新 IP 被安全拒绝(而不是继续建桶或 panic)。
func TestIPRateLimiterBucketMapBounded(t *testing.T) {
	rl := NewIPRateLimiter(1)

	for i := 0; i < maxBuckets; i++ {
		rl.allow(fmt.Sprintf("spoofed-ip-%d", i))
	}

	rl.mu.Lock()
	sizeAtCap := len(rl.buckets)
	rl.mu.Unlock()
	if sizeAtCap != maxBuckets {
		t.Fatalf("灌满硬上限后 map size 应等于 maxBuckets(%d), got %d", maxBuckets, sizeAtCap)
	}

	// 硬上限已满、且都是刚创建的活跃桶(sweep 回收不了任何一个):新 IP 应被拒绝
	// (安全侧,宁可误限也不无界建桶),不应放行/建桶。
	for i := 0; i < 100; i++ {
		if rl.allow(fmt.Sprintf("overflow-ip-%d", i)) {
			t.Fatalf("硬上限已满时,超限的新 IP(overflow-ip-%d)应被拒绝而非放行", i)
		}
	}

	rl.mu.Lock()
	sizeAfterOverflow := len(rl.buckets)
	rl.mu.Unlock()
	if sizeAfterOverflow > maxBuckets {
		t.Fatalf("超限请求后 buckets map 不应继续增长: size=%d max=%d", sizeAfterOverflow, maxBuckets)
	}
}