detect defer statement

testdata/src/sample/cleanup_test.go

@@ -9,8 +9,8 @@ func Test_Cleanup1(t *testing.T) { // want "Test_Cleanup1 should use t.Cleanup"
 	t.Parallel()
 
 	t.Run("Cleanup1_Sub1", func(t *testing.T) {
-		call("Cleanup1_Sub1")
 		t.Parallel()
+		call("Cleanup1_Sub1")
 	})
 
 	t.Run("Cleanup1_Sub2", func(t *testing.T) {
@@ -30,3 +30,18 @@ func Test_Cleanup2(t *testing.T) { // OK
 		call("Cleanup2_Sub2")
 	})
 }
+
+func Test_Cleanup3(t *testing.T) { // OK
+	t.Parallel()
+	call("Test_Cleanup3")
+
+	t.Run("Cleanup3_Sub1", func(t *testing.T) {
+		t.Parallel()
+		call("Cleanup3_Sub1")
+	})
+
+	t.Run("Cleanup3_Sub2", func(t *testing.T) {
+		t.Parallel()
+		call("Cleanup3_Sub2")
+	})
+}

tparallel.go

@@ -54,9 +54,11 @@ func run(pass *analysis.Pass) (interface{}, error) {
 			}
 		}
 
-		useCleanup := isCalled(top, cleanup)
-		if isPararellSub && !useCleanup {
-			pass.Reportf(top.Pos(), "%s should use t.Cleanup", top.Name())
+		if isDeferCalled(top) {
+			useCleanup := isCalled(top, cleanup)
+			if isPararellSub && !useCleanup {
+				pass.Reportf(top.Pos(), "%s should use t.Cleanup instead of defer", top.Name())
+			}
 		}
 
 		if isParallelTop == isPararellSub {
@@ -71,6 +73,18 @@ func run(pass *analysis.Pass) (interface{}, error) {
 	return nil, nil
 }
 
+func isDeferCalled(f *ssa.Function) bool {
+	for _, block := range f.Blocks {
+		for _, instr := range block.Instrs {
+			switch instr.(type) {
+			case *ssa.Defer:
+				return true
+			}
+		}
+	}
+	return false
+}
+
 func isCalled(f *ssa.Function, typ *types.Func) bool {
 	for _, block := range f.Blocks {
 		for _, instr := range block.Instrs {