fix coverage being under-reported due to over-counting of files

imsky · Jun 12, 2018 · 46c3ee1 · 46c3ee1
1 parent 6953aaf
commit 46c3ee1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -28,6 +28,7 @@ func GetLines() []Line {
 
 func ComputeTotalCoverage(sourceFiles map[string][]SourceFile) float64 {
 	var totalCoverage float64
+	totalFiles := 0
 
 	//todo: validate that each source file list of lines is same length
 	for _, fileSets := range sourceFiles {
@@ -49,11 +50,12 @@ func ComputeTotalCoverage(sourceFiles map[string][]SourceFile) float64 {
 		}
 
 		if totalLines > 0 {
+			totalFiles++
 			totalCoverage += float64(hitLines) / float64(totalLines)
 		}
 	}
 
-	return (totalCoverage / float64(len(sourceFiles))) * 100
+	return (totalCoverage / float64(totalFiles)) * 100
 }
 
 func Run(reports []string) {