Files
pangolin/server/internal/routing/profile_test.go
T

51 lines
1.6 KiB
Go

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)
}
}