1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142
|
// ⚡️ 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"
"github.com/stretchr/testify/require"
)
func Test_IsIPv4(t *testing.T) {
t.Parallel()
require.True(t, IsIPv4("255.255.255.255"))
require.True(t, IsIPv4("174.23.33.100"))
require.True(t, IsIPv4("127.0.0.1"))
require.True(t, IsIPv4("0.0.0.0"))
require.True(t, IsIPv4("1.1.1.1"))
require.False(t, IsIPv4(".0.0.0"))
require.False(t, IsIPv4("0.0.0."))
require.False(t, IsIPv4("0.0.0"))
require.False(t, IsIPv4(".0.0.0."))
require.False(t, IsIPv4("0.0.0.0.0"))
require.False(t, IsIPv4("0"))
require.False(t, IsIPv4(""))
require.False(t, IsIPv4("2345:0425:2CA1::0567:5673:23b5"))
require.False(t, IsIPv4("invalid"))
require.False(t, IsIPv4("189.12.34.260"))
require.False(t, IsIPv4("189.12.260.260"))
require.False(t, IsIPv4("189.260.260.260"))
require.False(t, IsIPv4("255.255.255.256"))
require.False(t, IsIPv4("999.999.999.999"))
require.False(t, IsIPv4("9999.9999.9999.9999"))
require.False(t, IsIPv4("192168.1.1"))
require.False(t, IsIPv4("192.1681.1"))
require.False(t, IsIPv4("192a168.1.1"))
}
func Test_IsIPv6(t *testing.T) {
t.Parallel()
// Valid Cases
require.True(t, IsIPv6("9396:9549:b4f7:8ed0:4791:1330:8c06:e62d"))
require.True(t, IsIPv6("2345:0425:2CA1::0567:5673:23b5"))
require.True(t, IsIPv6("2001:1:2:3:4:5:6:7"))
require.True(t, IsIPv6("::1"), "IPv6 loopback address with leading ellipsis")
require.True(t, IsIPv6("1::"), "Shortest possible IPv6 address with trailing ellipsis")
require.True(t, IsIPv6("2001:db8::8a2e:370:7334"), "Valid IPv6 address with ellipsis in the middle")
require.True(t, IsIPv6("::ffff:192.0.2.128"), "IPv4-mapped IPv6 address")
require.True(t, IsIPv6("::192.168.1.1"), "IPv4-compatible IPv6 address")
require.True(t, IsIPv6("ffff::"), "Address with a single non-zero group and leading ellipsis")
// Boundary values
require.True(t, IsIPv6("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"), "Maximum value for all groups")
// Invalid cases
require.False(t, IsIPv6("1::2::3"), "IPv6 address with multiple ellipses")
require.False(t, IsIPv6("2001::25de::cade"), "IPv6 address with multiple ellipses in the middle")
require.False(t, IsIPv6("g:a:b:c:d:e:f:1"), "Invalid character 'g' in address")
require.False(t, IsIPv6("::192.168.1.256"), "Invalid IPv4 segment in IPv6 address")
require.False(t, IsIPv6("1:2:3:4:5:6:7:8:9"), "IPv6 address too long")
require.False(t, IsIPv6("1:2:3"), "IPv6 address too short")
require.False(t, IsIPv6("1:::3:4"), "Consecutive colons not part of a valid ellipsis")
require.False(t, IsIPv6("::ffff:256.0.0.1"), "Embedded IPv4 address with invalid IPv4 segment")
require.False(t, IsIPv6("1111:2222:3333:4444:5555:6666:7777:8888:9999"), "IPv6 address with too many groups")
require.False(t, IsIPv6("2001:db8::8a2e:370g:7334"), "IPv6 address with invalid hex character 'g'")
require.False(t, IsIPv6("1.1.1.1"))
require.False(t, IsIPv6("2001:1:2:3:4:5:6:"))
require.False(t, IsIPv6(":1:2:3:4:5:6:"))
require.False(t, IsIPv6("1:2:3:4:5:6:"))
require.False(t, IsIPv6(""))
require.False(t, IsIPv6("invalid"))
}
func Test_IsIPv6_EdgeCases(t *testing.T) {
t.Parallel()
// hex group larger than 4 digits
require.False(t, IsIPv6("1:2:3:4:5:6:7:12345"))
// IPv4 part at wrong position
require.False(t, IsIPv6("1:2:3:4:1.2.3.4"))
// IPv4 part would exceed length
require.False(t, IsIPv6("1:2:3:4:5:6:7:1.2.3.4"))
// ellipsis present but no zero group
require.False(t, IsIPv6("1::2:3:4:5:6:7:8"))
}
func Test_IPWhitespace(t *testing.T) {
t.Parallel()
require.False(t, IsIPv4(" 1.1.1.1"))
require.False(t, IsIPv4("1.1.1.1 "))
require.False(t, IsIPv6(" ::1"))
require.False(t, IsIPv6("::1 "))
}
// go test -v -run=^$ -bench=Benchmark_IsIPv4 -benchmem -count=6
func Benchmark_IsIPv4(b *testing.B) {
ip := "174.23.33.100"
var res bool
b.Run("fiber", func(b *testing.B) {
for i := 0; i < b.N; i++ {
res = IsIPv4(ip)
}
require.True(b, res)
})
b.Run("default", func(b *testing.B) {
for i := 0; i < b.N; i++ {
res = net.ParseIP(ip) != nil
}
require.True(b, res)
})
}
// go test -v -run=^$ -bench=Benchmark_IsIPv6 -benchmem -count=6
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 i := 0; i < b.N; i++ {
res = IsIPv6(ip)
}
require.True(b, res)
})
b.Run("default", func(b *testing.B) {
for i := 0; i < b.N; i++ {
res = net.ParseIP(ip) != nil
}
require.True(b, res)
})
}
|