128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
package netconfig
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Validate the netplan configuration parser/writer functions.
|
|
func TestIfUpDown(t *testing.T) {
|
|
// Setup test file.
|
|
tmpDir, err := os.MkdirTemp("", "")
|
|
require.NoError(t, err)
|
|
configPath := filepath.Join(tmpDir, "interfaces")
|
|
testDir, err := filepath.Abs("./tests/ifupdown")
|
|
require.NoError(t, err)
|
|
resultsDir := filepath.Join(testDir, "results")
|
|
err = fileCopy(filepath.Join(testDir, "interfaces"), configPath)
|
|
require.NoError(t, err)
|
|
|
|
// Setup ifupdown and parse test file.
|
|
i := new(ifUpDown)
|
|
i.BaseConfig = configPath
|
|
i.BackupDir = configPath + ".backup"
|
|
err = i.ReadFile(i.BaseConfig)
|
|
require.NoError(t, err)
|
|
|
|
// Get the interfaces state.
|
|
interfaces, err := i.GetInterfaces()
|
|
require.NoError(t, err)
|
|
|
|
// Verify interfaces read from file.
|
|
err = testVerifyInterfaces(interfaces, resultsDir, 1)
|
|
require.NoError(t, err)
|
|
|
|
// Test setting the IP addresses on an interface.
|
|
err = i.SetIfaceAddresses(context.Background(), "test_eth0", []*net.IPNet{
|
|
{
|
|
IP: net.ParseIP("203.0.113.2"),
|
|
Mask: net.CIDRMask(24, 32),
|
|
},
|
|
{
|
|
IP: net.ParseIP("203.0.113.3"),
|
|
Mask: net.CIDRMask(24, 32),
|
|
},
|
|
{
|
|
IP: net.ParseIP("fc00::2"),
|
|
Mask: net.CIDRMask(64, 128),
|
|
},
|
|
}, net.ParseIP("203.0.113.6"), net.ParseIP("fc00::1"))
|
|
require.NoError(t, err)
|
|
|
|
// Test setting routes on an interface.
|
|
err = i.SetIfaceRoutes(context.Background(), "test_eth1", []*Route{
|
|
{
|
|
Destination: &net.IPNet{
|
|
IP: net.ParseIP("abcd:ef12:3455:3::"),
|
|
Mask: net.CIDRMask(64, 128),
|
|
},
|
|
Gateway: net.ParseIP("abcd:ef12:3456:3::1"),
|
|
Metric: 100,
|
|
},
|
|
{
|
|
Destination: &net.IPNet{
|
|
IP: net.ParseIP("10.253.2.0"),
|
|
Mask: net.CIDRMask(24, 32),
|
|
},
|
|
Gateway: net.ParseIP("1.2.3.1"),
|
|
Metric: 100,
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Verify we can read the modified files.
|
|
i.Interfaces = nil
|
|
err = i.ReadFile(i.BaseConfig)
|
|
require.NoError(t, err)
|
|
|
|
// Get the interfaces state.
|
|
interfaces, err = i.GetInterfaces()
|
|
require.NoError(t, err)
|
|
|
|
// Verify interfaces read from file.
|
|
err = testVerifyInterfaces(interfaces, resultsDir, 2)
|
|
require.NoError(t, err)
|
|
|
|
// Read the current file and expected state.
|
|
err = testVerifyResults(resultsDir, tmpDir, 1)
|
|
require.NoError(t, err)
|
|
|
|
// Test setting the IP addresses on an interface.
|
|
err = i.SetIfaceAddresses(context.Background(), "test_backend", []*net.IPNet{
|
|
{
|
|
IP: net.ParseIP("1.2.10.4"),
|
|
Mask: net.CIDRMask(24, 32),
|
|
},
|
|
}, net.ParseIP("1.2.10.254"), nil)
|
|
require.NoError(t, err)
|
|
|
|
// Test setting routes on an interface.
|
|
err = i.SetIfaceRoutes(context.Background(), "test_eth0", []*Route{})
|
|
require.NoError(t, err)
|
|
|
|
// Verify we can read the modified files.
|
|
i.Interfaces = nil
|
|
err = i.ReadFile(i.BaseConfig)
|
|
require.NoError(t, err)
|
|
|
|
// Get the interfaces state.
|
|
interfaces, err = i.GetInterfaces()
|
|
require.NoError(t, err)
|
|
|
|
// Verify interfaces read from file.
|
|
err = testVerifyInterfaces(interfaces, resultsDir, 3)
|
|
require.NoError(t, err)
|
|
|
|
// Read the current file and expected state.
|
|
err = testVerifyResults(resultsDir, tmpDir, 2)
|
|
require.NoError(t, err)
|
|
|
|
// Cleanup.
|
|
err = os.RemoveAll(tmpDir)
|
|
require.NoError(t, err)
|
|
}
|