[fix]code

adapter/provider/provider_test.go

@@ -0,0 +1,58 @@
+package provider
+
+import (
+	"github.com/Dreamacro/clash/constant"
+	rules "github.com/Dreamacro/clash/rule"
+
+	"github.com/stretchr/testify/assert"
+	"net"
+	"testing"
+	"time"
+)
+
+func setup() {
+	SetClassicalRuleParser(func(ruleType, rule string, params []string) (constant.Rule, error) {
+		if params == nil {
+			params = make([]string, 0)
+		}
+
+		return rules.ParseRule(ruleType, rule, "", params)
+	})
+}
+
+func TestDomain(t *testing.T) {
+	setup()
+	domainProvider := NewRuleSetProvider("test", Domain,
+		time.Duration(uint(100000)), NewFileVehicle("./domain.txt"))
+	assert.Nil(t, domainProvider.Initial())
+	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "youtube.com"}))
+	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "www.youtube.com"}))
+	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "test.youtube.com"}))
+	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "bcovlive-a.akamaihd.net"}))
+	assert.False(t, domainProvider.Match(&constant.Metadata{Host: "baidu.com"}))
+}
+
+func TestClassical(t *testing.T) {
+	setup()
+	classicalProvider := NewRuleSetProvider("test", Classical,
+		time.Duration(uint(100000)), NewFileVehicle("./classical.txt"))
+	assert.Nil(t, classicalProvider.Initial())
+	assert.True(t, classicalProvider.Match(&constant.Metadata{Host: "www.10010.com", AddrType: constant.AtypDomainName}))
+	assert.False(t, classicalProvider.Match(&constant.Metadata{Host: "google.com", AddrType: constant.AtypDomainName}))
+	assert.True(t, classicalProvider.Match(&constant.Metadata{Host: "analytics.strava.com", AddrType: constant.AtypDomainName}))
+	assert.True(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2a0b:b580::1")}))
+	assert.False(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2a0b:c582::1")}))
+	assert.True(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("1.255.62.34")}))
+	assert.False(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("103.65.41.199")}))
+}
+
+func TestIpCidr(t *testing.T) {
+	setup()
+	domainProvider := NewRuleSetProvider("test", IPCIDR,
+		time.Duration(uint(100000)), NewFileVehicle("./ipcidr.txt"))
+	assert.Nil(t, domainProvider.Initial())
+	assert.True(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("91.108.22.10")}))
+	assert.False(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("149.190.220.251")}))
+	assert.True(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2001:b28:f23f:f005::a")}))
+	assert.False(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2006:b28:f23f:f005::a")}))
+}