Add unit test case

Adding unit test for config package

Change-Id: I6897ac91f3ad55d2e48a0ab9255b224782fef5a5
Signed-off-by: Ritu Sood <ritu.sood@intel.com>

1 files changed, 163 insertions, 0 deletions

diff --git a/internal/pkg/config/config_test.go b/internal/pkg/config/config_test.go
new file mode 100644
index 0000000..32b1cc9
--- /dev/null
+++ b/internal/pkg/config/config_test.go
@@ -0,0 +1,163 @@
+package config
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/urfave/cli"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+func TestConfig(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Config Test Suite")
+}
+
+var _ = AfterSuite(func() {
+})
+
+var _ = Describe("Test Config", func() {
+	var app *cli.App
+	var cfgFile *os.File
+	var logFile *os.File
+	var kubecfgFile *os.File
+
+	BeforeEach(func() {
+		app = cli.NewApp()
+		app.Name = "test"
+		app.Flags = Flags
+
+		var err error
+		cfgFile, err = ioutil.TempFile("", "ovn4nfvconf-")
+		Expect(err).NotTo(HaveOccurred())
+		logFile, err = ioutil.TempFile("", "ovn4nfvlog-")
+		Expect(err).NotTo(HaveOccurred())
+		kubecfgFile, err = ioutil.TempFile("", "ovn4nfvkubecfg-")
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	AfterEach(func() {
+		os.Remove(cfgFile.Name())
+		os.Remove(logFile.Name())
+		os.Remove(kubecfgFile.Name())
+	})
+
+	It("uses expected defaults", func() {
+		app.Action = func(ctx *cli.Context) error {
+			cfgPath, err := InitConfig(ctx)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(cfgPath).To(Equal(cfgFile.Name()))
+
+			Expect(Default.MTU).To(Equal(1400))
+			Expect(Logging.File).To(Equal(""))
+			Expect(Logging.Level).To(Equal(4))
+			Expect(CNI.ConfDir).To(Equal("/etc/cni/net.d"))
+			Expect(CNI.Plugin).To(Equal("ovn4nfvk8s-cni"))
+			return nil
+		}
+		err := app.Run([]string{app.Name, "-config-file=" + cfgFile.Name(), "-k8s-kubeconfig=" + kubecfgFile.Name()})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	It("missing kubeconfig", func() {
+		app.Action = func(ctx *cli.Context) error {
+			_, err := InitConfig(ctx)
+			Expect(err).To(HaveOccurred())
+
+			return nil
+		}
+		err := app.Run([]string{app.Name, "-config-file=" + cfgFile.Name()})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	It("tests Config missing file", func() {
+		app.Action = func(ctx *cli.Context) error {
+			_, err := InitConfig(ctx)
+			Expect(err).To(MatchError("failed to open config file NoExistant: open NoExistant: no such file or directory"))
+			return nil
+		}
+
+		err := app.Run([]string{app.Name, "-config-file=NoExistant"})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	It("tests default config file", func() {
+		app.Action = func(ctx *cli.Context) error {
+			cfgPath, err := InitConfig(ctx)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(cfgPath).To(Equal(""))
+			return nil
+		}
+		err := app.Run([]string{app.Name, "-k8s-kubeconfig=" + kubecfgFile.Name(), "-logfile=" + logFile.Name()})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	It("overrides defaults with config file options", func() {
+		cfgData := fmt.Sprintf(`[default]
+mtu=1500
+
+[kubernetes]
+kubeconfig=%s
+
+[logging]
+loglevel=5
+logfile=%s
+
+[cni]
+conf-dir=/etc/cni/net.blah
+plugin=ovn-nfv-k8s-blah`, kubecfgFile.Name(), logFile.Name())
+		err := ioutil.WriteFile(cfgFile.Name(), []byte(cfgData), 0644)
+		Expect(err).NotTo(HaveOccurred())
+
+		app.Action = func(ctx *cli.Context) error {
+			var cfgPath string
+			cfgPath, err = InitConfig(ctx)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(cfgPath).To(Equal(cfgFile.Name()))
+
+			Expect(Default.MTU).To(Equal(1500))
+			Expect(Logging.File).To(Equal(logFile.Name()))
+			Expect(Logging.Level).To(Equal(5))
+			Expect(CNI.ConfDir).To(Equal("/etc/cni/net.blah"))
+			Expect(CNI.Plugin).To(Equal("ovn-nfv-k8s-blah"))
+			Expect(Kubernetes.Kubeconfig).To(Equal(kubecfgFile.Name()))
+
+			return nil
+		}
+		err = app.Run([]string{app.Name, "-config-file=" + cfgFile.Name()})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	It("overrides defaults with command line options", func() {
+		app.Action = func(ctx *cli.Context) error {
+			var cfgPath string
+			cfgPath, err := InitConfig(ctx)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(cfgPath).To(Equal(cfgFile.Name()))
+
+			Expect(Default.MTU).To(Equal(1500))
+			Expect(Logging.File).To(Equal(logFile.Name()))
+			Expect(Logging.Level).To(Equal(5))
+			Expect(CNI.ConfDir).To(Equal("/etc/cni/net.blah"))
+			Expect(CNI.Plugin).To(Equal("ovn-nfv-k8s-blah"))
+			Expect(Kubernetes.Kubeconfig).To(Equal(kubecfgFile.Name()))
+
+			return nil
+		}
+		args := []string{
+			app.Name,
+			"-config-file=" + cfgFile.Name(),
+			"-k8s-kubeconfig=" + kubecfgFile.Name(),
+			"-mtu=1500",
+			"-loglevel=5", "-logfile=" + logFile.Name(),
+			"-cni-conf-dir=/etc/cni/net.blah",
+			"-cni-plugin=ovn-nfv-k8s-blah"}
+
+		err := app.Run(args)
+		Expect(err).NotTo(HaveOccurred())
+	})
+})