From 8ea8613525171f705bcf7f3a9103b560dbb761fc Mon Sep 17 00:00:00 2001
From: Daniel Mendoza <daltemen@gmail.com>
Date: Mon, 9 Mar 2020 21:13:34 -0500
Subject: [PATCH] added support testing short coverage (#13)

* added support testing short coverage

* append short and tags flags in coverage command

* deleted unnecesary initialization

* Small change to wording for clarity.

Co-authored-by: Jordan Poles <jpoles1@users.noreply.github.com>
---
 README.md |  8 ++++++++
 main.go   | 22 ++++++++++++++++------
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 0e62c13..bdec92e 100644
--- a/README.md
+++ b/README.md
@@ -49,6 +49,14 @@ Test code coverage using build tags:
 
 <hr>
 
+<hr>
+
+Test code coverage, skipping short tests:
+
+`gopherbadger -short`
+
+<hr>
+
 ## Confused?
 
 Try running:
diff --git a/main.go b/main.go
index 074f9e8..d27034f 100644
--- a/main.go
+++ b/main.go
@@ -18,6 +18,11 @@ import (
 	"github.com/fatih/color"
 )
 
+const (
+	testCommand      = "go test ./... -coverprofile=coverage.out"
+	toolCoverCommand = "go tool cover -func=coverage.out"
+)
+
 func getCommandOutput(commandString string) chan float64 {
 	cmd := exec.Command("bash", "-c", commandString)
 	cmd.Stderr = os.Stderr
@@ -73,6 +78,7 @@ func main() {
 	coverageCommandFlag := flag.String("covercmd", "", "gocover command to run; must print coverage report to stdout")
 	manualCoverageFlag := flag.Float64("manualcov", -1.0, "A manually inputted coverage float.")
 	tagsFlag := flag.String("tags", "", "The build tests you'd like to include in your coverage")
+	shortFlag := flag.Bool("short", false, "It will skip tests marked as testing.Short()")
 	flag.Parse()
 
 	if !containsString(badgeStyles, *badgeStyleFlag) {
@@ -84,17 +90,21 @@ func main() {
 		ImageExtension: ".png",
 	}
 	var coverageFloat float64
-
 	coverageCommand := ""
 	if *coverageCommandFlag != "" {
 		coverageCommand = *coverageCommandFlag
-		if *tagsFlag != "" {
-			log.Println("Warning: When the covercmd flag is used the tags flag will be ignored.")
+		if *tagsFlag != "" || *shortFlag {
+			log.Println("Warning: When the covercmd flag is used the -tags and -short flags will be ignored.")
 		}
-	} else if *tagsFlag != "" {
-		coverageCommand = "go test ./... -tags \"" + *tagsFlag + "\" -coverprofile=coverage.out && go tool cover -func=coverage.out"
 	} else {
-		coverageCommand = "go test ./... -coverprofile=coverage.out && go tool cover -func=coverage.out"
+		flagsCommands := ""
+		if *tagsFlag != "" {
+			flagsCommands = flagsCommands + " -tags \"" + *tagsFlag + "\""
+		}
+		if *shortFlag {
+			flagsCommands = flagsCommands + " -short"
+		}
+		coverageCommand = fmt.Sprintf("%s %s && %s", testCommand, flagsCommands, toolCoverCommand)
 	}
 
 	if *manualCoverageFlag == -1 {