Merge pull request #194 from noborus/yaml-check

Added yaml check

config_test.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/spf13/viper"
+)
+
+// This test is a yaml file check.
+func Test_Config(t *testing.T) {
+	type args struct {
+		cfgFile string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantErr bool
+	}{
+		{
+			name: "test ov.yaml",
+			args: args{
+				cfgFile: "ov.yaml",
+			},
+			wantErr: false,
+		},
+		{
+			name: "test ov-less.yaml",
+			args: args{
+				cfgFile: "ov-less.yaml",
+			},
+			wantErr: false,
+		},
+		{
+			name: "test error",
+			args: args{
+				cfgFile: "config-fail.yaml",
+			},
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			viper.SetConfigFile(tt.args.cfgFile)
+			err := viper.ReadInConfig()
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Config error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+		})
+	}
+}

main.go

@@ -314,8 +314,10 @@ func initConfig() {
 	viper.AutomaticEnv() // read in environment variables that match
 
 	// If a config file is found, read it in.
-	_ = viper.ReadInConfig()
-
+	if err := viper.ReadInConfig(); err != nil {
+		fmt.Printf("%s: %s\n", viper.ConfigFileUsed(), err.Error())
+		os.Exit(1)
+	}
 	if err := viper.Unmarshal(&config); err != nil {
 		fmt.Println(err)
 		os.Exit(1)