3c59ef247f
The netip types can be used as map keys, unlike net.IP and friends, which is a very useful property to have for this application. Signed-off-by: Cory Snider <csnider@mirantis.com>
70 lines
2.4 KiB
Go
70 lines
2.4 KiB
Go
package ipbits
|
|
|
|
import (
|
|
"net/netip"
|
|
"testing"
|
|
)
|
|
|
|
func TestAdd(t *testing.T) {
|
|
tests := []struct {
|
|
in netip.Addr
|
|
x uint64
|
|
shift uint
|
|
want netip.Addr
|
|
}{
|
|
{netip.MustParseAddr("10.0.0.1"), 0, 0, netip.MustParseAddr("10.0.0.1")},
|
|
{netip.MustParseAddr("10.0.0.1"), 41, 0, netip.MustParseAddr("10.0.0.42")},
|
|
{netip.MustParseAddr("10.0.0.1"), 42, 16, netip.MustParseAddr("10.42.0.1")},
|
|
{netip.MustParseAddr("10.0.0.1"), 1, 7, netip.MustParseAddr("10.0.0.129")},
|
|
{netip.MustParseAddr("10.0.0.1"), 1, 24, netip.MustParseAddr("11.0.0.1")},
|
|
{netip.MustParseAddr("2001::1"), 0, 0, netip.MustParseAddr("2001::1")},
|
|
{netip.MustParseAddr("2001::1"), 0x41, 0, netip.MustParseAddr("2001::42")},
|
|
{netip.MustParseAddr("2001::1"), 1, 7, netip.MustParseAddr("2001::81")},
|
|
{netip.MustParseAddr("2001::1"), 0xcafe, 96, netip.MustParseAddr("2001:cafe::1")},
|
|
{netip.MustParseAddr("2001::1"), 1, 112, netip.MustParseAddr("2002::1")},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
if got := Add(tt.in, tt.x, tt.shift); tt.want != got {
|
|
t.Errorf("%v + (%v << %v) = %v; want %v", tt.in, tt.x, tt.shift, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkAdd(b *testing.B) {
|
|
do := func(b *testing.B, addr netip.Addr) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = Add(addr, uint64(i), 0)
|
|
}
|
|
}
|
|
|
|
b.Run("IPv4", func(b *testing.B) { do(b, netip.IPv4Unspecified()) })
|
|
b.Run("IPv6", func(b *testing.B) { do(b, netip.IPv6Unspecified()) })
|
|
}
|
|
|
|
func TestField(t *testing.T) {
|
|
tests := []struct {
|
|
in netip.Addr
|
|
u, v uint
|
|
want uint64
|
|
}{
|
|
{netip.MustParseAddr("1.2.3.4"), 0, 8, 1},
|
|
{netip.MustParseAddr("1.2.3.4"), 8, 16, 2},
|
|
{netip.MustParseAddr("1.2.3.4"), 16, 24, 3},
|
|
{netip.MustParseAddr("1.2.3.4"), 24, 32, 4},
|
|
{netip.MustParseAddr("1.2.3.4"), 0, 32, 0x01020304},
|
|
{netip.MustParseAddr("1.2.3.4"), 0, 28, 0x102030},
|
|
{netip.MustParseAddr("1234:5678:9abc:def0::7654:3210"), 0, 8, 0x12},
|
|
{netip.MustParseAddr("1234:5678:9abc:def0::7654:3210"), 8, 16, 0x34},
|
|
{netip.MustParseAddr("1234:5678:9abc:def0::7654:3210"), 16, 24, 0x56},
|
|
{netip.MustParseAddr("1234:5678:9abc:def0::7654:3210"), 64, 128, 0x76543210},
|
|
{netip.MustParseAddr("1234:5678:9abc:def0:beef::7654:3210"), 48, 80, 0xdef0beef},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
if got := Field(tt.in, tt.u, tt.v); got != tt.want {
|
|
t.Errorf("Field(%v, %v, %v) = %v (0x%[4]x); want %v (0x%[5]x)", tt.in, tt.u, tt.v, got, tt.want)
|
|
}
|
|
}
|
|
}
|