Add some tests to the main file

main.go

@@ -92,19 +92,19 @@ func parseSubCommands() (string, subCommand) {
 	}
 
 	if len(os.Args) < 2 {
-		listCmds()
+		listSubCommands()
 	}
 
 	subCmd, subCmdExists := cmds[os.Args[1]]
 
 	if !subCmdExists {
-		listCmds()
+		listSubCommands()
 	}
 
 	return os.Args[1], subCmd
 }
 
-func listCmds() {
+func listSubCommands() {
 	cmds := getSubCommands()
 
 	fmt.Fprintf(flag.CommandLine.Output(), "Usage of %s:\n", os.Args[0])
@@ -114,5 +114,7 @@ func listCmds() {
 		fmt.Printf("    %s\n", cmd.desc)
 	}
 
-	os.Exit(1)
+	if flag.Lookup("test.v") == nil {
+		os.Exit(1)
+	}
 }

main_test.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMain(t *testing.T) {
+	main()
+}
+
+func TestParseSubCommands(t *testing.T) {
+	os.Args[1] = "migrate:db"
+	_, v := parseSubCommands()
+	assert.NotEmpty(t, v.desc)
+
+	os.Args[1] = "bogus"
+	_, v = parseSubCommands()
+	assert.Empty(t, v.desc)
+}
+
+func TestListSubCommands(t *testing.T) {
+	listSubCommands()
+}