feat(routing): 档案校验 + 规范化(白名单/上限/去重)

2026-07-28 07:31:55 +08:00
parent 04b02da78d
commit b98aa2335b
2 changed files with 128 additions and 0 deletions
@@ -0,0 +1,50 @@
+package routing
+
+import "testing"
+
+func TestValidate(t *testing.T) {
+	ok := Default()
+	ok.Rules = []Rule{{Type: "domain_suffix", Value: "example.com", Action: "direct", Enabled: true}}
+	if e := ok.Validate(); len(e) != 0 {
+		t.Fatalf("valid profile got errors %v", e)
+	}
+
+	bad := Default()
+	bad.Mode = "weird"   // 非法 mode
+	bad.Final = "reject" // final 只能 proxy|direct
+	bad.Rules = []Rule{
+		{Type: "ip_cidr", Value: "not-a-cidr", Action: "proxy", Enabled: true}, // CIDR 非法
+		{Type: "geosite", Value: "netflix", Action: "direct", Enabled: true},   // 不在白名单(仅 cn)
+		{Type: "bogus", Value: "x", Action: "direct", Enabled: true},           // type 非法
+	}
+	errs := bad.Validate()
+	if len(errs) < 5 {
+		t.Fatalf("want >=5 field errors, got %d: %v", len(errs), errs)
+	}
+}
+
+func TestValidateCountLimit(t *testing.T) {
+	p := Default()
+	for i := 0; i < 201; i++ {
+		p.Rules = append(p.Rules, Rule{Type: "domain", Value: "a.com", Action: "proxy", Enabled: true})
+	}
+	if e := p.Validate(); len(e) == 0 {
+		t.Fatal("want count-limit error")
+	}
+}
+
+func TestNormalizeDedupAndTrim(t *testing.T) {
+	p := Default()
+	p.Rules = []Rule{
+		{Type: "domain", Value: "  a.com  ", Action: "proxy", Enabled: true},
+		{Type: "domain", Value: "a.com", Action: "proxy", Enabled: false},
+		{Type: "domain", Value: "b.com", Action: "proxy", Enabled: true},
+	}
+	p.Normalize()
+	if len(p.Rules) != 2 {
+		t.Fatalf("want 2 rules after dedup, got %d: %v", len(p.Rules), p.Rules)
+	}
+	if p.Rules[0].Value != "a.com" || p.Rules[1].Value != "b.com" {
+		t.Fatalf("want first-occurrence order preserved, got %v", p.Rules)
+	}
+}