1.4.5.2

EdgeCommon/pkg/dnsconfigs/ns_route_ranges_test.go

@@ -0,0 +1,135 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package dnsconfigs_test
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/dnsconfigs"
+	"github.com/iwind/TeaGo/assert"
+	"net"
+	"testing"
+)
+
+func TestNSRouteRangeIPRange_Contains(t *testing.T) {
+	var a = assert.NewAssertion(t)
+
+	// ipv4
+	{
+		var r = &dnsconfigs.NSRouteRangeIPRange{
+			IPFrom: "192.168.1.100",
+			IPTo:   "192.168.3.200",
+		}
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(r.Contains(net.ParseIP("aaa")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.1.200")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.3.200")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.4.1")))
+		a.IsFalse(r.Contains(net.ParseIP("::1")))
+	}
+
+	// ipv6
+	{
+		var prefix = "1:2:3:4:5:6"
+		var r = &dnsconfigs.NSRouteRangeIPRange{
+			IPFrom: prefix + ":1:8",
+			IPTo:   prefix + ":5:10",
+		}
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(r.Contains(net.ParseIP("aaa")))
+		a.IsTrue(r.Contains(net.ParseIP(prefix + ":3:4")))
+		a.IsTrue(r.Contains(net.ParseIP(prefix + ":5:9")))
+		a.IsTrue(r.Contains(net.ParseIP(prefix + ":5:10")))
+		a.IsTrue(r.Contains(net.ParseIP(prefix + ":4:8")))
+		a.IsFalse(r.Contains(net.ParseIP(prefix + ":5:11")))
+	}
+
+	{
+		var r = &dnsconfigs.NSRouteRangeCIDR{
+			CIDR: "192.168.2.1/24",
+		}
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(r.Contains(net.ParseIP("aaa")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.1")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.254")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.100")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.3.1")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.1.1")))
+	}
+
+	// reverse ipv4
+	{
+		var r = &dnsconfigs.NSRouteRangeIPRange{
+			IPFrom: "192.168.1.100",
+			IPTo:   "192.168.3.200",
+		}
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(r.Contains(net.ParseIP("aaa")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.1.200")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.3.200")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.4.1")))
+	}
+
+	// reverse cidr
+	{
+		var r = &dnsconfigs.NSRouteRangeCIDR{
+			CIDR: "192.168.2.1/24",
+		}
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(r.Contains(net.ParseIP("aaa")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.1")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.254")))
+		a.IsTrue(r.Contains(net.ParseIP("192.168.2.100")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.3.1")))
+		a.IsFalse(r.Contains(net.ParseIP("192.168.1.1")))
+	}
+}
+
+type testNSIPResolver struct {
+}
+
+func (this *testNSIPResolver) Resolve(ip net.IP) (countryId int64, provinceId int64, cityId int64, providerId int64) {
+	return 1, 2, 3, 4
+}
+
+func TestNSRouteRangeRegion_Contains(t *testing.T) {
+	{
+		var r = &dnsconfigs.NSRouteRangeRegion{
+			Regions:   nil,
+			Connector: "",
+		}
+		r.Regions = append(r.Regions, &dnsconfigs.RouteRegion{
+			Type: "country",
+			Id:   1,
+			Name: "1",
+		})
+		r.Regions = append(r.Regions, &dnsconfigs.RouteRegion{
+			Type: "province",
+			Id:   2,
+			Name: "2",
+		})
+		r.SetRegionResolver(&testNSIPResolver{})
+		err := r.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		t.Log(r.Contains(net.ParseIP("1.1.1.1")))
+
+		r.Connector = "AND"
+		t.Log(r.Contains(net.ParseIP("1.1.1.1")))
+	}
+}