Adds ability to update golden files

cmd/root_test.go

@@ -2,6 +2,7 @@ package cmd_test
 
 import (
 	"bytes"
+	"flag"
 	"fmt"
 	"io/ioutil"
 	"os"
@@ -11,6 +12,14 @@ import (
 	"github.com/sebradloff/rawk8stfc/cmd"
 )
 
+var (
+	updateFlag bool
+)
+
+func init() {
+	flag.BoolVar(&updateFlag, "update", false, "Set this flag to update the golden files.")
+}
+
 func TestRoot(t *testing.T) {
 	testName := "one-obj"
 
@@ -26,6 +35,15 @@ func TestRoot(t *testing.T) {
 		t.Errorf("running root command failed. err = %v", err)
 	}
 
+	if updateFlag {
+		cmd.Flags().Set("k8sFile", k8sFile)
+		cmd.Flags().Set("outputFile", goldenFile)
+		err := cmd.RunE(cmd, []string{})
+		if err != nil {
+			t.Errorf("running root command failed. err = %v", err)
+		}
+	}
+
 	got, err := os.Open(outputFile)
 	if err != nil {
 		t.Fatalf("failed to open the got file: %s. err = %v", outputFile, err)