55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package sa
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestIncrementIP(t *testing.T) {
|
|
testCases := []struct {
|
|
ip string
|
|
index int
|
|
expected string
|
|
}{
|
|
{"0.0.0.0", 128, "0.0.0.1"},
|
|
{"0.0.0.255", 128, "0.0.1.0"},
|
|
{"127.0.0.1", 128, "127.0.0.2"},
|
|
{"1.2.3.4", 120, "1.2.4.4"},
|
|
{"::1", 128, "::2"},
|
|
{"2002:1001:4008::", 128, "2002:1001:4008::1"},
|
|
{"2002:1001:4008::", 48, "2002:1001:4009::"},
|
|
{"2002:1001:ffff::", 48, "2002:1002::"},
|
|
{"ffff:ffff:ffff::", 48, "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"},
|
|
}
|
|
for _, tc := range testCases {
|
|
ip := net.ParseIP(tc.ip).To16()
|
|
actual := incrementIP(ip, tc.index)
|
|
expectedIP := net.ParseIP(tc.expected)
|
|
if !actual.Equal(expectedIP) {
|
|
t.Errorf("Expected incrementIP(%s, %d) to be %s, instead got %s",
|
|
tc.ip, tc.index, expectedIP, actual.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPRange(t *testing.T) {
|
|
testCases := []struct {
|
|
ip string
|
|
expectedBegin string
|
|
expectedEnd string
|
|
}{
|
|
{"28.45.45.28", "28.45.45.28", "28.45.45.29"},
|
|
{"2002:1001:4008::", "2002:1001:4008::", "2002:1001:4009::"},
|
|
}
|
|
for _, tc := range testCases {
|
|
ip := net.ParseIP(tc.ip)
|
|
expectedBegin := net.ParseIP(tc.expectedBegin)
|
|
expectedEnd := net.ParseIP(tc.expectedEnd)
|
|
actualBegin, actualEnd := ipRange(ip)
|
|
if !expectedBegin.Equal(actualBegin) || !expectedEnd.Equal(actualEnd) {
|
|
t.Errorf("Expected ipRange(%s) to be (%s, %s), got (%s, %s)",
|
|
tc.ip, tc.expectedBegin, tc.expectedEnd, actualBegin, actualEnd)
|
|
}
|
|
}
|
|
}
|