From fcc6d71be2c3ce7d9305c04fc2e87af554571bac Mon Sep 17 00:00:00 2001
From: Dmitry Vyukov <dvyukov@google.com>
Date: Mon, 22 Feb 2021 20:37:25 +0100
Subject: go.mod: update golangci-lint to v1.37

---
 vendor/github.com/moricho/tparallel/testmap.go | 63 ++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 vendor/github.com/moricho/tparallel/testmap.go

(limited to 'vendor/github.com/moricho/tparallel/testmap.go')

diff --git a/vendor/github.com/moricho/tparallel/testmap.go b/vendor/github.com/moricho/tparallel/testmap.go
new file mode 100644
index 000000000..fa9bed708
--- /dev/null
+++ b/vendor/github.com/moricho/tparallel/testmap.go
@@ -0,0 +1,63 @@
+package tparallel
+
+import (
+	"go/types"
+	"strings"
+
+	"github.com/gostaticanalysis/analysisutil"
+	"golang.org/x/tools/go/analysis/passes/buildssa"
+	"golang.org/x/tools/go/ssa"
+
+	"github.com/moricho/tparallel/pkg/ssainstr"
+)
+
+// getTestMap gets a set of a top-level test and its sub-tests
+func getTestMap(ssaanalyzer *buildssa.SSA, testTyp types.Type) map[*ssa.Function][]*ssa.Function {
+	testMap := map[*ssa.Function][]*ssa.Function{}
+
+	trun := analysisutil.MethodOf(testTyp, "Run")
+	for _, f := range ssaanalyzer.SrcFuncs {
+		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 && ssainstr.HasArgs(instr, types.NewPointer(testTyp)) {
+					if instrs, ok := ssainstr.LookupCalled(instr, trun); ok {
+						for _, v := range instrs {
+							testMap[f] = appendTestMap(testMap[f], v)
+						}
+					}
+				} else if called {
+					testMap[f] = appendTestMap(testMap[f], instr)
+				}
+			}
+		}
+	}
+
+	return testMap
+}
+
+// appendTestMap converts ssa.Instruction to ssa.Function and append it to a given sub-test slice
+func appendTestMap(subtests []*ssa.Function, instr ssa.Instruction) []*ssa.Function {
+	call, ok := instr.(ssa.CallInstruction)
+	if !ok {
+		return subtests
+	}
+
+	ssaCall := call.Value()
+	for _, arg := range ssaCall.Call.Args {
+		switch arg := arg.(type) {
+		case *ssa.Function:
+			subtests = append(subtests, arg)
+		case *ssa.MakeClosure:
+			fn, _ := arg.Fn.(*ssa.Function)
+			subtests = append(subtests, fn)
+		}
+	}
+
+	return subtests
+}
-- 
cgit mrf-deployment