chore: import upstream snapshot with attribution

2026-07-13 13:00:08 +08:00
commit 6345505628
1516 changed files with 473769 additions and 0 deletions
@@ -0,0 +1,138 @@
+package agent
+
+import (
+	"testing"
+	"time"
+)
+
+// TestClassificationCache_GetSet 测试缓存的基本功能
+func TestClassificationCache_GetSet(t *testing.T) {
+	cache := newClassificationCache()
+
+	// 测试缓存未命中
+	_, ok := cache.Get("hello")
+	if ok {
+		t.Error("expected cache miss for new input")
+	}
+
+	// 测试缓存设置和命中
+	cache.Set("hello", false)
+	got, ok := cache.Get("hello")
+	if !ok {
+		t.Error("expected cache hit after Set")
+	}
+	if got != false {
+		t.Errorf("Get() = %v, want false", got)
+	}
+
+	// 测试不同输入
+	cache.Set("fix bug", true)
+	got2, ok2 := cache.Get("fix bug")
+	if !ok2 {
+		t.Error("expected cache hit for second input")
+	}
+	if got2 != true {
+		t.Errorf("Get() = %v, want true", got2)
+	}
+}
+
+// TestClassificationCache_TTL 测试缓存过期
+func TestClassificationCache_TTL(t *testing.T) {
+	cache := &classificationCache{
+		entries: make(map[string]cacheEntry),
+		maxSize: 100,
+		ttl:     100 * time.Millisecond, // 短 TTL 用于测试
+	}
+
+	cache.Set("hello", false)
+
+	// 立即获取应该命中
+	_, ok := cache.Get("hello")
+	if !ok {
+		t.Error("expected cache hit immediately after Set")
+	}
+
+	// 等待 TTL 过期
+	time.Sleep(150 * time.Millisecond)
+
+	// 应该过期
+	_, ok = cache.Get("hello")
+	if ok {
+		t.Error("expected cache miss after TTL expiration")
+	}
+}
+
+// TestClassificationCache_Clear 测试缓存清除
+func TestClassificationCache_Clear(t *testing.T) {
+	cache := newClassificationCache()
+
+	cache.Set("hello", false)
+	cache.Set("fix bug", true)
+
+	// 清除前应该命中
+	_, ok1 := cache.Get("hello")
+	_, ok2 := cache.Get("fix bug")
+	if !ok1 || !ok2 {
+		t.Error("expected cache hits before Clear")
+	}
+
+	cache.Clear()
+
+	// 清除后应该未命中
+	_, ok1 = cache.Get("hello")
+	_, ok2 = cache.Get("fix bug")
+	if ok1 || ok2 {
+		t.Error("expected cache misses after Clear")
+	}
+}
+
+// TestClassificationCache_LRUEviction 测试 LRU 淘汰
+func TestClassificationCache_LRUEviction(t *testing.T) {
+	cache := &classificationCache{
+		entries: make(map[string]cacheEntry),
+		maxSize: 5, // 小的最大值用于测试
+		ttl:     5 * time.Minute,
+	}
+
+	// 填充到最大容量
+	for i := 0; i < 5; i++ {
+		cache.Set(string(rune('a'+i)), i%2 == 0)
+	}
+
+	if len(cache.entries) != 5 {
+		t.Errorf("expected 5 entries, got %d", len(cache.entries))
+	}
+
+	// 添加第 6 个元素应该触发淘汰
+	cache.Set("f", true)
+
+	// 缓存大小应该减少（淘汰了 20%）
+	if len(cache.entries) > 5 {
+		t.Errorf("expected cache size <= 5 after eviction, got %d", len(cache.entries))
+	}
+}
+
+// TestNormalizeInputForCache 测试输入规范化
+func TestNormalizeInputForCache(t *testing.T) {
+	tests := []struct {
+		input1 string
+		input2 string
+		same   bool
+	}{
+		{"hello", "hello", true},
+		{"hello", "Hello", true},      // 大小写
+		{"hello", " hello ", true},    // 空格
+		{"hello", "hello!", false},    // 不同内容
+		{"fix bug", "fix  bug", true}, // 多余空格（SHA256 会不同，但规范化相同）
+	}
+
+	for _, tt := range tests {
+		hash1 := normalizeInputForCache(tt.input1)
+		hash2 := normalizeInputForCache(tt.input2)
+		same := hash1 == hash2
+		if same != tt.same {
+			t.Errorf("normalizeInputForCache(%q) == normalizeInputForCache(%q): got %v, want %v",
+				tt.input1, tt.input2, same, tt.same)
+		}
+	}
+}