shad-go/tools/testtool/commands/list.go
2020-02-13 23:57:34 +03:00

125 lines
2.9 KiB
Go

package commands
import (
"log"
"os"
"sort"
"strings"
"golang.org/x/tools/go/packages"
)
// getPackageFiles returns absolute paths for all files in rootPackage and it's subpackages
// including tests and non-go files.
func getPackageFiles(rootPackage string, buildFlags []string) map[string]struct{} {
cfg := &packages.Config{
Dir: rootPackage,
Mode: packages.NeedFiles,
BuildFlags: buildFlags,
Tests: true,
}
pkgs, err := packages.Load(cfg, "./...")
if err != nil {
log.Fatalf("unable to load packages %s: %s", rootPackage, err)
}
if packages.PrintErrors(pkgs) > 0 {
os.Exit(1)
}
files := make(map[string]struct{})
for _, p := range pkgs {
for _, f := range p.GoFiles {
files[f] = struct{}{}
}
//for _, f := range p.OtherFiles {
// files[f] = struct{}{}
//}
}
return files
}
// listTestFiles returns absolute paths for all _test.go files of the package
// including the ones with "private" build tag.
func listTestFiles(rootPackage string) []string {
files := getPackageFiles(rootPackage, []string{"-tags", "private"})
var tests []string
for f := range files {
if strings.HasSuffix(f, "_test.go") {
tests = append(tests, f)
}
}
sort.Strings(tests)
return tests
}
// listProtectedFiles returns absolute paths for all files of the package
// protected by "!change" build tag.
func listProtectedFiles(rootPackage string) []string {
allFiles := getPackageFiles(rootPackage, nil)
allFilesWithoutProtected := getPackageFiles(rootPackage, []string{"-tags", "change"})
var protectedFiles []string
for f := range allFiles {
if _, ok := allFilesWithoutProtected[f]; !ok {
protectedFiles = append(protectedFiles, f)
}
}
sort.Strings(protectedFiles)
return protectedFiles
}
// listPrivateFiles returns absolute paths for all files of the package
// protected by "private,solution" build tag.
func listPrivateFiles(rootPackage string) []string {
allFiles := getPackageFiles(rootPackage, []string{})
allWithPrivate := getPackageFiles(rootPackage, []string{"-tags", "private,solution"})
var files []string
for f := range allWithPrivate {
if _, isPublic := allFiles[f]; !isPublic {
files = append(files, f)
}
}
sort.Strings(files)
return files
}
func listTestsAndBinaries(rootDir string, buildFlags []string) (binaries, tests map[string]struct{}) {
cfg := &packages.Config{
Dir: rootDir,
Mode: packages.NeedName | packages.NeedFiles,
BuildFlags: buildFlags,
Tests: true,
}
pkgs, err := packages.Load(cfg, "./...")
if err != nil {
log.Fatalf("unable to load packages %s: %s", rootDir, err)
}
if packages.PrintErrors(pkgs) > 0 {
os.Exit(1)
}
tests = map[string]struct{}{}
binaries = map[string]struct{}{}
for _, p := range pkgs {
if p.Name != "main" {
continue
}
if strings.HasSuffix(p.PkgPath, ".test") {
tests[strings.TrimSuffix(p.PkgPath, ".test")] = struct{}{}
} else {
binaries[p.PkgPath] = struct{}{}
}
}
return
}