|
@@ -0,0 +1,94 @@
|
|
|
|
|
+package strmatcher_test
|
|
|
|
|
+
|
|
|
|
|
+import (
|
|
|
|
|
+ "reflect"
|
|
|
|
|
+ "testing"
|
|
|
|
|
+
|
|
|
|
|
+ "v2ray.com/core/common"
|
|
|
|
|
+ . "v2ray.com/core/common/strmatcher"
|
|
|
|
|
+)
|
|
|
|
|
+
|
|
|
|
|
+// See https://github.com/v2fly/v2ray-core/issues/92#issuecomment-673238489
|
|
|
|
|
+func TestMatcherGroup(t *testing.T) {
|
|
|
|
|
+ rules := []struct {
|
|
|
|
|
+ Type Type
|
|
|
|
|
+ Domain string
|
|
|
|
|
+ }{
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Regex,
|
|
|
|
|
+ Domain: "apis\\.us$",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Substr,
|
|
|
|
|
+ Domain: "apis",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Domain,
|
|
|
|
|
+ Domain: "googleapis.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Domain,
|
|
|
|
|
+ Domain: "com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Full,
|
|
|
|
|
+ Domain: "www.baidu.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Substr,
|
|
|
|
|
+ Domain: "apis",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Domain,
|
|
|
|
|
+ Domain: "googleapis.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Full,
|
|
|
|
|
+ Domain: "fonts.googleapis.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Full,
|
|
|
|
|
+ Domain: "www.baidu.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Type: Domain,
|
|
|
|
|
+ Domain: "example.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ }
|
|
|
|
|
+ cases := []struct {
|
|
|
|
|
+ Input string
|
|
|
|
|
+ Output []uint32
|
|
|
|
|
+ }{
|
|
|
|
|
+ {
|
|
|
|
|
+ Input: "www.baidu.com",
|
|
|
|
|
+ Output: []uint32{5, 9, 4},
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Input: "fonts.googleapis.com",
|
|
|
|
|
+ Output: []uint32{8, 3, 7, 4, 2, 6},
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Input: "example.googleapis.com",
|
|
|
|
|
+ Output: []uint32{3, 7, 4, 2, 6},
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Input: "testapis.us",
|
|
|
|
|
+ Output: []uint32{1, 2, 6},
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ Input: "example.com",
|
|
|
|
|
+ Output: []uint32{10, 4},
|
|
|
|
|
+ },
|
|
|
|
|
+ }
|
|
|
|
|
+ matcherGroup := &MatcherGroup{}
|
|
|
|
|
+ for _, rule := range rules {
|
|
|
|
|
+ matcher, err := rule.Type.New(rule.Domain)
|
|
|
|
|
+ common.Must(err)
|
|
|
|
|
+ matcherGroup.Add(matcher)
|
|
|
|
|
+ }
|
|
|
|
|
+ for _, test := range cases {
|
|
|
|
|
+ if m := matcherGroup.Match(test.Input); !reflect.DeepEqual(m, test.Output) {
|
|
|
|
|
+ t.Error("unexpected output: ", m, " for test case ", test)
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+}
|