| package docker |
| |
| import ( |
| "net" |
| "testing" |
| ) |
| |
| func TestNetworkRange(t *testing.T) { |
| // Simple class C test |
| _, network, _ := net.ParseCIDR("192.168.0.1/24") |
| first, last := networkRange(network) |
| if !first.Equal(net.ParseIP("192.168.0.0")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("192.168.0.255")) { |
| t.Error(last.String()) |
| } |
| if size, err := networkSize(network.Mask); err != nil || size != 256 { |
| t.Error(size, err) |
| } |
| |
| // Class A test |
| _, network, _ = net.ParseCIDR("10.0.0.1/8") |
| first, last = networkRange(network) |
| if !first.Equal(net.ParseIP("10.0.0.0")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("10.255.255.255")) { |
| t.Error(last.String()) |
| } |
| if size, err := networkSize(network.Mask); err != nil || size != 16777216 { |
| t.Error(size, err) |
| } |
| |
| // Class A, random IP address |
| _, network, _ = net.ParseCIDR("10.1.2.3/8") |
| first, last = networkRange(network) |
| if !first.Equal(net.ParseIP("10.0.0.0")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("10.255.255.255")) { |
| t.Error(last.String()) |
| } |
| |
| // 32bit mask |
| _, network, _ = net.ParseCIDR("10.1.2.3/32") |
| first, last = networkRange(network) |
| if !first.Equal(net.ParseIP("10.1.2.3")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("10.1.2.3")) { |
| t.Error(last.String()) |
| } |
| if size, err := networkSize(network.Mask); err != nil || size != 1 { |
| t.Error(size, err) |
| } |
| |
| // 31bit mask |
| _, network, _ = net.ParseCIDR("10.1.2.3/31") |
| first, last = networkRange(network) |
| if !first.Equal(net.ParseIP("10.1.2.2")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("10.1.2.3")) { |
| t.Error(last.String()) |
| } |
| if size, err := networkSize(network.Mask); err != nil || size != 2 { |
| t.Error(size, err) |
| } |
| |
| // 26bit mask |
| _, network, _ = net.ParseCIDR("10.1.2.3/26") |
| first, last = networkRange(network) |
| if !first.Equal(net.ParseIP("10.1.2.0")) { |
| t.Error(first.String()) |
| } |
| if !last.Equal(net.ParseIP("10.1.2.63")) { |
| t.Error(last.String()) |
| } |
| if size, err := networkSize(network.Mask); err != nil || size != 64 { |
| t.Error(size, err) |
| } |
| } |
| |
| func TestConversion(t *testing.T) { |
| ip := net.ParseIP("127.0.0.1") |
| i, err := ipToInt(ip) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if i == 0 { |
| t.Fatal("converted to zero") |
| } |
| conv, err := intToIp(i) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if !ip.Equal(conv) { |
| t.Error(conv.String()) |
| } |
| } |
| |
| func TestIPAllocator(t *testing.T) { |
| gwIP, n, _ := net.ParseCIDR("127.0.0.1/29") |
| alloc, err := newIPAllocator(&net.IPNet{IP: gwIP, Mask: n.Mask}) |
| if err != nil { |
| t.Fatal(err) |
| } |
| var lastIP net.IP |
| for i := 0; i < 5; i++ { |
| ip, err := alloc.Acquire() |
| if err != nil { |
| t.Fatal(err) |
| } |
| lastIP = ip |
| } |
| ip, err := alloc.Acquire() |
| if err == nil { |
| t.Fatal("There shouldn't be any IP addresses at this point") |
| } |
| // Release 1 IP |
| alloc.Release(lastIP) |
| ip, err = alloc.Acquire() |
| if err != nil { |
| t.Fatal(err) |
| } |
| if !ip.Equal(lastIP) { |
| t.Fatal(ip.String()) |
| } |
| } |