modify getTestMap func

moricho · Sep 3, 2020 · 76d5ee2 · 76d5ee2
1 parent c448147
commit 76d5ee2
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/tparallel.go b/tparallel.go
@@ -77,14 +77,15 @@ func getTestMap(ssaanalyzer *buildssa.SSA, testTyp types.Type) map[*ssa.Function
 
 	trun := analysisutil.MethodOf(testTyp, "Run")
 	for _, f := range ssaanalyzer.SrcFuncs {
-		if strings.HasPrefix(f.Name(), "Test") && f.Parent() == (*ssa.Function)(nil) {
-			testMap[f] = []*ssa.Function{}
-			for _, block := range f.Blocks {
-				for _, instr := range block.Instrs {
-					called := analysisutil.Called(instr, nil, trun)
-					if called {
-						testMap[f] = appendTestMap(testMap[f], instr)
-					}
+		if !strings.HasPrefix(f.Name(), "Test") || !(f.Parent() == (*ssa.Function)(nil)) {
+			continue
+		}
+		testMap[f] = []*ssa.Function{}
+		for _, block := range f.Blocks {
+			for _, instr := range block.Instrs {
+				called := analysisutil.Called(instr, nil, trun)
+				if called {
+					testMap[f] = appendTestMap(testMap[f], instr)
 				}
 			}
 		}
@@ -106,5 +107,6 @@ func appendTestMap(subtests []*ssa.Function, instr ssa.Instruction) []*ssa.Funct
 			subtests = append(subtests, arg)
 		}
 	}
+
 	return subtests
 }