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
|
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)
}
})
}
}
|
