clash/component/domain-trie/trie_test.go

88 lines
1.8 KiB
Go
Raw Normal View History

2019-07-14 11:29:58 +00:00
package trie
import (
"net"
"testing"
"github.com/stretchr/testify/assert"
2019-07-14 11:29:58 +00:00
)
2019-07-15 10:00:51 +00:00
var localIP = net.IP{127, 0, 0, 1}
2019-07-14 11:29:58 +00:00
func TestTrie_Basic(t *testing.T) {
tree := New()
domains := []string{
"example.com",
"google.com",
}
for _, domain := range domains {
2019-07-15 10:00:51 +00:00
tree.Insert(domain, localIP)
2019-07-14 11:29:58 +00:00
}
node := tree.Search("example.com")
assert.NotNil(t, node)
assert.True(t, node.Data.(net.IP).Equal(localIP))
assert.NotNil(t, tree.Insert("", localIP))
2019-07-14 11:29:58 +00:00
}
func TestTrie_Wildcard(t *testing.T) {
tree := New()
domains := []string{
"*.example.com",
"sub.*.example.com",
"*.dev",
".org",
".example.net",
2020-04-24 15:49:19 +00:00
".apple.*",
2019-07-14 11:29:58 +00:00
}
for _, domain := range domains {
2019-07-15 10:00:51 +00:00
tree.Insert(domain, localIP)
2019-07-14 11:29:58 +00:00
}
assert.NotNil(t, tree.Search("sub.example.com"))
assert.NotNil(t, tree.Search("sub.foo.example.com"))
assert.NotNil(t, tree.Search("test.org"))
assert.NotNil(t, tree.Search("test.example.net"))
2020-04-24 15:49:19 +00:00
assert.NotNil(t, tree.Search("test.apple.com"))
assert.Nil(t, tree.Search("foo.sub.example.com"))
assert.Nil(t, tree.Search("foo.example.dev"))
assert.Nil(t, tree.Search("example.com"))
}
2019-07-14 11:29:58 +00:00
func TestTrie_Priority(t *testing.T) {
tree := New()
domains := []string{
".dev",
"example.dev",
"*.example.dev",
"test.example.dev",
2019-07-14 11:29:58 +00:00
}
assertFn := func(domain string, data int) {
node := tree.Search(domain)
assert.NotNil(t, node)
assert.Equal(t, data, node.Data)
2019-07-14 11:29:58 +00:00
}
for idx, domain := range domains {
tree.Insert(domain, idx)
2019-07-14 11:29:58 +00:00
}
2019-07-15 10:00:51 +00:00
assertFn("test.dev", 0)
assertFn("foo.bar.dev", 0)
assertFn("example.dev", 1)
assertFn("foo.example.dev", 2)
assertFn("test.example.dev", 3)
2019-07-14 11:29:58 +00:00
}
func TestTrie_Boundary(t *testing.T) {
tree := New()
2019-07-15 10:00:51 +00:00
tree.Insert("*.dev", localIP)
2019-07-14 11:29:58 +00:00
assert.NotNil(t, tree.Insert(".", localIP))
assert.NotNil(t, tree.Insert("..dev", localIP))
assert.Nil(t, tree.Search("dev"))
2019-07-14 11:29:58 +00:00
}