tests: add util/net coverage

Signed-off-by: Francesco Giudici <[email protected]>

Author: fgiudici

pkg/util/net_test.go

@@ -0,0 +1,114 @@
+/*
+Copyright © 2022 - 2024 SUSE LLC
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package util
+
+import (
+	"bytes"
+	"fmt"
+	"net"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+
+	utilmocks "github.com/rancher/elemental-operator/pkg/util/mocks"
+	gomock "go.uber.org/mock/gomock"
+)
+
+var _ = Describe("GetIPByIfName", func() {
+	var ctrl *gomock.Controller
+	var netctrl *utilmocks.MockNetController
+
+	BeforeEach(func() {
+		ctrl = gomock.NewController(GinkgoT())
+		netctrl = utilmocks.NewMockNetController(ctrl)
+	})
+	It("should return IPv4 address if available", func() {
+		netctrl.EXPECT().GetInterfaceAddresses("eth0").Return(
+			[]net.IPNet{
+				{
+					IP: []byte{0x20, 0x1, 0x11, 0x11, 0x22, 0x22, 0x33, 0x33,
+						0x44, 0x44, 0x55, 0x55, 0x66, 0x66, 0x77, 0x77},
+				},
+				{
+					IP: []byte{192, 168, 1, 10},
+				},
+			},
+		)
+
+		ret := getIPByIfName("eth0", netctrl)
+		Expect(ret).To(Equal("192.168.1.10"))
+	})
+	It("should return IPv6 address if IPv4 is not available", func() {
+		netctrl.EXPECT().GetInterfaceAddresses("eth0").Return(
+			[]net.IPNet{
+				{
+					IP: []byte{0x20, 0x1, 0x11, 0x11, 0x22, 0x22, 0x33, 0x33,
+						0x44, 0x44, 0x55, 0x55, 0x66, 0x66, 0x77, 0x77},
+				},
+			},
+		)
+
+		ret := getIPByIfName("eth0", netctrl)
+		Expect(ret).To(Equal("2001:1111:2222:3333:4444:5555:6666:7777"))
+	})
+	It("should return the empty string when no IP addresses are available", func() {
+		netctrl.EXPECT().GetInterfaceAddresses("eth0").Return([]net.IPNet{})
+		ret := getIPByIfName("eth0", netctrl)
+		Expect(ret).To(BeEmpty())
+	})
+	It("should return the empty string if returned addresses are invalid", func() {
+		netctrl.EXPECT().GetInterfaceAddresses("eth0").Return(
+			[]net.IPNet{
+				{
+					IP: []byte{0x20, 0x1, 0x11, 0x11, 0x22, 0x22, 0x33, 0x33,
+						0x44, 0x44},
+				},
+			},
+		)
+		ret := getIPByIfName("eth0", netctrl)
+		Expect(ret).To(BeEmpty())
+	})
+})
+
+var _ = Describe("GetNMActivatedDevices", func() {
+	var ctrl *gomock.Controller
+	var netctrl *utilmocks.MockNetController
+
+	BeforeEach(func() {
+		ctrl = gomock.NewController(GinkgoT())
+		netctrl = utilmocks.NewMockNetController(ctrl)
+	})
+	It("should return only devices activated by NetworkManager", func() {
+		buf := &bytes.Buffer{}
+		buf.Write([]byte("eth0:connected\n"))
+		buf.Write([]byte("lo:connected (externally)\n"))
+		buf.Write([]byte("tun0:connected (externally)\n"))
+		buf.Write([]byte("garbage\n"))
+		buf.Write([]byte("mynic:connected\n"))
+		buf.Write([]byte("eth1:unmanaged\n"))
+		netctrl.EXPECT().GetNMDeviceState().Return(buf, nil)
+		ret := getNMActivatedDevices(netctrl)
+		Expect(ret).To(Equal([]string{"eth0", "mynic"}))
+	})
+	It("should return no devices if NetworkManager returns error", func() {
+		buf := &bytes.Buffer{}
+		buf.Write([]byte("eth0:connected\n"))
+		netctrl.EXPECT().GetNMDeviceState().Return(buf, fmt.Errorf("nmcli failed"))
+		ret := getNMActivatedDevices(netctrl)
+		Expect(ret).To(BeEmpty())
+	})
+})