package ipamutils import ( "net" "testing" _ "github.com/docker/docker/libnetwork/testutils" "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")) }