Unit tests for main and config

Low value, but still...

Author: bpineau

config/config_test.go

@@ -0,0 +1,30 @@
+package config
+
+import (
+	"os"
+	"testing"
+
+	"github.com/bpineau/katafygio/pkg/log"
+)
+
+const nonExistentPath = "\\/hopefully/non/existent/path"
+
+func TestConfig(t *testing.T) {
+	conf := &KfConfig{
+		DryRun: true,
+		Logger: log.New("info", "", "test"),
+	}
+
+	err := conf.Init("http://127.0.0.1", nonExistentPath)
+	if err == nil {
+		t.Error("conf.Init() should fail on non existent kubeconfig path")
+	}
+
+	here, _ := os.Getwd()
+	_ = os.Setenv("HOME", here+"/../assets")
+	err = conf.Init("", "")
+	if err != nil {
+		t.Error("conf.Init() with no arguments should find a .kube/config in $HOME")
+	}
+
+}

main_test.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/bpineau/katafygio/cmd"
+)
+
+func TestMain(t *testing.T) {
+	var ok = true
+
+	privateExitHandler = func(c int) {
+		ok = false
+	}
+
+	//cmd.FakeCS = true
+	cmd.RootCmd.SetOutput(new(bytes.Buffer))
+
+	// test with normal exit
+	cmd.RootCmd.SetArgs([]string{"--help"})
+	main()
+
+	if !ok {
+		t.Errorf("main() failed")
+	}
+
+	// test with failure
+	cmd.RootCmd.SetArgs([]string{"--unexpected-arg"})
+	main()
+
+	if ok {
+		t.Errorf("main() should fail with unexpected arguments")
+	}
+}
+
+func TestExitWrapper(t *testing.T) {
+	var ok = false
+
+	privateExitHandler = func(c int) {
+		ok = true
+	}
+
+	ExitWrapper(1)
+
+	if !ok {
+		t.Errorf("Error in ExitWrapper()")
+	}
+}