wip

Author: linkdata

main.go

@@ -19,21 +19,21 @@ var (
 var ErrMissingPctFlag = errors.New("missing required flag -pct")
 
 func mainfn() int {
-	err := ErrMissingPctFlag
-	if *flagPct != "" {
+	vs, err := gitsemver.New(*flagGit)
+	if err == nil {
+		if *flagDebug {
+			vs.DebugOut = os.Stderr
+		}
 		repoDir := os.ExpandEnv(flag.Arg(0))
 		if repoDir == "" {
 			repoDir = "."
 		}
-		var vs *gitsemver.GitSemVer
-		if vs, err = gitsemver.New(*flagGit); err == nil {
-			if *flagDebug {
-				vs.DebugOut = os.Stderr
-			}
-			if repoDir, err = vs.Git.CheckGitRepo(repoDir); err == nil {
-				var vi gitsemver.VersionInfo
-				if vi, err = vs.GetVersion(repoDir); err == nil {
-
+		if repoDir, err = vs.Git.CheckGitRepo(repoDir); err == nil {
+			var vi gitsemver.VersionInfo
+			if vi, err = vs.GetVersion(repoDir); err == nil {
+				if *flagPct == "" {
+					err = ErrMissingPctFlag
+				} else {
 					fmt.Println(vi.Branch)
 					return 0
 				}

main_test.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"os"
 	"testing"
 )
 
@@ -11,6 +12,7 @@ func TestMainFn(t *testing.T) {
 }
 
 func TestMainError(t *testing.T) {
+	os.Args = append(os.Args, "/proc/.nonexistant")
 	exitFn = func(i int) {
 		if i == 0 {
 			t.Error(i)