65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"go/ast"
|
|
"go/parser"
|
|
"go/token"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/posener/complete"
|
|
"github.com/posener/complete/match"
|
|
)
|
|
|
|
// predictTest predict test names.
|
|
// it searches in the current directory for all the go test files
|
|
// and then all the relevant function names.
|
|
// for test names use prefix of 'Test' or 'Example', and for benchmark
|
|
// test names use 'Benchmark'
|
|
func predictTest(funcPrefix ...string) complete.Predicate {
|
|
return func(last string) []match.Matcher {
|
|
tests := testNames(funcPrefix)
|
|
options := make([]match.Matcher, len(tests))
|
|
for i := range tests {
|
|
options[i] = match.Prefix(tests[i])
|
|
}
|
|
return options
|
|
}
|
|
}
|
|
|
|
// get all test names in current directory
|
|
func testNames(funcPrefix []string) (tests []string) {
|
|
filepath.Walk("./", func(path string, info os.FileInfo, err error) error {
|
|
// if not a test file, skip
|
|
if !strings.HasSuffix(path, "_test.go") {
|
|
return nil
|
|
}
|
|
// inspect test file and append all the test names
|
|
tests = append(tests, testsInFile(funcPrefix, path)...)
|
|
return nil
|
|
})
|
|
return
|
|
}
|
|
|
|
func testsInFile(funcPrefix []string, path string) (tests []string) {
|
|
fset := token.NewFileSet()
|
|
f, err := parser.ParseFile(fset, path, nil, 0)
|
|
if err != nil {
|
|
complete.Log("Failed parsing %s: %s", path, err)
|
|
return nil
|
|
}
|
|
for _, d := range f.Decls {
|
|
if f, ok := d.(*ast.FuncDecl); ok {
|
|
name := f.Name.String()
|
|
for _, prefix := range funcPrefix {
|
|
if strings.HasPrefix(name, prefix) {
|
|
tests = append(tests, name)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|