network: Add a testcase for setupDNS

setupDNS adds DNS to the Kata guest VM. Add a testcase for that function. Fixes: #635 Signed-off-by: Nitesh Konkar <[email protected]>
kata-containers · Sep 11, 2019 · e01f23c · e01f23c
1 parent d733185
commit e01f23c
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/network_test.go b/network_test.go
@@ -7,9 +7,12 @@
 package main
 
 import (
+	"io/ioutil"
 	"net"
+	"os"
 	"reflect"
 	"runtime"
+	"strings"
 	"testing"
 
 	"github.com/kata-containers/agent/pkg/types"
@@ -422,3 +425,33 @@ func TestListRoutesWithTwoInterfacesSameSubnet(t *testing.T) {
 		"Route listed didn't match: got %+v, expecting %+v", results.Routes[2], expectedRoutes[2])
 
 }
+
+// As mounting errors out in permission denied, so test kataGuestSandboxDNSFile contents only.
+func TestSetupDNS(t *testing.T) {
+	skipUnlessRoot(t)
+
+	tmpfile, err := ioutil.TempFile("", "resolv.conf")
+	assert.NoError(t, err)
+	guestDNSFile = tmpfile.Name()
+
+	tmpfile, err = ioutil.TempFile("", "resolv.conf")
+	assert.NoError(t, err)
+	kataGuestSandboxDNSFile = tmpfile.Name()
+
+	defer os.RemoveAll(guestDNSFile)
+	defer os.RemoveAll(kataGuestSandboxDNSFile)
+
+	dns := []string{
+		"nameserver 8.8.8.8",
+		"nameserver 8.8.4.4",
+	}
+
+	err = setupDNS(dns)
+	assert.NoError(t, err)
+
+	content, err := ioutil.ReadFile(guestDNSFile)
+	assert.NoError(t, err)
+
+	expectedDNS := strings.Split(string(content), "\n")
+	assert.Equal(t, dns, expectedDNS)
+}