added network config

2019-11-29 19:53:17 +01:00
parent ce335164ff
commit d7e0591576
6 changed files with 310 additions and 4 deletions
--- a/network_test.go
+++ b/network_test.go
@@ -0,0 +1,116 @@
+package alpine_builder
+
+import (
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func init() {
+	networkConfig = "test_interfaces"
+}
+
+func TestGetNetworkInfo(t *testing.T) {
+	ass := assert.New(t)
+
+	// file does not exist
+	_, err := GetNetworkInfo()
+	ass.EqualError(err, "failed to open network config: open test_interfaces: no such file or directory")
+
+	// invalid config file
+	ass.NoError(ioutil.WriteFile(networkConfig, []byte(""), os.ModePerm))
+	_, err = GetNetworkInfo()
+	ass.EqualError(err, "invalid network config")
+
+	// dynamic config
+	ass.NoError(ioutil.WriteFile(networkConfig, []byte(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet dhcp`), os.ModePerm))
+	info, err := GetNetworkInfo()
+	ass.NoError(err)
+	ass.False(info.IsStatic)
+
+	// static config
+	ass.NoError(ioutil.WriteFile(networkConfig, []byte(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet static
+  address 1.2.3.4
+  netmask 255.255.255.0
+  gateway 4.3.2.1`), os.ModePerm))
+	info, err = GetNetworkInfo()
+	ass.NoError(err)
+	ass.True(info.IsStatic)
+	ass.Equal("1.2.3.4", info.Address)
+	ass.Equal("255.255.255.0", info.Netmask)
+	ass.Equal("4.3.2.1", info.Gateway)
+
+	_ = os.Remove(networkConfig)
+}
+
+func TestNetworkEnableDHCP(t *testing.T) {
+	ass := assert.New(t)
+
+	// file does not exist
+	_, err := GetNetworkInfo()
+	ass.EqualError(err, "failed to open network config: open test_interfaces: no such file or directory")
+
+	ass.NoError(ioutil.WriteFile(networkConfig, []byte(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet static
+  address 1.2.3.4
+  netmask 255.255.255.0
+  gateway 4.3.2.1`), os.ModePerm))
+
+	ass.NoError(NetworkEnableDHCP())
+
+	data, err := ioutil.ReadFile(networkConfig)
+	ass.NoError(err)
+
+	ass.Equal(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet dhcp
+`, string(data))
+
+	_ = os.Remove(networkConfig)
+}
+
+func TestNetworkSetStatic(t *testing.T) {
+	ass := assert.New(t)
+
+	// file does not exist
+	_, err := GetNetworkInfo()
+	ass.EqualError(err, "failed to open network config: open test_interfaces: no such file or directory")
+
+	ass.NoError(ioutil.WriteFile(networkConfig, []byte(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet dhcp`), os.ModePerm))
+
+	ass.NoError(NetworkSetStatic("1.2.3.4", "255.255.255.0", "4.3.2.1"))
+
+	data, err := ioutil.ReadFile(networkConfig)
+	ass.NoError(err)
+
+	ass.Equal(`auto lo
+iface lo inet loopback
+
+auto eth0
+iface eth0 inet static
+  address 1.2.3.4
+  netmask 255.255.255.0
+  gateway 4.3.2.1
+`, string(data))
+
+	_ = os.Remove(networkConfig)
+}