118 lines
3.5 KiB
Go
118 lines
3.5 KiB
Go
package ipamutils
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"gotest.tools/v3/assert"
|
|
is "gotest.tools/v3/assert/cmp"
|
|
)
|
|
|
|
func initBroadPredefinedNetworks() []*net.IPNet {
|
|
pl := make([]*net.IPNet, 0, 31)
|
|
mask := []byte{255, 255, 0, 0}
|
|
for i := 17; i < 32; i++ {
|
|
pl = append(pl, &net.IPNet{IP: []byte{172, byte(i), 0, 0}, Mask: mask})
|
|
}
|
|
mask20 := []byte{255, 255, 240, 0}
|
|
for i := 0; i < 16; i++ {
|
|
pl = append(pl, &net.IPNet{IP: []byte{192, 168, byte(i << 4), 0}, Mask: mask20})
|
|
}
|
|
return pl
|
|
}
|
|
|
|
func initGranularPredefinedNetworks() []*net.IPNet {
|
|
pl := make([]*net.IPNet, 0, 256*256)
|
|
mask := []byte{255, 255, 255, 0}
|
|
for i := 0; i < 256; i++ {
|
|
for j := 0; j < 256; j++ {
|
|
pl = append(pl, &net.IPNet{IP: []byte{10, byte(i), byte(j), 0}, Mask: mask})
|
|
}
|
|
}
|
|
return pl
|
|
}
|
|
|
|
func initGlobalScopeNetworks() []*net.IPNet {
|
|
pl := make([]*net.IPNet, 0, 256*256)
|
|
mask := []byte{255, 255, 255, 0}
|
|
for i := 0; i < 256; i++ {
|
|
for j := 0; j < 256; j++ {
|
|
pl = append(pl, &net.IPNet{IP: []byte{30, byte(i), byte(j), 0}, Mask: mask})
|
|
}
|
|
}
|
|
return pl
|
|
}
|
|
|
|
func TestDefaultNetwork(t *testing.T) {
|
|
for _, nw := range PredefinedGlobalScopeDefaultNetworks {
|
|
if ones, bits := nw.Mask.Size(); bits != 32 || ones != 24 {
|
|
t.Fatalf("Unexpected size for network in granular list: %v", nw)
|
|
}
|
|
}
|
|
|
|
for _, nw := range PredefinedLocalScopeDefaultNetworks {
|
|
if ones, bits := nw.Mask.Size(); bits != 32 || (ones != 20 && ones != 16) {
|
|
t.Fatalf("Unexpected size for network in broad list: %v", nw)
|
|
}
|
|
}
|
|
|
|
originalBroadNets := initBroadPredefinedNetworks()
|
|
m := make(map[string]bool)
|
|
for _, v := range originalBroadNets {
|
|
m[v.String()] = true
|
|
}
|
|
for _, nw := range PredefinedLocalScopeDefaultNetworks {
|
|
_, ok := m[nw.String()]
|
|
assert.Check(t, ok)
|
|
delete(m, nw.String())
|
|
}
|
|
|
|
assert.Check(t, is.Len(m, 0))
|
|
|
|
originalGranularNets := initGranularPredefinedNetworks()
|
|
|
|
m = make(map[string]bool)
|
|
for _, v := range originalGranularNets {
|
|
m[v.String()] = true
|
|
}
|
|
for _, nw := range PredefinedGlobalScopeDefaultNetworks {
|
|
_, ok := m[nw.String()]
|
|
assert.Check(t, ok)
|
|
delete(m, nw.String())
|
|
}
|
|
|
|
assert.Check(t, is.Len(m, 0))
|
|
}
|
|
|
|
func TestConfigGlobalScopeDefaultNetworks(t *testing.T) {
|
|
err := ConfigGlobalScopeDefaultNetworks([]*NetworkToSplit{{"30.0.0.0/8", 24}})
|
|
assert.NilError(t, err)
|
|
|
|
originalGlobalScopeNetworks := initGlobalScopeNetworks()
|
|
m := make(map[string]bool)
|
|
for _, v := range originalGlobalScopeNetworks {
|
|
m[v.String()] = true
|
|
}
|
|
for _, nw := range PredefinedGlobalScopeDefaultNetworks {
|
|
_, ok := m[nw.String()]
|
|
assert.Check(t, ok)
|
|
delete(m, nw.String())
|
|
}
|
|
|
|
assert.Check(t, is.Len(m, 0))
|
|
}
|
|
|
|
func TestInitAddressPools(t *testing.T) {
|
|
err := ConfigLocalScopeDefaultNetworks([]*NetworkToSplit{{"172.80.0.0/16", 24}, {"172.90.0.0/16", 24}})
|
|
assert.NilError(t, err)
|
|
|
|
// Check for Random IPAddresses in PredefinedLocalScopeDefaultNetworks ex: first , last and middle
|
|
assert.Check(t, is.Len(PredefinedLocalScopeDefaultNetworks, 512), "Failed to find PredefinedLocalScopeDefaultNetworks")
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[0].String(), "172.80.0.0/24"))
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[127].String(), "172.80.127.0/24"))
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[255].String(), "172.80.255.0/24"))
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[256].String(), "172.90.0.0/24"))
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[383].String(), "172.90.127.0/24"))
|
|
assert.Check(t, is.Equal(PredefinedLocalScopeDefaultNetworks[511].String(), "172.90.255.0/24"))
|
|
}
|