mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-11-16 03:32:33 +08:00
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package trie_test
|
|
|
|
import (
|
|
"golang.org/x/exp/slices"
|
|
"testing"
|
|
|
|
"github.com/metacubex/mihomo/component/trie"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testDump(t *testing.T, tree *trie.DomainTrie[struct{}], set *trie.DomainSet) {
|
|
var dataSrc []string
|
|
tree.Foreach(func(domain string, data struct{}) bool {
|
|
dataSrc = append(dataSrc, domain)
|
|
return true
|
|
})
|
|
slices.Sort(dataSrc)
|
|
var dataSet []string
|
|
set.Foreach(func(key string) bool {
|
|
dataSet = append(dataSet, key)
|
|
return true
|
|
})
|
|
slices.Sort(dataSet)
|
|
assert.Equal(t, dataSrc, dataSet)
|
|
}
|
|
|
|
func TestDomainSet(t *testing.T) {
|
|
tree := trie.New[struct{}]()
|
|
domainSet := []string{
|
|
"baidu.com",
|
|
"google.com",
|
|
"www.google.com",
|
|
"test.a.net",
|
|
"test.a.oc",
|
|
"Mijia Cloud",
|
|
".qq.com",
|
|
"+.cn",
|
|
}
|
|
|
|
for _, domain := range domainSet {
|
|
assert.NoError(t, tree.Insert(domain, struct{}{}))
|
|
}
|
|
assert.False(t, tree.IsEmpty())
|
|
set := tree.NewDomainSet()
|
|
assert.NotNil(t, set)
|
|
assert.True(t, set.Has("test.cn"))
|
|
assert.True(t, set.Has("cn"))
|
|
assert.True(t, set.Has("Mijia Cloud"))
|
|
assert.True(t, set.Has("test.a.net"))
|
|
assert.True(t, set.Has("www.qq.com"))
|
|
assert.True(t, set.Has("google.com"))
|
|
assert.False(t, set.Has("qq.com"))
|
|
assert.False(t, set.Has("www.baidu.com"))
|
|
testDump(t, tree, set)
|
|
}
|
|
|
|
func TestDomainSetComplexWildcard(t *testing.T) {
|
|
tree := trie.New[struct{}]()
|
|
domainSet := []string{
|
|
"+.baidu.com",
|
|
"+.a.baidu.com",
|
|
"www.baidu.com",
|
|
"+.bb.baidu.com",
|
|
"test.a.net",
|
|
"test.a.oc",
|
|
"www.qq.com",
|
|
}
|
|
|
|
for _, domain := range domainSet {
|
|
assert.NoError(t, tree.Insert(domain, struct{}{}))
|
|
}
|
|
assert.False(t, tree.IsEmpty())
|
|
set := tree.NewDomainSet()
|
|
assert.NotNil(t, set)
|
|
assert.False(t, set.Has("google.com"))
|
|
assert.True(t, set.Has("www.baidu.com"))
|
|
assert.True(t, set.Has("test.test.baidu.com"))
|
|
testDump(t, tree, set)
|
|
}
|
|
|
|
func TestDomainSetWildcard(t *testing.T) {
|
|
tree := trie.New[struct{}]()
|
|
domainSet := []string{
|
|
"*.*.*.baidu.com",
|
|
"www.baidu.*",
|
|
"stun.*.*",
|
|
"*.*.qq.com",
|
|
"test.*.baidu.com",
|
|
"*.apple.com",
|
|
}
|
|
|
|
for _, domain := range domainSet {
|
|
assert.NoError(t, tree.Insert(domain, struct{}{}))
|
|
}
|
|
assert.False(t, tree.IsEmpty())
|
|
set := tree.NewDomainSet()
|
|
assert.NotNil(t, set)
|
|
assert.True(t, set.Has("www.baidu.com"))
|
|
assert.True(t, set.Has("test.test.baidu.com"))
|
|
assert.True(t, set.Has("test.test.qq.com"))
|
|
assert.True(t, set.Has("stun.ab.cd"))
|
|
assert.False(t, set.Has("test.baidu.com"))
|
|
assert.False(t, set.Has("www.google.com"))
|
|
assert.False(t, set.Has("a.www.google.com"))
|
|
assert.False(t, set.Has("test.qq.com"))
|
|
assert.False(t, set.Has("test.test.test.qq.com"))
|
|
testDump(t, tree, set)
|
|
}
|