mirror of
https://github.com/SagerNet/sing.git
synced 2025-04-03 20:07:38 +03:00
80 lines
2.7 KiB
Go
80 lines
2.7 KiB
Go
package domain_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/sagernet/sing/common/domain"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMatcher(t *testing.T) {
|
|
t.Parallel()
|
|
testDomain := []string{"example.com", "example.org"}
|
|
testDomainSuffix := []string{".com.cn", ".org.cn", "sagernet.org"}
|
|
matcher := domain.NewMatcher(testDomain, testDomainSuffix, false)
|
|
require.NotNil(t, matcher)
|
|
require.True(t, matcher.Match("example.com"))
|
|
require.True(t, matcher.Match("example.org"))
|
|
require.False(t, matcher.Match("example.cn"))
|
|
require.True(t, matcher.Match("example.com.cn"))
|
|
require.True(t, matcher.Match("example.org.cn"))
|
|
require.False(t, matcher.Match("com.cn"))
|
|
require.False(t, matcher.Match("org.cn"))
|
|
require.True(t, matcher.Match("sagernet.org"))
|
|
require.True(t, matcher.Match("sing-box.sagernet.org"))
|
|
dDomain, dDomainSuffix := matcher.Dump()
|
|
require.Equal(t, testDomain, dDomain)
|
|
require.Equal(t, testDomainSuffix, dDomainSuffix)
|
|
}
|
|
|
|
func TestMatcherLegacy(t *testing.T) {
|
|
t.Parallel()
|
|
testDomain := []string{"example.com", "example.org"}
|
|
testDomainSuffix := []string{".com.cn", ".org.cn", "sagernet.org"}
|
|
matcher := domain.NewMatcher(testDomain, testDomainSuffix, true)
|
|
require.NotNil(t, matcher)
|
|
require.True(t, matcher.Match("example.com"))
|
|
require.True(t, matcher.Match("example.org"))
|
|
require.False(t, matcher.Match("example.cn"))
|
|
require.True(t, matcher.Match("example.com.cn"))
|
|
require.True(t, matcher.Match("example.org.cn"))
|
|
require.False(t, matcher.Match("com.cn"))
|
|
require.False(t, matcher.Match("org.cn"))
|
|
require.True(t, matcher.Match("sagernet.org"))
|
|
require.True(t, matcher.Match("sing-box.sagernet.org"))
|
|
dDomain, dDomainSuffix := matcher.Dump()
|
|
require.Equal(t, testDomain, dDomain)
|
|
require.Equal(t, testDomainSuffix, dDomainSuffix)
|
|
}
|
|
|
|
type simpleRuleSet struct {
|
|
Rules []struct {
|
|
Domain []string `json:"domain"`
|
|
DomainSuffix []string `json:"domain_suffix"`
|
|
}
|
|
}
|
|
|
|
func TestDumpLarge(t *testing.T) {
|
|
t.Parallel()
|
|
response, err := http.Get("https://raw.githubusercontent.com/MetaCubeX/meta-rules-dat/sing/geo/geosite/cn.json")
|
|
require.NoError(t, err)
|
|
defer response.Body.Close()
|
|
var ruleSet simpleRuleSet
|
|
err = json.NewDecoder(response.Body).Decode(&ruleSet)
|
|
require.NoError(t, err)
|
|
domainList := ruleSet.Rules[0].Domain
|
|
domainSuffixList := ruleSet.Rules[0].DomainSuffix
|
|
require.Len(t, ruleSet.Rules, 1)
|
|
require.True(t, len(domainList)+len(domainSuffixList) > 0)
|
|
sort.Strings(domainList)
|
|
sort.Strings(domainSuffixList)
|
|
matcher := domain.NewMatcher(domainList, domainSuffixList, false)
|
|
require.NotNil(t, matcher)
|
|
dDomain, dDomainSuffix := matcher.Dump()
|
|
require.Equal(t, domainList, dDomain)
|
|
require.Equal(t, domainSuffixList, dDomainSuffix)
|
|
}
|