teeset: import first cli test

cmd/coca_test.go

+package cmd
+
+import (
+	"bytes"
+	shellwords "github.com/mattn/go-shellwords"
+	"github.com/spf13/cobra"
+	"os"
+	"strings"
+	"testing"
+)
+
+// cmdTestCase describes a test case that works with releases.
+type cmdTestCase struct {
+	name      string
+	cmd       string
+	golden    string
+	wantError bool
+}
+
+func runTestCmd(t *testing.T, tests []cmdTestCase) {
+	t.Helper()
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			defer resetEnv()()
+
+			t.Log("running cmd: ", tt.cmd)
+			_, output, err := executeActionCommandC(tt.cmd)
+			if (err != nil) != tt.wantError {
+				t.Errorf("expected error, got '%v'", err)
+			}
+			if tt.golden != "" {
+				AssertGoldenString(t, output, tt.golden)
+			}
+		})
+	}
+}
+
+func executeActionCommandC(cmd string) (*cobra.Command, string, error) {
+	args, err := shellwords.Parse(cmd)
+	if err != nil {
+		return nil, "", err
+	}
+	buf := new(bytes.Buffer)
+
+	rootCmd.SetOutput(buf)
+	rootCmd.SetArgs(args)
+
+	c, err := rootCmd.ExecuteC()
+
+	return c, buf.String(), err
+}
+
+func resetEnv() func() {
+	origEnv := os.Environ()
+	return func() {
+		os.Clearenv()
+		for _, pair := range origEnv {
+			kv := strings.SplitN(pair, "=", 2)
+			os.Setenv(kv[0], kv[1])
+		}
+	}
+}

cmd/test_helper.go

+package cmd
+
+// based on https://github.com/helm/helm
+// license: apache 2.0
+// just copy
+
+import (
+	"bytes"
+	"flag"
+	"io/ioutil"
+	"path/filepath"
+
+	"github.com/pkg/errors"
+)
+
+// UpdateGolden writes out the golden files with the latest values, rather than failing the test.
+var updateGolden = flag.Bool("update", false, "update golden files")
+
+// TestingT describes a testing object compatible with the critical functions from the testing.T type
+type TestingT interface {
+	Fatal(...interface{})
+	Fatalf(string, ...interface{})
+	HelperT
+}
+
+// HelperT describes a test with a helper function
+type HelperT interface {
+	Helper()
+}
+
+// AssertGoldenBytes asserts that the give actual content matches the contents of the given filename
+func AssertGoldenBytes(t TestingT, actual []byte, filename string) {
+	t.Helper()
+
+	if err := compare(actual, path(filename)); err != nil {
+		t.Fatalf("%v", err)
+	}
+}
+
+// AssertGoldenString asserts that the given string matches the contents of the given file.
+func AssertGoldenString(t TestingT, actual, filename string) {
+	t.Helper()
+
+	if err := compare([]byte(actual), path(filename)); err != nil {
+		t.Fatalf("%v", err)
+	}
+}
+
+// AssertGoldenFile assers that the content of the actual file matches the contents of the expected file
+func AssertGoldenFile(t TestingT, actualFileName string, expectedFilename string) {
+	t.Helper()
+
+	actual, err := ioutil.ReadFile(actualFileName)
+	if err != nil {
+		t.Fatalf("%v", err)
+	}
+	AssertGoldenBytes(t, actual, expectedFilename)
+}
+
+func path(filename string) string {
+	if filepath.IsAbs(filename) {
+		return filename
+	}
+	return filepath.Join("testdata", filename)
+}
+
+func compare(actual []byte, filename string) error {
+	if err := update(filename, actual); err != nil {
+		return err
+	}
+
+	expected, err := ioutil.ReadFile(filename)
+	if err != nil {
+		return errors.Wrapf(err, "unable to read testdata %s", filename)
+	}
+	if !bytes.Equal(expected, actual) {
+		return errors.Errorf("does not match golden file %s\n\nWANT:\n'%s'\n\nGOT:\n'%s'\n", filename, expected, actual)
+	}
+	return nil
+}
+
+
+func update(filename string, in []byte) error {
+	if !*updateGolden {
+		return nil
+	}
+	return ioutil.WriteFile(filename, normalize(in), 0666)
+}
+
+func normalize(in []byte) []byte {
+	return bytes.Replace(in, []byte("\r\n"), []byte("\n"), -1)
+}
\ No newline at end of file

cmd/version_test.go

+package cmd
+
+import (
+	"testing"
+)
+
+func TestVersion(t *testing.T) {
+	tests := []cmdTestCase{{
+		name:   "version",
+		cmd:    "version",
+		golden: "output/version.txt",
+	}}
+	runTestCmd(t, tests)
+}
\ No newline at end of file

go.mod

@@ -8,11 +8,13 @@ require (
 	github.com/boyter/scc v2.10.1+incompatible
 	github.com/dbaggerman/cuba v0.3.2 // indirect
 	github.com/iancoleman/strcase v0.0.0-20191112232945-16388991a334
+	github.com/mattn/go-shellwords v1.0.6
 	github.com/minio/blake2b-simd v0.0.0-20160723061019-3f5f724cb5b1 // indirect
 	github.com/monochromegane/go-gitignore v0.0.0-20160105113617-38717d0a108c // indirect
 	github.com/olekukonko/tablewriter v0.0.4
 	github.com/onsi/ginkgo v1.10.3 // indirect
 	github.com/onsi/gomega v1.7.1
+	github.com/pkg/errors v0.8.1
 	github.com/pkg/profile v1.3.0
 	github.com/sabhiram/go-gitignore v0.0.0-20180611051255-d3107576ba94
 	github.com/spf13/cobra v0.0.5

go.sum

@@ -37,6 +37,8 @@ github.com/karrick/godirwalk v1.10.12/go.mod h1:RoGL9dQei4vP9ilrpETWE8CLOZ1kiN0L
 github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
 github.com/mattn/go-runewidth v0.0.7 h1:Ei8KR0497xHyKJPAv59M1dkC+rOZCMBJ+t3fZ+twI54=
 github.com/mattn/go-runewidth v0.0.7/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
+github.com/mattn/go-shellwords v1.0.6 h1:9Jok5pILi5S1MnDirGVTufYGtksUs/V2BWUP3ZkeUUI=
+github.com/mattn/go-shellwords v1.0.6/go.mod h1:3xCvwCdWdlDJUrvuMn7Wuy9eWs4pE8vqg+NOMyg4B2o=
 github.com/minio/blake2b-simd v0.0.0-20160723061019-3f5f724cb5b1 h1:lYpkrQH5ajf0OXOcUbGjvZxxijuBwbbmlSxLiuofa+g=
 github.com/minio/blake2b-simd v0.0.0-20160723061019-3f5f724cb5b1/go.mod h1:pD8RvIylQ358TN4wwqatJ8rNavkEINozVn9DtGI3dfQ=
 github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
@@ -51,6 +53,8 @@ github.com/onsi/ginkgo v1.10.3/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+
 github.com/onsi/gomega v1.7.1 h1:K0jcRCwNQM3vFGh1ppMtDh/+7ApJrjldlX8fA0jDTLQ=
 github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY=
 github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
+github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
+github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pkg/profile v1.3.0 h1:OQIvuDgm00gWVWGTf4m4mCt6W1/0YqU7Ntg0mySWgaI=
 github.com/pkg/profile v1.3.0/go.mod h1:hJw3o1OdXxsrSjjVksARp5W95eeEaEfptyVZyv6JUPA=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=