go-network-configurator/networkScripts_test.go

package netconfig

import (
	"context"
	"net"
	"os"
	"path/filepath"
	"strings"
	"testing"
)

// Validate the network scripts configuration parser/writer functions.
func TestNetworkScripts(t *testing.T) {
	// Setup test file.
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		t.Fatal(err)
	}
	testDir, err := filepath.Abs("./tests/networkScripts")
	if err != nil {
		t.Fatal(err)
	}
	resultsDir := filepath.Join(testDir, "results")

	// Try to read the test directory for files to copy to our temporary directory.
	entries, err := os.ReadDir(testDir)
	if err != nil {
		t.Fatal(err)
	}
	for _, entry := range entries {
		if !strings.HasPrefix(entry.Name(), "ifcfg-") && !strings.HasPrefix(entry.Name(), "route-") && !strings.HasPrefix(entry.Name(), "route6-") {
			continue
		}

		configPath := filepath.Join(testDir, entry.Name())
		tmpPath := filepath.Join(tmpDir, entry.Name())
		err = fileCopy(configPath, tmpPath)
		if err != nil {
			t.Fatal(err)
		}
	}

	// Setup ifupdown and parse test file.
	ns, err := newNetworkScriptsWithConfig(tmpDir)
	if err != nil {
		t.Fatal(err)
	}

	// Get the interfaces state.
	interfaces, err := ns.GetInterfaces()
	if err != nil {
		t.Fatal(err)
	}

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 1)
	if err != nil {
		t.Error(err)
	}

	// Test setting the IP addresses on an interface.
	err = ns.SetIfaceAddresses(context.Background(), "test_eth0.1556", []*net.IPNet{
		{
			IP:   net.ParseIP("1.2.3.4"),
			Mask: net.CIDRMask(24, 32),
		},
		{
			IP:   net.ParseIP("1.2.3.43"),
			Mask: net.CIDRMask(24, 32),
		},
		{
			IP:   net.ParseIP("fc00::2"),
			Mask: net.CIDRMask(64, 128),
		},
	}, net.ParseIP("1.2.3.1"), net.ParseIP("fc00::1"))
	if err != nil {
		t.Fatal(err)
	}

	// Test setting routes on an interface.
	err = ns.SetIfaceRoutes(context.Background(), "test_eth2", []*Route{
		{
			Destination: &net.IPNet{
				IP:   net.ParseIP("abcd:ef12:3455:10::"),
				Mask: net.CIDRMask(64, 128),
			},
			Gateway: net.ParseIP("abcd:ef12:3456:10::1"),
			Metric:  100,
		},
		{
			Destination: &net.IPNet{
				IP:   net.ParseIP("10.253.2.0"),
				Mask: net.CIDRMask(24, 32),
			},
			Gateway: net.ParseIP("203.0.113.22"),
			Metric:  100,
		},
	})
	if err != nil {
		t.Fatal(err)
	}

	// Verify we can re-read configurations.
	ns, err = newNetworkScriptsWithConfig(tmpDir)
	if err != nil {
		t.Fatal(err)
	}

	// Get the interfaces state.
	interfaces, err = ns.GetInterfaces()
	if err != nil {
		t.Fatal(err)
	}

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 2)
	if err != nil {
		t.Error(err)
	}

	// Read the current file and expected state.
	err = testVerifyResults(resultsDir, tmpDir, 1)
	if err != nil {
		t.Error(err)
	}

	// Test setting the IP addresses on an interface.
	err = ns.SetIfaceAddresses(context.Background(), "test_eth0", []*net.IPNet{
		{
			IP:   net.ParseIP("1.2.10.4"),
			Mask: net.CIDRMask(24, 32),
		},
	}, net.ParseIP("1.2.10.254"), nil)
	if err != nil {
		t.Fatal(err)
	}

	// Test setting routes on an interface.
	err = ns.SetIfaceRoutes(context.Background(), "test_eth0.1556", []*Route{})
	if err != nil {
		t.Fatal(err)
	}

	// Verify we can re-read configurations.
	ns, err = newNetworkScriptsWithConfig(tmpDir)
	if err != nil {
		t.Fatal(err)
	}

	// Get the interfaces state.
	interfaces, err = ns.GetInterfaces()
	if err != nil {
		t.Fatal(err)
	}

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 3)
	if err != nil {
		t.Error(err)
	}

	// Read the current file and expected state.
	err = testVerifyResults(resultsDir, tmpDir, 2)
	if err != nil {
		t.Error(err)
	}

	// Cleanup.
	err = os.RemoveAll(tmpDir)
	if err != nil {
		t.Fatal(err)
	}
}