mirror of https://github.com/XTLS/Xray-core.git
94 lines
1.5 KiB
Go
94 lines
1.5 KiB
Go
package strmatcher_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/xtls/xray-core/common"
|
|
. "github.com/xtls/xray-core/common/strmatcher"
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|