pkg may be nil at this point, assume main package

Этот коммит содержится в:
gedi 2016-06-16 10:19:53 +03:00
родитель c9a9f3c1da
коммит a6a318322b

Просмотреть файл

@ -234,6 +234,7 @@ func uniqStringList(strs []string) (unique []string) {
func buildTestMain(pkg *build.Package) ([]byte, bool, error) { func buildTestMain(pkg *build.Package) ([]byte, bool, error) {
var contexts []string var contexts []string
var importPath string var importPath string
name := "main"
if nil != pkg { if nil != pkg {
ctxs, err := processPackageTestFiles( ctxs, err := processPackageTestFiles(
pkg.TestGoFiles, pkg.TestGoFiles,
@ -244,13 +245,14 @@ func buildTestMain(pkg *build.Package) ([]byte, bool, error) {
} }
contexts = ctxs contexts = ctxs
importPath = pkg.ImportPath importPath = pkg.ImportPath
name = pkg.Name
} }
data := struct { data := struct {
Name string Name string
Contexts []string Contexts []string
ImportPath string ImportPath string
}{pkg.Name, contexts, importPath} }{name, contexts, importPath}
var buf bytes.Buffer var buf bytes.Buffer
if err := runnerTemplate.Execute(&buf, data); err != nil { if err := runnerTemplate.Execute(&buf, data); err != nil {