1 files changed, 70 insertions, 0 deletions
diff --git a/libpod/network/netconflist_test.go b/libpod/network/netconflist_test.go
index 5893bf985..6bf1a9777 100644
--- a/libpod/network/netconflist_test.go
+++ b/libpod/network/netconflist_test.go
@@ -1,6 +1,7 @@
 package network
 
 import (
+	"net"
 	"reflect"
 	"testing"
 )
@@ -36,3 +37,72 @@ func TestNewIPAMDefaultRoute(t *testing.T) {
 		})
 	}
 }
+
+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",
+				},
+			},
+		},
+		{
+			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",
+				},
+			},
+		},
+		{
+			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)
+			}
+		})
+	}
+}