warewulf/internal/pkg/node/methods_test.go

package node

import (
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_IpCIDR(t *testing.T) {
	tests := map[string]struct {
		ipaddr  net.IP
		netmask net.IP
		cidr    string
	}{
		"nil": {
			ipaddr:  nil,
			netmask: nil,
			cidr:    "",
		},
		"ip only": {
			ipaddr:  net.ParseIP("192.168.1.1"),
			netmask: nil,
			cidr:    "",
		},
		"netmask only": {
			ipaddr:  nil,
			netmask: net.ParseIP("255.255.255.0"),
			cidr:    "",
		},
		"working": {
			ipaddr:  net.ParseIP("192.168.1.1"),
			netmask: net.ParseIP("255.255.255.0"),
			cidr:    "192.168.1.1/24",
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			n := new(NetDev)
			n.Ipaddr = tt.ipaddr
			n.Netmask = tt.netmask
			assert.Equal(t, tt.cidr, n.IpCIDR())
		})
	}
}

func Test_IpCIDR6(t *testing.T) {
	tests := map[string]struct {
		ipaddr net.IP
		prefix string
		cidr   string
	}{
		"nil": {
			ipaddr: nil,
			prefix: "",
			cidr:   "",
		},
		"ip only": {
			ipaddr: net.ParseIP("fd00:10::1"),
			prefix: "",
			cidr:   "",
		},
		"netmask only": {
			ipaddr: nil,
			prefix: "64",
			cidr:   "",
		},
		"ipv4": {
			ipaddr: net.ParseIP("10.0.0.1"),
			prefix: "64",
			cidr:   "",
		},
		"invalid prefix type": {
			ipaddr: net.ParseIP("fd00:10::1"),
			prefix: "string",
			cidr:   "",
		},
		"invalid prefix": {
			ipaddr: net.ParseIP("fd00:10::1"),
			prefix: "129",
			cidr:   "",
		},
		"working": {
			ipaddr: net.ParseIP("fd00:10::1"),
			prefix: "64",
			cidr:   "fd00:10::1/64",
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			n := new(NetDev)
			n.Ipaddr6 = tt.ipaddr
			n.PrefixLen6 = tt.prefix
			assert.Equal(t, tt.cidr, n.IpCIDR6())
		})
	}
}

func Test_Empty(t *testing.T) {
	var netdev NetDev
	var netdevPtr *NetDev

	t.Run("test for empty", func(t *testing.T) {
		if ObjectIsEmpty(netdev) != true {
			t.Errorf("netdev must be empty")
		}
	})
	t.Run("test for non empty", func(t *testing.T) {
		netdev.Device = "foo"
		if ObjectIsEmpty(netdev) == true {
			t.Errorf("netdev must be non empty")
		}
	})
	t.Run("test for nil pointer", func(t *testing.T) {
		if ObjectIsEmpty(netdevPtr) != true {
			t.Errorf("netdev must be empty")
		}
	})
	t.Run("test for pointer assigned", func(t *testing.T) {
		netdev.Ipaddr = net.ParseIP("10.10.10.1")
		netdevPtr = &netdev
		if ObjectIsEmpty(netdevPtr) == true {
			t.Errorf("netdev must be empty")
		}
	})
}

func Test_Node_Expand_Flatten(t *testing.T) {
	node := new(Node)

	assert.Nil(t, node.Ipmi)
	assert.Nil(t, node.Kernel)
	assert.Nil(t, node.NetDevs)
	assert.Nil(t, node.Tags)

	t.Run("test expand", func(t *testing.T) {
		node.Expand()
		assert.Equal(t, map[string]string{}, node.Tags)
		assert.Equal(t, map[string]string{}, node.Ipmi.Tags)
		assert.Equal(t, "", node.Kernel.Version)
		assert.Len(t, node.Kernel.Args, 0)
		assert.Equal(t, map[string]*NetDev{}, node.NetDevs)
	})

	t.Run("test flatten", func(t *testing.T) {
		node.Flatten()
		assert.Nil(t, node.Ipmi)
		assert.Nil(t, node.Kernel)
		assert.Nil(t, node.NetDevs)
		assert.Nil(t, node.Tags)
	})
}

func Test_Profile_Expand_Flatten(t *testing.T) {
	profile := new(Profile)

	assert.Nil(t, profile.Ipmi)
	assert.Nil(t, profile.Kernel)
	assert.Nil(t, profile.NetDevs)
	assert.Nil(t, profile.Tags)

	t.Run("test expand", func(t *testing.T) {
		profile.Expand()
		assert.Equal(t, map[string]string{}, profile.Tags)
		assert.Equal(t, map[string]string{}, profile.Ipmi.Tags)
		assert.Equal(t, "", profile.Kernel.Version)
		assert.Len(t, profile.Kernel.Args, 0)
		assert.Equal(t, map[string]*NetDev{}, profile.NetDevs)
	})

	t.Run("test flatten", func(t *testing.T) {
		profile.Flatten()
		assert.Nil(t, profile.Ipmi)
		assert.Nil(t, profile.Kernel)
		assert.Nil(t, profile.NetDevs)
		assert.Nil(t, profile.Tags)
	})
}