Merge pull request #10 from award28/issue_9

Issue #9 Allow code coverage with build tags

README.md

@@ -43,6 +43,12 @@ Manually set the coverage value (note: do not include %):
 
 <hr>
 
+Test code coverage using build tags:
+
+`gopherbadger -tags "unit"`
+
+<hr>
+
 ## Confused?
 
 Try running:

main.go

@@ -70,8 +70,9 @@ func main() {
 	badgeStyleFlag := flag.String("style", "flat", "Badge style from list: ["+strings.Join(badgeStyles, ",")+"]")
 	updateMdFilesFlag := flag.String("md", "", "A list of markdown filepaths for badge updates.")
 	coveragePrefixFlag := flag.String("prefix", "Go", "A prefix to specify the coverage in your badge.")
-	coverageCommandFlag := flag.String("covercmd", "go test ./... -coverprofile=coverage.out && go tool cover -func=coverage.out", "gocover command to run; must print coverage report to stdout")
+	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")
 	flag.Parse()
 
 	if !containsString(badgeStyles, *badgeStyleFlag) {
@@ -83,8 +84,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.")
+		}
+	} 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"
+	}
+
 	if *manualCoverageFlag == -1 {
-		coverageFloat = <-getCommandOutput(*coverageCommandFlag)
+		coverageFloat = <-getCommandOutput(coverageCommand)
 	} else {
 		coverageFloat = *manualCoverageFlag
 	}