support table test

moricho · Sep 4, 2020 · 255baa9 · 255baa9
1 parent d1ab6fe
commit 255baa9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/testdata/src/sample/cleanup_test.go b/testdata/src/sample/cleanup_test.go
@@ -1,8 +1,6 @@
 package sample
 
-import (
-	"testing"
-)
+import "testing"
 
 func Test_Cleanup1(t *testing.T) { // want "Test_Cleanup1 should use t.Cleanup"
 	teardown := setup("Test_Cleanup1")

diff --git a/tparallel.go b/tparallel.go
@@ -116,6 +116,9 @@ func appendTestMap(subtests []*ssa.Function, instr ssa.Instruction) []*ssa.Funct
 		switch arg := arg.(type) {
 		case *ssa.Function:
 			subtests = append(subtests, arg)
+		case *ssa.MakeClosure:
+			fn, _ := arg.Fn.(*ssa.Function)
+			subtests = append(subtests, fn)
 		}
 	}