fiber/utils/ips_test.go

92 lines
2.4 KiB
Go

// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
// 🤖 Github Repository: https://github.com/gofiber/fiber
// 📌 API Documentation: https://docs.gofiber.io
package utils
import (
"net"
"testing"
)
func Test_IsIPv4(t *testing.T) {
t.Parallel()
AssertEqual(t, true, IsIPv4("174.23.33.100"))
AssertEqual(t, true, IsIPv4("127.0.0.1"))
AssertEqual(t, true, IsIPv4("127.255.255.255"))
AssertEqual(t, true, IsIPv4("0.0.0.0"))
AssertEqual(t, false, IsIPv4(".0.0.0"))
AssertEqual(t, false, IsIPv4("0.0.0."))
AssertEqual(t, false, IsIPv4("0.0.0"))
AssertEqual(t, false, IsIPv4(".0.0.0."))
AssertEqual(t, false, IsIPv4("0.0.0.0.0"))
AssertEqual(t, false, IsIPv4("0"))
AssertEqual(t, false, IsIPv4(""))
AssertEqual(t, false, IsIPv4("2345:0425:2CA1::0567:5673:23b5"))
AssertEqual(t, false, IsIPv4("invalid"))
AssertEqual(t, false, IsIPv4("189.12.34.260"))
AssertEqual(t, false, IsIPv4("189.12.260.260"))
AssertEqual(t, false, IsIPv4("189.260.260.260"))
AssertEqual(t, false, IsIPv4("999.999.999.999"))
AssertEqual(t, false, IsIPv4("9999.9999.9999.9999"))
}
// go test -v -run=^$ -bench=UnsafeString -benchmem -count=2
func Benchmark_IsIPv4(b *testing.B) {
ip := "174.23.33.100"
var res bool
b.Run("fiber", func(b *testing.B) {
for n := 0; n < b.N; n++ {
res = IsIPv4(ip)
}
AssertEqual(b, true, res)
})
b.Run("default", func(b *testing.B) {
for n := 0; n < b.N; n++ {
res = net.ParseIP(ip) != nil
}
AssertEqual(b, true, res)
})
}
func Test_IsIPv6(t *testing.T) {
t.Parallel()
AssertEqual(t, true, IsIPv6("9396:9549:b4f7:8ed0:4791:1330:8c06:e62d"))
AssertEqual(t, true, IsIPv6("2345:0425:2CA1::0567:5673:23b5"))
AssertEqual(t, true, IsIPv6("2001:1:2:3:4:5:6:7"))
AssertEqual(t, false, IsIPv6("1.1.1.1"))
AssertEqual(t, false, IsIPv6("2001:1:2:3:4:5:6:"))
AssertEqual(t, false, IsIPv6(":1:2:3:4:5:6:"))
AssertEqual(t, false, IsIPv6("1:2:3:4:5:6:"))
AssertEqual(t, false, IsIPv6(""))
AssertEqual(t, false, IsIPv6("invalid"))
}
// go test -v -run=^$ -bench=UnsafeString -benchmem -count=2
func Benchmark_IsIPv6(b *testing.B) {
ip := "9396:9549:b4f7:8ed0:4791:1330:8c06:e62d"
var res bool
b.Run("fiber", func(b *testing.B) {
for n := 0; n < b.N; n++ {
res = IsIPv6(ip)
}
AssertEqual(b, true, res)
})
b.Run("default", func(b *testing.B) {
for n := 0; n < b.N; n++ {
res = net.ParseIP(ip) != nil
}
AssertEqual(b, true, res)
})
}