mirror of
https://github.com/TeaOSLab/EdgeCommon.git
synced 2025-11-04 05:00:24 +08:00
260 lines
6.5 KiB
Go
260 lines
6.5 KiB
Go
// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cloud .
|
|
|
|
package iputils_test
|
|
|
|
import (
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/iputils"
|
|
"github.com/iwind/TeaGo/assert"
|
|
)
|
|
|
|
func TestIP_ParseIP(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
|
|
{
|
|
var i = iputils.ParseIP("127.0.0.1")
|
|
a.IsTrue(i.IsIPv4())
|
|
a.IsFalse(i.IsIPv6())
|
|
a.IsTrue(i.IsValid())
|
|
a.IsTrue(iputils.IsIPv4("127.0.0.1"))
|
|
a.IsFalse(iputils.IsIPv6("127.0.0.1"))
|
|
t.Log(i.String(), i.ToLong())
|
|
t.Log("raw:", i.Raw())
|
|
a.IsTrue(iputils.IsValid("127.0.0.1"))
|
|
}
|
|
|
|
{
|
|
var i = iputils.ParseIP("0.0.0.1")
|
|
a.IsTrue(i.IsIPv4())
|
|
a.IsFalse(i.IsIPv6())
|
|
t.Log(i.String(), i.ToLong())
|
|
}
|
|
|
|
for j := 0; j < 3; j++ /** repeat test **/ {
|
|
var i = iputils.ParseIP("::1")
|
|
a.IsFalse(i.IsIPv4())
|
|
a.IsTrue(i.IsIPv6())
|
|
a.IsTrue(i.IsValid())
|
|
t.Log(i.String(), i.ToLong())
|
|
}
|
|
|
|
{
|
|
{
|
|
var i = iputils.ParseIP("2001:db8:0:1::1:101")
|
|
t.Log(i.String(), i.ToLong())
|
|
a.IsFalse(i.IsIPv4())
|
|
a.IsTrue(i.IsIPv6())
|
|
a.IsFalse(iputils.IsIPv4("2001:db8:0:1::1:101"))
|
|
a.IsTrue(iputils.IsIPv6("2001:db8:0:1::1:101"))
|
|
a.IsTrue(i.IsValid())
|
|
}
|
|
|
|
{
|
|
var i = iputils.ParseIP("2001:db8:0:1::1:102")
|
|
t.Log(i.String(), i.ToLong())
|
|
a.IsFalse(i.IsIPv4())
|
|
a.IsTrue(i.IsIPv6())
|
|
a.IsTrue(i.IsValid())
|
|
}
|
|
|
|
{
|
|
var i = iputils.ParseIP("2001:db8:0:1::2:101")
|
|
t.Log(i.String(), i.ToLong())
|
|
a.IsFalse(i.IsIPv4())
|
|
a.IsTrue(i.IsIPv6())
|
|
a.IsTrue(i.IsValid())
|
|
a.IsTrue(iputils.IsValid("2001:db8:0:1::2:101"))
|
|
}
|
|
}
|
|
|
|
{
|
|
var i = iputils.ParseIP("WRONG IP")
|
|
t.Log(i.String(), i.ToLong())
|
|
a.IsFalse(i.IsIPv4())
|
|
a.IsFalse(i.IsIPv6())
|
|
a.IsFalse(i.IsValid())
|
|
a.IsFalse(iputils.IsValid("WRONG IP"))
|
|
a.IsFalse(iputils.IsIPv4("WRONG IP"))
|
|
a.IsFalse(iputils.IsIPv6("WRONG IP"))
|
|
}
|
|
}
|
|
|
|
func TestIP_Mod(t *testing.T) {
|
|
for _, ip := range []string{
|
|
"127.0.0.1",
|
|
"::1",
|
|
"2001:db8:0:1::1:101",
|
|
"2001:db8:0:1::1:102",
|
|
"WRONG IP",
|
|
} {
|
|
var i = iputils.ParseIP(ip)
|
|
t.Log(ip, "=>", i.ToLong(), "=>", i.Mod(5))
|
|
}
|
|
}
|
|
|
|
func TestIP_Compare(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("127.0.0.1")
|
|
var i2 = iputils.ParseIP("127.0.0.1")
|
|
a.IsTrue(i1.Compare(i2) == 0)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("127.0.0.1")
|
|
var i2 = iputils.ParseIP("127.0.0.2")
|
|
a.IsTrue(i1.Compare(i2) == -1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("127.0.0.2")
|
|
var i2 = iputils.ParseIP("127.0.0.1")
|
|
a.IsTrue(i1.Compare(i2) == 1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("2001:db8:0:1::101")
|
|
var i2 = iputils.ParseIP("127.0.0.1")
|
|
a.IsTrue(i1.Compare(i2) == 1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("127.0.0.1")
|
|
var i2 = iputils.ParseIP("2001:db8:0:1::101")
|
|
a.IsTrue(i1.Compare(i2) == -1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("2001:db8:0:1::101")
|
|
var i2 = iputils.ParseIP("2001:db8:0:1::101")
|
|
a.IsTrue(i1.Compare(i2) == 0)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("2001:db8:0:1::101")
|
|
var i2 = iputils.ParseIP("2001:db8:0:1::102")
|
|
a.IsTrue(i1.Compare(i2) == -1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("2001:db8:0:1::102")
|
|
var i2 = iputils.ParseIP("2001:db8:0:1::101")
|
|
a.IsTrue(i1.Compare(i2) == 1)
|
|
}
|
|
|
|
{
|
|
var i1 = iputils.ParseIP("2001:db8:0:1::2:100")
|
|
var i2 = iputils.ParseIP("2001:db8:0:1::1:101")
|
|
a.IsTrue(i1.Compare(i2) == 1)
|
|
}
|
|
}
|
|
|
|
func TestIP_Between(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
a.IsTrue(iputils.ParseIP("127.0.0.2").Between(iputils.ParseIP("127.0.0.1"), iputils.ParseIP("127.0.0.3")))
|
|
a.IsTrue(iputils.ParseIP("127.0.0.1").Between(iputils.ParseIP("127.0.0.1"), iputils.ParseIP("127.0.0.3")))
|
|
a.IsFalse(iputils.ParseIP("127.0.0.2").Between(iputils.ParseIP("127.0.0.3"), iputils.ParseIP("127.0.0.4")))
|
|
a.IsFalse(iputils.ParseIP("127.0.0.5").Between(iputils.ParseIP("127.0.0.3"), iputils.ParseIP("127.0.0.4")))
|
|
a.IsFalse(iputils.ParseIP("127.0.0.2").Between(iputils.ParseIP("127.0.0.3"), iputils.ParseIP("127.0.0.1")))
|
|
}
|
|
|
|
func TestIP_ToLong(t *testing.T) {
|
|
for _, ip := range []string{
|
|
"127.0.0.1",
|
|
"192.168.1.100",
|
|
"::1",
|
|
"fd00:6868:6868:0:10ac:d056:3bf6:7452",
|
|
"fd00:6868:6868:0:10ac:d056:3bf6:7453",
|
|
"2001:0db8:85a3:0000:0000:8a2e:0370:7334",
|
|
"2001:db8:0:1::101",
|
|
"2001:db8:0:2::101",
|
|
"wrong ip",
|
|
} {
|
|
var goIP = iputils.ParseIP(ip)
|
|
t.Log(ip, "=>", "\n", goIP.String(), "\n", "=>", "\n", "long1:", goIP.ToLong(), "\n", "long2:", iputils.ToLong(ip), "\n", "little long:", iputils.ToLittleLong(ip))
|
|
}
|
|
}
|
|
|
|
func TestIP_CompareLong(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
a.IsTrue(iputils.CompareLong("1", "2") == -1)
|
|
a.IsTrue(iputils.CompareLong("11", "2") == 1)
|
|
a.IsTrue(iputils.CompareLong("11", "22") == -1)
|
|
a.IsTrue(iputils.CompareLong("22", "101") == -1)
|
|
a.IsTrue(iputils.CompareLong("33", "22") == 1)
|
|
a.IsTrue(iputils.CompareLong("101", "22") == 1)
|
|
a.IsTrue(iputils.CompareLong("22", "22") == 0)
|
|
}
|
|
|
|
func TestIP_Memory(t *testing.T) {
|
|
var list []iputils.IP
|
|
|
|
var stat1 = &runtime.MemStats{}
|
|
runtime.ReadMemStats(stat1)
|
|
|
|
for i := 0; i < 1_000_000; i++ {
|
|
list = append(list, iputils.ParseIP("fd00:6868:6868:0:10ac:d056:3bf6:7452"))
|
|
}
|
|
|
|
// runtime.GC()
|
|
|
|
var stat2 = &runtime.MemStats{}
|
|
runtime.ReadMemStats(stat2)
|
|
|
|
t.Log((stat2.Alloc-stat1.Alloc)>>10, "KB", (stat2.HeapInuse-stat1.HeapInuse)>>10, "KB")
|
|
|
|
// hold the memory
|
|
for _, v := range list {
|
|
_ = v
|
|
}
|
|
}
|
|
|
|
func TestToBytes(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
a.IsTrue(len(iputils.ToBytes("a")) == 0)
|
|
a.IsTrue(len(iputils.ToBytes("192.168.1.100")) == 4)
|
|
a.IsTrue(len(iputils.ToBytes("::1")) == 16)
|
|
}
|
|
|
|
func TestCompareIP(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
a.IsTrue(iputils.CompareIP("a", "b") == 0)
|
|
a.IsTrue(iputils.CompareIP("192.168.1.100", "192.168.1.1") > 0)
|
|
a.IsTrue(iputils.CompareIP("192.168.1.100", "10.168.1.1") > 0)
|
|
a.IsTrue(iputils.CompareIP("192.168.1.100", "192.168.2.1") < 0)
|
|
a.IsTrue(iputils.CompareIP("192.168.1.100", "::1") < 0)
|
|
a.IsTrue(iputils.CompareIP("::1", "192.168.1.100") > 0)
|
|
a.IsTrue(iputils.CompareIP("192.168.1.100", "192.168.1.100") == 0)
|
|
}
|
|
|
|
func TestIsSameVersion(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
a.IsTrue(iputils.IsSameVersion("192.168.1.1", "10.0.0.1"))
|
|
a.IsTrue(iputils.IsSameVersion("::1", "::5"))
|
|
a.IsFalse(iputils.IsSameVersion("192.168.1.1", "::5"))
|
|
}
|
|
|
|
func BenchmarkParse(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
iputils.ParseIP("fd00:6868:6868:0:10ac:d056:3bf6:7452")
|
|
}
|
|
}
|
|
|
|
func BenchmarkToLongV4(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
iputils.ToLong("192.168.2.100")
|
|
}
|
|
}
|
|
|
|
func BenchmarkToLongV6(b *testing.B) {
|
|
runtime.GOMAXPROCS(1)
|
|
b.ReportAllocs()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
iputils.ToLong("fd00:6868:6868:0:10ac:d056:3bf6:7452")
|
|
}
|
|
}
|