package network import ( "net" "reflect" "testing" ) func TestNewIPAMDefaultRoute(t *testing.T) { tests := []struct { name string isIPv6 bool want IPAMRoute }{ { name: "IPv4 default route", isIPv6: false, want: IPAMRoute{defaultIPv4Route}, }, { name: "IPv6 default route", isIPv6: true, want: IPAMRoute{defaultIPv6Route}, }, } for _, tt := range tests { tt := tt t.Run(tt.name, func(t *testing.T) { got, err := NewIPAMDefaultRoute(tt.isIPv6) if err != nil { t.Errorf("no error expected: %v", err) } if !reflect.DeepEqual(got, tt.want) { t.Errorf("NewIPAMDefaultRoute() = %v, want %v", got, tt.want) } }) } } func TestNewIPAMLocalHostRange(t *testing.T) { tests := []struct { name string subnet *net.IPNet ipRange *net.IPNet gw net.IP want []IPAMLocalHostRangeConf }{ { name: "IPv4 subnet", subnet: &net.IPNet{IP: net.IPv4(192, 168, 0, 0), Mask: net.IPv4Mask(255, 255, 255, 0)}, want: []IPAMLocalHostRangeConf{ { Subnet: "192.168.0.0/24", Gateway: "192.168.0.1", }, }, }, { name: "IPv4 subnet, range and gateway", subnet: &net.IPNet{IP: net.IPv4(192, 168, 0, 0), Mask: net.IPv4Mask(255, 255, 255, 0)}, ipRange: &net.IPNet{IP: net.IPv4(192, 168, 0, 128), Mask: net.IPv4Mask(255, 255, 255, 128)}, gw: net.ParseIP("192.168.0.10"), want: []IPAMLocalHostRangeConf{ { Subnet: "192.168.0.0/24", RangeStart: "192.168.0.129", RangeEnd: "192.168.0.255", Gateway: "192.168.0.10", }, }, }, { name: "IPv6 subnet", subnet: &net.IPNet{IP: net.ParseIP("2001:DB8::"), Mask: net.IPMask(net.ParseIP("ffff:ffff:ffff::"))}, want: []IPAMLocalHostRangeConf{ { Subnet: "2001:db8::/48", Gateway: "2001:db8::1", }, }, }, { name: "IPv6 subnet, range and gateway", subnet: &net.IPNet{IP: net.ParseIP("2001:DB8::"), Mask: net.IPMask(net.ParseIP("ffff:ffff:ffff::"))}, ipRange: &net.IPNet{IP: net.ParseIP("2001:DB8:1:1::"), Mask: net.IPMask(net.ParseIP("ffff:ffff:ffff:ffff::"))}, gw: net.ParseIP("2001:DB8::2"), want: []IPAMLocalHostRangeConf{ { Subnet: "2001:db8::/48", RangeStart: "2001:db8:1:1::1", RangeEnd: "2001:db8:1:1:ffff:ffff:ffff:ffff", Gateway: "2001:db8::2", }, }, }, } for _, tt := range tests { tt := tt t.Run(tt.name, func(t *testing.T) { got, err := NewIPAMLocalHostRange(tt.subnet, tt.ipRange, tt.gw) if err != nil { t.Errorf("no error expected: %v", err) } if !reflect.DeepEqual(got, tt.want) { t.Errorf("NewIPAMLocalHostRange() = %v, want %v", got, tt.want) } }) } }