use testing hack to mock git call

nate smith · nate smith · commit f6afe1b576b0 · 2019-10-30T16:34:23.000-05:00
diff --git a/git/git.go b/git/git.go
@@ -216,9 +216,17 @@ func LocalBranches() ([]string, error) {
 	return branches, nil
 }
 
+var GitCommand = func(args ...string) *exec.Cmd {
+	return exec.Command("git", args...)
+}
+
 func UncommittedChangeCount() (int, error) {
-	statusCmd := exec.Command("git", "status", "--porcelain")
+	statusCmd := GitCommand("status", "--porcelain")
+	fmt.Println("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
+	fmt.Printf("%+v", statusCmd)
 	output, err := statusCmd.Output()
+	fmt.Println(string(output))
+	fmt.Println("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
 	if err != nil {
 		return 0, fmt.Errorf("failed to run git status: %s", err)
 	}
diff --git a/git/git_test.go b/git/git_test.go
@@ -0,0 +1,62 @@
+package git
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"testing"
+)
+
+func StubbedGit(args ...string) *exec.Cmd {
+	cs := []string{"-test.run=TestHelperProcess", "--", "git"}
+	cs = append(cs, args...)
+	env := []string{
+		"GO_WANT_HELPER_PROCESS=1",
+	}
+
+	cmd := exec.Command(os.Args[0], cs...)
+	cmd.Env = append(env, os.Environ()...)
+	fmt.Printf("%+v", cmd.Env)
+	return cmd
+}
+
+func TestHelperProcess(*testing.T) {
+	if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
+		return
+	}
+	defer os.Exit(0)
+	args := os.Args
+	for len(args) > 0 {
+		if args[0] == "--" {
+			args = args[1:]
+			break
+		}
+		args = args[1:]
+	}
+	cmd, args := args[0], args[1:]
+	switch cmd {
+	case "git":
+		fmt.Println("fart")
+		fmt.Println("town")
+	}
+
+}
+
+func Test_UncommittedChangeCount(t *testing.T) {
+	origGitCommand := GitCommand
+	defer func() {
+		GitCommand = origGitCommand
+	}()
+
+	GitCommand = StubbedGit
+
+	ucc, err := UncommittedChangeCount()
+	if err != nil {
+		t.Fatalf("failed to run UncommittedChangeCount: %s", err)
+	}
+
+	if ucc != 10 {
+		t.Errorf("got unexpected ucc value: %d", ucc)
+	}
+
+}
