136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
// 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")))
|
|
}
|
|
}
|