mirror of https://github.com/XTLS/Xray-core
77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
package strmatcher_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
. "github.com/xtls/xray-core/common/strmatcher"
|
|
)
|
|
|
|
func TestDomainMatcherGroup(t *testing.T) {
|
|
g := new(DomainMatcherGroup)
|
|
g.Add("example.com", 1)
|
|
g.Add("google.com", 2)
|
|
g.Add("x.a.com", 3)
|
|
g.Add("a.b.com", 4)
|
|
g.Add("c.a.b.com", 5)
|
|
g.Add("x.y.com", 4)
|
|
g.Add("x.y.com", 6)
|
|
|
|
testCases := []struct {
|
|
Domain string
|
|
Result []uint32
|
|
}{
|
|
{
|
|
Domain: "x.example.com",
|
|
Result: []uint32{1},
|
|
},
|
|
{
|
|
Domain: "y.com",
|
|
Result: nil,
|
|
},
|
|
{
|
|
Domain: "a.b.com",
|
|
Result: []uint32{4},
|
|
},
|
|
{ // Matches [c.a.b.com, a.b.com]
|
|
Domain: "c.a.b.com",
|
|
Result: []uint32{5, 4},
|
|
},
|
|
{
|
|
Domain: "c.a..b.com",
|
|
Result: nil,
|
|
},
|
|
{
|
|
Domain: ".com",
|
|
Result: nil,
|
|
},
|
|
{
|
|
Domain: "com",
|
|
Result: nil,
|
|
},
|
|
{
|
|
Domain: "",
|
|
Result: nil,
|
|
},
|
|
{
|
|
Domain: "x.y.com",
|
|
Result: []uint32{4, 6},
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
r := g.Match(testCase.Domain)
|
|
if !reflect.DeepEqual(r, testCase.Result) {
|
|
t.Error("Failed to match domain: ", testCase.Domain, ", expect ", testCase.Result, ", but got ", r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEmptyDomainMatcherGroup(t *testing.T) {
|
|
g := new(DomainMatcherGroup)
|
|
r := g.Match("example.com")
|
|
if len(r) != 0 {
|
|
t.Error("Expect [], but ", r)
|
|
}
|
|
}
|