2014-01-22 21:05:20 -05:00
|
|
|
|
package ipallocator
|
|
|
|
|
|
|
|
|
|
import (
|
2014-01-22 22:34:47 -05:00
|
|
|
|
"fmt"
|
2014-01-22 21:05:20 -05:00
|
|
|
|
"net"
|
|
|
|
|
"testing"
|
|
|
|
|
)
|
|
|
|
|
|
2014-01-22 22:34:47 -05:00
|
|
|
|
func reset() {
|
|
|
|
|
allocatedIPs = networkSet{}
|
|
|
|
|
availableIPS = networkSet{}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestRequestNewIps(t *testing.T) {
|
|
|
|
|
defer reset()
|
|
|
|
|
network := &net.IPNet{
|
|
|
|
|
IP: []byte{192, 168, 0, 1},
|
|
|
|
|
Mask: []byte{255, 255, 255, 0},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for i := 2; i < 10; i++ {
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if expected := fmt.Sprintf("192.168.0.%d", i); ip.String() != expected {
|
|
|
|
|
t.Fatalf("Expected ip %s got %s", expected, ip.String())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestReleaseIp(t *testing.T) {
|
|
|
|
|
defer reset()
|
|
|
|
|
network := &net.IPNet{
|
|
|
|
|
IP: []byte{192, 168, 0, 1},
|
|
|
|
|
Mask: []byte{255, 255, 255, 0},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := ReleaseIP(network, ip); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestGetReleasedIp(t *testing.T) {
|
|
|
|
|
defer reset()
|
|
|
|
|
network := &net.IPNet{
|
|
|
|
|
IP: []byte{192, 168, 0, 1},
|
|
|
|
|
Mask: []byte{255, 255, 255, 0},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
value := ip.String()
|
|
|
|
|
if err := ReleaseIP(network, ip); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ip, err = RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if ip.String() != value {
|
|
|
|
|
t.Fatalf("Expected to receive same ip %s got %s", value, ip.String())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestRequesetSpecificIp(t *testing.T) {
|
|
|
|
|
defer reset()
|
|
|
|
|
network := &net.IPNet{
|
|
|
|
|
IP: []byte{192, 168, 0, 1},
|
|
|
|
|
Mask: []byte{255, 255, 255, 0},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ip := net.ParseIP("192.168.1.5")
|
|
|
|
|
|
|
|
|
|
if _, err := RequestIP(network, &ip); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
}
|
2014-01-23 04:31:38 -05:00
|
|
|
|
|
|
|
|
|
func TestConversion(t *testing.T) {
|
|
|
|
|
ip := net.ParseIP("127.0.0.1")
|
|
|
|
|
i := ipToInt(&ip)
|
|
|
|
|
if i == 0 {
|
|
|
|
|
t.Fatal("converted to zero")
|
|
|
|
|
}
|
|
|
|
|
conv := intToIP(i)
|
|
|
|
|
if !ip.Equal(*conv) {
|
|
|
|
|
t.Error(conv.String())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestIPAllocator(t *testing.T) {
|
|
|
|
|
expectedIPs := []net.IP{
|
|
|
|
|
0: net.IPv4(127, 0, 0, 2),
|
|
|
|
|
1: net.IPv4(127, 0, 0, 3),
|
|
|
|
|
2: net.IPv4(127, 0, 0, 4),
|
|
|
|
|
3: net.IPv4(127, 0, 0, 5),
|
|
|
|
|
4: net.IPv4(127, 0, 0, 6),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
gwIP, n, _ := net.ParseCIDR("127.0.0.1/29")
|
|
|
|
|
network := &net.IPNet{IP: gwIP, Mask: n.Mask}
|
|
|
|
|
// Pool after initialisation (f = free, u = used)
|
|
|
|
|
// 2(f) - 3(f) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// Check that we get 5 IPs, from 127.0.0.2–127.0.0.6, in that
|
|
|
|
|
// order.
|
|
|
|
|
for i := 0; i < 5; i++ {
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
assertIPEquals(t, &expectedIPs[i], ip)
|
|
|
|
|
}
|
|
|
|
|
// Before loop begin
|
|
|
|
|
// 2(f) - 3(f) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 0
|
|
|
|
|
// 2(u) - 3(f) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 1
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 2
|
|
|
|
|
// 2(u) - 3(u) - 4(u) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 3
|
|
|
|
|
// 2(u) - 3(u) - 4(u) - 5(u) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 4
|
|
|
|
|
// 2(u) - 3(u) - 4(u) - 5(u) - 6(u)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// Check that there are no more IPs
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err == nil {
|
|
|
|
|
t.Fatalf("There shouldn't be any IP addresses at this point, got %s\n", ip)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Release some IPs in non-sequential order
|
|
|
|
|
if err := ReleaseIP(network, &expectedIPs[3]); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
// 2(u) - 3(u) - 4(u) - 5(f) - 6(u)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
if err := ReleaseIP(network, &expectedIPs[2]); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(f) - 6(u)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
if err := ReleaseIP(network, &expectedIPs[4]); err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// Make sure that IPs are reused in sequential order, starting
|
|
|
|
|
// with the first released IP
|
|
|
|
|
newIPs := make([]*net.IP, 3)
|
|
|
|
|
for i := 0; i < 3; i++ {
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
newIPs[i] = ip
|
|
|
|
|
}
|
|
|
|
|
// Before loop begin
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(f) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 0
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(u) - 6(f)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 1
|
|
|
|
|
// 2(u) - 3(u) - 4(f) - 5(u) - 6(u)
|
|
|
|
|
// ↑
|
|
|
|
|
|
|
|
|
|
// After i = 2
|
|
|
|
|
// 2(u) - 3(u) - 4(u) - 5(u) - 6(u)
|
|
|
|
|
// ↑
|
|
|
|
|
|
2014-01-23 08:22:32 -05:00
|
|
|
|
// Reordered these because the new set will always return the
|
|
|
|
|
// lowest ips first and not in the order that they were released
|
2014-01-23 05:22:13 -05:00
|
|
|
|
assertIPEquals(t, &expectedIPs[2], newIPs[0])
|
|
|
|
|
assertIPEquals(t, &expectedIPs[3], newIPs[1])
|
|
|
|
|
assertIPEquals(t, &expectedIPs[4], newIPs[2])
|
2014-01-23 04:31:38 -05:00
|
|
|
|
|
|
|
|
|
_, err = RequestIP(network, nil)
|
|
|
|
|
if err == nil {
|
|
|
|
|
t.Fatal("There shouldn't be any IP addresses at this point")
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-01-31 02:05:16 -05:00
|
|
|
|
func TestAllocateFirstIP(t *testing.T) {
|
|
|
|
|
defer reset()
|
|
|
|
|
network := &net.IPNet{
|
|
|
|
|
IP: []byte{192, 168, 0, 0},
|
|
|
|
|
Mask: []byte{255, 255, 255, 0},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
firstIP := network.IP.To4().Mask(network.Mask)
|
|
|
|
|
first := ipToInt(&firstIP) + 1
|
|
|
|
|
|
|
|
|
|
ip, err := RequestIP(network, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
allocated := ipToInt(ip)
|
|
|
|
|
|
|
|
|
|
if allocated == first {
|
|
|
|
|
t.Fatalf("allocated ip should not equal first ip: %d == %d", first, allocated)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-01-23 04:31:38 -05:00
|
|
|
|
func assertIPEquals(t *testing.T, ip1, ip2 *net.IP) {
|
|
|
|
|
if !ip1.Equal(*ip2) {
|
|
|
|
|
t.Fatalf("Expected IP %s, got %s", ip1, ip2)
|
|
|
|
|
}
|
|
|
|
|
}
|