flannel/pkg/ip/ipnet_test.go

package ip

import (
	"encoding/json"
	"net"
	"testing"
)

func mkIP4Net(s string, plen uint) IP4Net {
	ip, err := ParseIP4(s)
	if err != nil {
		panic(err)
	}
	return IP4Net{ip, plen}
}

func mkIP4(s string) IP4 {
	ip, err := ParseIP4(s)
	if err != nil {
		panic(err)
	}
	return ip
}

func TestIP4(t *testing.T) {
	nip := net.ParseIP("1.2.3.4")
	ip := FromIP(nip)
	a, b, c, d := ip.Octets()
	if a != 1 || b != 2 || c != 3 || d != 4 {
		t.Error("FromIP failed")
	}

	ip, err := ParseIP4("1.2.3.4")
	if err != nil {
		t.Error("ParseIP4 failed with: ", err)
	} else {
		a, b, c, d := ip.Octets()
		if a != 1 || b != 2 || c != 3 || d != 4 {
			t.Error("ParseIP4 failed")
		}
	}

	if ip.ToIP().String() != "1.2.3.4" {
		t.Error("ToIP failed")
	}

	if ip.String() != "1.2.3.4" {
		t.Error("String failed")
	}

	if ip.StringSep("*") != "1*2*3*4" {
		t.Error("StringSep failed")
	}

	j, err := json.Marshal(ip)
	if err != nil {
		t.Error("Marshal of IP4 failed: ", err)
	} else if string(j) != `"1.2.3.4"` {
		t.Error("Marshal of IP4 failed with unexpected value: ", j)
	}
}

func TestIP4Net(t *testing.T) {
	n1 := mkIP4Net("1.2.3.0", 24)

	if n1.ToIPNet().String() != "1.2.3.0/24" {
		t.Error("ToIPNet failed")
	}

	if !n1.Overlaps(n1) {
		t.Errorf("%s does not overlap %s", n1, n1)
	}

	n2 := mkIP4Net("1.2.0.0", 16)
	if !n1.Overlaps(n2) {
		t.Errorf("%s does not overlap %s", n1, n2)
	}

	n2 = mkIP4Net("1.2.4.0", 24)
	if n1.Overlaps(n2) {
		t.Errorf("%s overlaps %s", n1, n2)
	}

	n2 = mkIP4Net("7.2.4.0", 22)
	if n1.Overlaps(n2) {
		t.Errorf("%s overlaps %s", n1, n2)
	}

	if !n1.Contains(mkIP4("1.2.3.0")) {
		t.Error("Contains failed")
	}

	if !n1.Contains(mkIP4("1.2.3.4")) {
		t.Error("Contains failed")
	}

	if n1.Contains(mkIP4("1.2.4.0")) {
		t.Error("Contains failed")
	}

	j, err := json.Marshal(n1)
	if err != nil {
		t.Error("Marshal of IP4Net failed: ", err)
	} else if string(j) != `"1.2.3.0/24"` {
		t.Error("Marshal of IP4Net failed with unexpected value: ", j)
	}
}