5bcc9176 |
package main
import (
"bufio"
"bytes"
"flag"
"fmt"
"os"
"os/exec"
"path/filepath"
"regexp"
"sort"
"strings"
"log"
)
var reLine = regexp.MustCompile("^\\s*(\\w+)\\s+(\\w+)\\s+(.+)$")
func main() {
log.SetFlags(0)
prefix := flag.String("prefix", "", "the function prefix to scan for; if not specified will use *<dirBasename>.Test*")
help := flag.Bool("help", false, "display help")
flag.Parse()
if *help {
flag.PrintDefaults()
return
}
args := flag.Args()
if len(args) == 0 {
log.Fatalf("Must specify the name of a single directory, e.g. ./test/integration")
}
|
2ff1ecf3 |
test, err := filepath.Abs(args[0]) |
5bcc9176 |
if err != nil { |
2ff1ecf3 |
log.Fatalf("Unable to make path %q absolute: %v", args[0], err) |
5bcc9176 |
}
if _, err := os.Stat(test); err != nil {
log.Fatalf("No test executable %q exits, did you run `go test -c` on the named package?", test)
}
cmd := exec.Command("go", "tool", "nm", "-sort", "name", "-n", test)
out, err := cmd.CombinedOutput()
if err != nil {
log.Fatalf("Can't get `go tool nm` output from %q: %v", test, err)
}
names := []string{}
scanner := bufio.NewScanner(bytes.NewReader(out))
for scanner.Scan() {
line := scanner.Text()
match := reLine.FindStringSubmatch(line)
if len(match) == 0 {
//log.Printf("mismatch: %s", line)
continue
}
// ignore non code segments |
9d045f67 |
if match[2] != "t" && match[2] != "T" {
// log.Printf("non-code line: %s",line) |
5bcc9176 |
continue
}
name := match[3]
// there are always two segments per function, ignore the extra one
if strings.HasSuffix(name, ".f") {
continue
}
segments := strings.SplitAfter(name, "/")
// expect a package and name
if len(segments) < 2 {
//log.Printf("root_package: %s", name)
continue
}
last := segments[len(segments)-1]
_ = segments[len(segments)-2]
// two parts
parts := strings.Split(last, ".")
if len(parts) != 2 {
//log.Printf("bad_name: %s", last)
continue
}
test := parts[1]
if len(*prefix) == 0 {
if !strings.HasPrefix(test, "Test") {
continue
}
} else {
if !strings.HasPrefix(name, *prefix) {
continue
}
}
names = append(names, test)
}
|
4ac18055 |
if len(names) == 0 {
log.Fatalf("No tests found!")
} |
5bcc9176 |
sort.Sort(sort.StringSlice(names))
for _, test := range names {
fmt.Printf("%s\n", test)
}
if err := scanner.Err(); err != nil {
log.Fatalf("Unable to scan the symbol output: %v", err)
}
} |