imports: make tests use a helper, read from disk more
Replace the repeated Process/err check/output block with a test helper
eliminating a lot of copy/pasting. Change many tests to get their input
from disk rather than the argument to Process; this is mostly me being
selfish: go/packages doesn't support creating new packages with overlays
and that breaks a lot of these tests.
Change-Id: Ib1b7908c3d86d17323a94f1ee4ce074e264327b4
Reviewed-on: https://go-review.googlesource.com/c/141680
Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
Run-TryBot: Heschi Kreinick <heschi@google.com>
TryBot-Result: Gobot Gobot <gobot@golang.org>
diff --git a/imports/fix_test.go b/imports/fix_test.go
index 8eb5e50..bb3d158 100644
--- a/imports/fix_test.go
+++ b/imports/fix_test.go
@@ -1247,37 +1247,29 @@
func TestSimpleCases(t *testing.T) {
defer func(lp string) { LocalPrefix = lp }(LocalPrefix)
LocalPrefix = "local,github.com/local"
-
- testConfig{
- // Skeleton non-stdlib packages for use during testing.
- // Each includes one arbitrary symbol, e.g. the first declaration in the first file.
- // Try not to add more without a good reason.
- gopathFiles: map[string]string{
- "appengine/x.go": "package appengine\nfunc Main(){}\n",
- "appengine/datastore/x.go": "package datastore\nvar ErrInvalidEntityType error\n",
- "rsc.io/p/x.go": "package p\nfunc P(){}\n",
- "code.google.com/p/snappy-go/snappy/x.go": "package snappy\nvar ErrCorrupt error\n",
- },
- }.test(t, func(t *goimportTest) {
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- options := &Options{
- TabWidth: 8,
- TabIndent: true,
- Comments: true,
- Fragment: true,
- FormatOnly: tt.formatOnly,
- }
- buf, err := Process(tt.name+".go", []byte(tt.in), options)
- if err != nil {
- t.Fatalf("Process(...) = %v", err)
- }
- if got := string(buf); got != tt.out {
- t.Errorf("output differs:\nGOT:\n%s\nWANT:\n%s\n", got, tt.out)
- }
- })
- }
- })
+ for _, tt := range tests {
+ t.Run(tt.name, func(t *testing.T) {
+ options := &Options{
+ TabWidth: 8,
+ TabIndent: true,
+ Comments: true,
+ Fragment: true,
+ FormatOnly: tt.formatOnly,
+ }
+ testConfig{
+ // Skeleton non-stdlib packages for use during testing.
+ // Each includes one arbitrary symbol, e.g. the first declaration in the first file.
+ // Try not to add more without a good reason.
+ gopathFiles: map[string]string{
+ "appengine/x.go": "package appengine\nfunc Main(){}\n",
+ "appengine/datastore/x.go": "package datastore\nvar ErrInvalidEntityType error\n",
+ "rsc.io/p/x.go": "package p\nfunc P(){}\n",
+ "code.google.com/p/snappy-go/snappy/x.go": "package snappy\nvar ErrCorrupt error\n",
+ "x/x.go": tt.in,
+ },
+ }.processTest(t, "x/x.go", nil, options, tt.out)
+ })
+ }
}
@@ -1321,26 +1313,17 @@
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
+ options := &Options{
+ TabWidth: 8,
+ TabIndent: true,
+ Comments: true,
+ Fragment: true,
+ }
testConfig{
gopathFiles: map[string]string{
"x.go": tt.in,
},
- }.test(t, func(t *goimportTest) {
- options := &Options{
- TabWidth: 8,
- TabIndent: true,
- Comments: true,
- Fragment: true,
- }
- buf, err := Process(filepath.Join(t.gopath, "/src/x.go"), nil, options)
- if err != nil {
- t.Errorf("Process(...) = %v", tt.name, err)
-
- }
- if got := string(buf); got != tt.out {
- t.Errorf("output differs:\nGOT:\n%s\nWANT:\n%s\n", tt.name, got, tt.out)
- }
- })
+ }.processTest(t, "x.go", nil, options, tt.out)
})
}
@@ -1354,23 +1337,14 @@
t.Skipf("skipping test on %q as there are no symlinks", runtime.GOOS)
}
- testConfig{
- gopathFiles: map[string]string{
- "../target/f.go": "package mypkg\nvar Foo = 123\n",
- "x/mypkg": "LINK:../../target", // valid symlink
- "x/apkg": "LINK:..", // symlink loop
- },
- }.test(t, func(t *goimportTest) {
- build.Default.GOPATH = t.gopath
-
- input := `package p
+ const input = `package p
var (
_ = fmt.Print
_ = mypkg.Foo
)
`
- output := `package p
+ const want = `package p
import (
"fmt"
@@ -1382,14 +1356,15 @@
_ = mypkg.Foo
)
`
- buf, err := Process(t.gopath+"/src/myotherpkg/toformat.go", []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != output {
- t.Fatalf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, output)
- }
- })
+
+ testConfig{
+ gopathFiles: map[string]string{
+ "../target/f.go": "package mypkg\nvar Foo = 123\n",
+ "x/mypkg": "LINK:../../target", // valid symlink
+ "x/apkg": "LINK:..", // symlink loop
+ "myotherpackage/toformat.go": input,
+ },
+ }.processTest(t, "myotherpackage/toformat.go", nil, nil, want)
}
func TestImportSymlinksWithIgnore(t *testing.T) {
@@ -1398,22 +1373,14 @@
t.Skipf("skipping test on %q as there are no symlinks", runtime.GOOS)
}
- testConfig{
- gopathFiles: map[string]string{
- "../target/f.go": "package mypkg\nvar Foo = 123\n",
- "x/mypkg": "LINK:../../target", // valid symlink
- "x/apkg": "LINK:..", // symlink loop
- ".goimportsignore": "x/mypkg\n",
- },
- }.test(t, func(t *goimportTest) {
- input := `package p
+ const input = `package p
var (
_ = fmt.Print
_ = mypkg.Foo
)
`
- output := `package p
+ const want = `package p
import "fmt"
@@ -1422,20 +1389,21 @@
_ = mypkg.Foo
)
`
- buf, err := Process(t.gopath+"/src/myotherpkg/toformat.go", []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != output {
- t.Fatalf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, output)
- }
- })
+
+ testConfig{
+ gopathFiles: map[string]string{
+ "../target/f.go": "package mypkg\nvar Foo = 123\n",
+ "x/mypkg": "LINK:../../target", // valid symlink
+ "x/apkg": "LINK:..", // symlink loop
+ "myotherpkg/toformat.go": input,
+ ".goimportsignore": "x/mypkg\n",
+ },
+ }.processTest(t, "myotherpkg/toformat.go", nil, nil, want)
}
// Test for x/y/v2 convention for package y.
func TestModuleVersion(t *testing.T) {
- testConfig{}.test(t, func(t *goimportTest) {
- input := `package p
+ const input = `package p
import (
"fmt"
@@ -1448,14 +1416,12 @@
_ = foo.Foo
)
`
- buf, err := Process(filepath.Join(t.gopath, "src/mypkg.com/outpkg/toformat.go"), []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != input {
- t.Fatalf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, input)
- }
- })
+
+ testConfig{
+ gopathFiles: map[string]string{
+ "mypkg.com/outpkg/toformat.go": input,
+ },
+ }.processTest(t, "mypkg.com/outpkg/toformat.go", nil, nil, input)
}
// Test for correctly identifying the name of a vendored package when it
@@ -1463,12 +1429,7 @@
// "mypkg.com/mypkg.v1" would be removed if goimports wasn't able to detect
// that the package name is "mypkg".
func TestVendorPackage(t *testing.T) {
- testConfig{
- gopathFiles: map[string]string{
- "mypkg.com/outpkg/vendor/mypkg.com/mypkg.v1/f.go": "package mypkg\nvar Foo = 123\n",
- },
- }.test(t, func(t *goimportTest) {
- input := `package p
+ const input = `package p
import (
"fmt"
@@ -1481,32 +1442,12 @@
_ = mypkg.Foo
)
`
- buf, err := Process(filepath.Join(t.gopath, "src/mypkg.com/outpkg/toformat.go"), []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != input {
- t.Fatalf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, input)
- }
- })
-}
-
-func withEmptyGoPath(fn func()) {
- scanOnce = sync.Once{}
-
- oldGOPATH := build.Default.GOPATH
- oldGOROOT := build.Default.GOROOT
- oldCompiler := build.Default.Compiler
- build.Default.GOPATH = ""
- build.Default.Compiler = "gc"
-
- defer func() {
- build.Default.GOPATH = oldGOPATH
- build.Default.GOROOT = oldGOROOT
- build.Default.Compiler = oldCompiler
- }()
-
- fn()
+ testConfig{
+ gopathFiles: map[string]string{
+ "mypkg.com/outpkg/vendor/mypkg.com/mypkg.v1/f.go": "package mypkg\nvar Foo = 123\n",
+ "mypkg.com/outpkg/toformat.go": input,
+ },
+ }.processTest(t, "mypkg.com/outpkg/toformat.go", nil, nil, input)
}
func TestInternal(t *testing.T) {
@@ -1521,31 +1462,21 @@
var _ = race.Acquire
`
- tc := testConfig{
+ // Packages under the same directory should be able to use internal packages.
+ testConfig{
gopathFiles: map[string]string{
"foo/internal/race/x.go": "package race\n func Acquire(){}\n",
+ "foo/bar/x.go": input,
},
- }
- // Packages under the same directory should be able to use internal packages.
- tc.test(t, func(t *goimportTest) {
- buf, err := Process(filepath.Join(t.gopath, "src/foo/bar/x.go"), []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != importAdded {
- t.Errorf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, importAdded)
- }
- })
+ }.processTest(t, "foo/bar/x.go", nil, nil, importAdded)
+
// Packages outside the same directory should not.
- tc.test(t, func(t *goimportTest) {
- buf, err := Process(filepath.Join(t.gopath, "src/bar/x.go"), []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != input {
- t.Errorf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, input)
- }
- })
+ testConfig{
+ gopathFiles: map[string]string{
+ "foo/internal/race/x.go": "package race\n func Acquire(){}\n",
+ "bar/x.go": input,
+ },
+ }.processTest(t, "bar/x.go", nil, nil, input)
}
func TestProcessVendor(t *testing.T) {
@@ -1562,16 +1493,9 @@
testConfig{
gopathFiles: map[string]string{
"vendor/golang.org/x/net/http2/hpack/huffman.go": "package hpack\nfunc HuffmanDecode() { }\n",
+ "bar/x.go": input,
},
- }.test(t, func(t *goimportTest) {
- buf, err := Process(filepath.Join(t.gopath, "src/bar/x.go"), []byte(input), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if got := string(buf); got != want {
- t.Errorf("results differ\nGOT:\n%s\nWANT:\n%s\n", got, want)
- }
- })
+ }.processTest(t, "bar/x.go", nil, nil, want)
}
func TestFindStdlib(t *testing.T) {
@@ -1617,6 +1541,7 @@
}
func mustTempDir(t *testing.T, prefix string) string {
+ t.Helper()
dir, err := ioutil.TempDir("", prefix)
if err != nil {
t.Fatal(err)
@@ -1644,6 +1569,8 @@
}
func (c testConfig) test(t *testing.T, fn func(*goimportTest)) {
+ t.Helper()
+
goroot := c.goroot
gopath := c.gopath
@@ -1663,19 +1590,39 @@
t.Fatal(err)
}
- withEmptyGoPath(func() {
- if goroot != "" {
- build.Default.GOROOT = goroot
- }
- build.Default.GOPATH = gopath
+ scanOnce = sync.Once{}
- it := &goimportTest{
- T: t,
- goroot: build.Default.GOROOT,
- gopath: gopath,
- ctx: &build.Default,
- }
- fn(it)
+ oldGOPATH := build.Default.GOPATH
+ oldGOROOT := build.Default.GOROOT
+ oldCompiler := build.Default.Compiler
+ build.Default.GOPATH = ""
+ build.Default.Compiler = "gc"
+
+ defer func() {
+ build.Default.GOPATH = oldGOPATH
+ build.Default.GOROOT = oldGOROOT
+ build.Default.Compiler = oldCompiler
+ }()
+
+ if goroot != "" {
+ build.Default.GOROOT = goroot
+ }
+ build.Default.GOPATH = gopath
+
+ it := &goimportTest{
+ T: t,
+ goroot: build.Default.GOROOT,
+ gopath: gopath,
+ ctx: &build.Default,
+ }
+ fn(it)
+}
+
+func (c testConfig) processTest(t *testing.T, file string, contents []byte, opts *Options, want string) {
+ t.Helper()
+ c.test(t, func(t *goimportTest) {
+ t.Helper()
+ t.process(file, contents, opts, want)
})
}
@@ -1686,30 +1633,37 @@
gopath string
}
+func (t *goimportTest) process(file string, contents []byte, opts *Options, want string) {
+ t.Helper()
+ buf, err := Process(filepath.Join(t.gopath, "src", file), contents, opts)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if string(buf) != want {
+ t.Errorf("Got:\n%s\nWant:\n%s", buf, want)
+ }
+}
+
// Tests that added imports are renamed when the import path's base doesn't
// match its package name. For example, we want to generate:
//
// import cloudbilling "google.golang.org/api/cloudbilling/v1"
func TestRenameWhenPackageNameMismatch(t *testing.T) {
- testConfig{
- gopathFiles: map[string]string{
- "foo/bar/v1/x.go": "package bar \n const X = 1",
- },
- }.test(t, func(t *goimportTest) {
- buf, err := Process(t.gopath+"/src/test/t.go", []byte("package main \n const Y = bar.X"), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- const want = `package main
+ const input = `package main
+ const Y = bar.X`
+
+ const want = `package main
import bar "foo/bar/v1"
const Y = bar.X
`
- if string(buf) != want {
- t.Errorf("Got:\n%s\nWant:\n%s", buf, want)
- }
- })
+ testConfig{
+ gopathFiles: map[string]string{
+ "foo/bar/v1/x.go": "package bar \n const X = 1",
+ "test/t.go": input,
+ },
+ }.processTest(t, "test/t.go", nil, nil, want)
}
// Tests that the LocalPrefix option causes imports
@@ -1796,35 +1750,31 @@
tt.config.test(t, func(t *goimportTest) {
defer func(s string) { LocalPrefix = s }(LocalPrefix)
LocalPrefix = tt.localPrefix
- buf, err := Process(t.gopath+"/src/test/t.go", []byte(tt.src), &Options{})
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != tt.want {
- t.Errorf("Got:\n%s\nWant:\n%s", buf, tt.want)
- }
+ t.process("test/t.go", []byte(tt.src), nil, tt.want)
})
}
}
// Tests that "package documentation" files are ignored.
func TestIgnoreDocumentationPackage(t *testing.T) {
+ const input = `package x
+
+const Y = foo.X
+`
+ const want = `package x
+
+import "foo"
+
+const Y = foo.X
+`
+
testConfig{
gopathFiles: map[string]string{
"foo/foo.go": "package foo\nconst X = 1\n",
"foo/doc.go": "package documentation \n // just to confuse things\n",
+ "x/x.go": input,
},
- }.test(t, func(t *goimportTest) {
- const in = "package x\n\nconst Y = foo.X\n"
- const want = "package x\n\nimport \"foo\"\n\nconst Y = foo.X\n"
- buf, err := Process(t.gopath+"/src/x/x.go", []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != want {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", in, want)
- }
- })
+ }.processTest(t, "x/x.go", nil, nil, want)
}
// Tests importPathToNameGoPathParse and in particular that it stops
@@ -1852,47 +1802,62 @@
}
func TestIgnoreConfiguration(t *testing.T) {
+ const input = `package x
+
+const _ = pkg.X
+`
+ const want = `package x
+
+import "otherwise-longer-so-worse.example.net/foo/pkg"
+
+const _ = pkg.X
+`
+
testConfig{
gopathFiles: map[string]string{
".goimportsignore": "# comment line\n\n example.net", // tests comment, blank line, whitespace trimming
"example.net/pkg/pkg.go": "package pkg\nconst X = 1",
"otherwise-longer-so-worse.example.net/foo/pkg/pkg.go": "package pkg\nconst X = 1",
+ "x/x.go": input,
},
- }.test(t, func(t *goimportTest) {
- const in = "package x\n\nconst _ = pkg.X\n"
- const want = "package x\n\nimport \"otherwise-longer-so-worse.example.net/foo/pkg\"\n\nconst _ = pkg.X\n"
- buf, err := Process(t.gopath+"/src/x/x.go", []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != want {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", buf, want)
- }
- })
+ }.processTest(t, "x/x.go", nil, nil, want)
}
// Skip "node_modules" directory.
func TestSkipNodeModules(t *testing.T) {
+ const input = `package x
+
+const _ = pkg.X
+`
+ const want = `package x
+
+import "otherwise-longer.net/not_modules/pkg"
+
+const _ = pkg.X
+`
+
testConfig{
gopathFiles: map[string]string{
"example.net/node_modules/pkg/a.go": "package pkg\nconst X = 1",
"otherwise-longer.net/not_modules/pkg/a.go": "package pkg\nconst X = 1",
+ "x/x.go": input,
},
- }.test(t, func(t *goimportTest) {
- const in = "package x\n\nconst _ = pkg.X\n"
- const want = "package x\n\nimport \"otherwise-longer.net/not_modules/pkg\"\n\nconst _ = pkg.X\n"
- buf, err := Process(t.gopath+"/src/x/x.go", []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != want {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", buf, want)
- }
- })
+ }.processTest(t, "x/x.go", nil, nil, want)
}
// golang.org/issue/16458 -- if GOROOT is a prefix of GOPATH, GOPATH is ignored.
func TestGoRootPrefixOfGoPath(t *testing.T) {
+ const input = `package x
+
+const _ = foo.X
+`
+ const want = `package x
+
+import "example.com/foo"
+
+const _ = foo.X
+`
+
dir := mustTempDir(t, "importstest")
defer os.RemoveAll(dir)
testConfig{
@@ -1900,19 +1865,9 @@
gopath: filepath.Join(dir, "gopath"),
gopathFiles: map[string]string{
"example.com/foo/pkg.go": "package foo\nconst X = 1",
+ "x/x.go": input,
},
- }.test(t, func(t *goimportTest) {
- const in = "package x\n\nconst _ = foo.X\n"
- const want = "package x\n\nimport \"example.com/foo\"\n\nconst _ = foo.X\n"
- buf, err := Process(t.gopath+"/src/x/x.go", []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != want {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", buf, want)
- }
- })
-
+ }.processTest(t, "x/x.go", nil, nil, want)
}
// Tests that package global variables with the same name and function name as
@@ -1939,18 +1894,10 @@
testConfig{
gopathFiles: map[string]string{
+ "pkg/uses.go": usesGlobal,
"pkg/global.go": declaresGlobal,
},
- }.test(t, func(t *goimportTest) {
- buf, err := Process(
- filepath.Join(t.gopath, "src/pkg/uses.go"), []byte(usesGlobal), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != usesGlobal {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", buf, usesGlobal)
- }
- })
+ }.processTest(t, "pkg/uses.go", nil, nil, usesGlobal)
}
// Tests that sibling files - other files in the same package - can provide an
@@ -1994,23 +1941,12 @@
}
`
- const pkg = "siblingimporttest"
- const siblingFile = pkg + "/needs_import.go"
testConfig{
gopathFiles: map[string]string{
- siblingFile: need,
- pkg + "/provides_import.go": provide,
+ "p/needs_import.go": need,
+ "p/provides_import.go": provide,
},
- }.test(t, func(t *goimportTest) {
- buf, err := Process(
- t.gopath+"/src/"+siblingFile, []byte(need), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(buf) != want {
- t.Errorf("wrong output.\ngot:\n%q\nwant:\n%q\n", buf, want)
- }
- })
+ }.processTest(t, "p/needs_import.go", nil, nil, want)
}
func TestPkgIsCandidate(t *testing.T) {
@@ -2169,28 +2105,24 @@
// should be the closest match.
// https://golang.org/issues/17557
func TestLocalPackagePromotion(t *testing.T) {
- testConfig{
- gopathFiles: map[string]string{
- "config.net/config/config.go": "package config\n type SystemConfig struct {}", // Will match but should not be first choice
- "mycompany.net/config/config.go": "package config\n type SystemConfig struct {}", // Will match but should not be first choice
- "mycompany.net/tool/config/config.go": "package config\n type SystemConfig struct {}", // Local package should be promoted over shorter package
- },
- }.test(t, func(t *goimportTest) {
- const in = "package main\n var c = &config.SystemConfig{}"
- const want = `package main
+ const input = `package main
+var c = &config.SystemConfig{}
+`
+ const want = `package main
import "mycompany.net/tool/config"
var c = &config.SystemConfig{}
`
- got, err := Process(filepath.Join(t.gopath, "src", "mycompany.net/tool/main.go"), []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(got) != want {
- t.Errorf("Process = %q; want %q", got, want)
- }
- })
+
+ testConfig{
+ gopathFiles: map[string]string{
+ "config.net/config/config.go": "package config\n type SystemConfig struct {}", // Will match but should not be first choice
+ "mycompany.net/config/config.go": "package config\n type SystemConfig struct {}", // Will match but should not be first choice
+ "mycompany.net/tool/config/config.go": "package config\n type SystemConfig struct {}", // Local package should be promoted over shorter package
+ "mycompany.net/tool/main.go": input,
+ },
+ }.processTest(t, "mycompany.net/tool/main.go", nil, nil, want)
}
// Tests FindImportInLocalGoFiles looks at the import lines for other Go files in the
@@ -2199,49 +2131,37 @@
// standard library.
// https://golang.org/issues/17557
func TestFindImportInLocalGoFiles(t *testing.T) {
- testConfig{
- gopathFiles: map[string]string{
- "bytes.net/bytes/bytes.go": "package bytes\n type Buffer struct {}", // Should be selected over standard library
- "mycompany.net/tool/io.go": "package main\n import \"bytes.net/bytes\"\n var _ = &bytes.Buffer{}", // Contains package import that will cause stdlib to be ignored
- "mycompany.net/tool/err.go": "package main\n import \"bogus.net/bytes\"\n var _ = &bytes.Buffer{}", // Contains import which is not resolved, so it is ignored
- },
- }.test(t, func(t *goimportTest) {
- const in = "package main\n var _ = &bytes.Buffer{}"
- const want = `package main
+ const input = `package main
+ var _ = &bytes.Buffer{}`
+
+ const want = `package main
import "bytes.net/bytes"
var _ = &bytes.Buffer{}
`
- got, err := Process(filepath.Join(t.gopath, "src", "mycompany.net/tool/main.go"), []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(got) != want {
- t.Errorf("Process = got %q; want %q", got, want)
- }
- })
+ testConfig{
+ gopathFiles: map[string]string{
+ "bytes.net/bytes/bytes.go": "package bytes\n type Buffer struct {}", // Should be selected over standard library
+ "mycompany.net/tool/io.go": "package main\n import \"bytes.net/bytes\"\n var _ = &bytes.Buffer{}", // Contains package import that will cause stdlib to be ignored
+ "mycompany.net/tool/err.go": "package main\n import \"bogus.net/bytes\"\n var _ = &bytes.Buffer{}", // Contains import which is not resolved, so it is ignored
+ "mycompany.net/tool/main.go": input,
+ },
+ }.processTest(t, "mycompany.net/tool/main.go", nil, nil, want)
}
func TestImportNoGoFiles(t *testing.T) {
- testConfig{
- gopathFiles: map[string]string{},
- }.test(t, func(t *goimportTest) {
- const in = "package main\n var _ = &bytes.Buffer{}"
- const want = `package main
+ const input = `package main
+ var _ = &bytes.Buffer{}`
+
+ const want = `package main
import "bytes"
var _ = &bytes.Buffer{}
`
- got, err := Process(filepath.Join(t.gopath, "src", "mycompany.net/tool/main.go"), []byte(in), nil)
- if err != nil {
- t.Fatal(err)
- }
- if string(got) != want {
- t.Errorf("Process = got %q; want %q", got, want)
- }
- })
+
+ testConfig{}.processTest(t, "mycompany.net/tool/main.go", []byte(input), nil, want)
}
// Ensures a token as large as 500000 bytes can be handled
@@ -2249,7 +2169,7 @@
func TestProcessLargeToken(t *testing.T) {
largeString := strings.Repeat("x", 500000)
- in := `package testimports
+ input := `package testimports
import (
"fmt"
@@ -2262,12 +2182,6 @@
// end
`
- out, err := Process("foo", []byte(in), nil)
-
- if err != nil {
- t.Errorf("Process returned error.\n got:\n%v\nwant:\nnil", err)
- }
-
want := `package testimports
import (
@@ -2281,7 +2195,8 @@
// end
`
- if got := string(out); got != want {
- t.Errorf("Process returned unexpected result.\ngot:\n%.100v\nwant:\n%.100v", got, want)
- }
+
+ testConfig{
+ gopathFiles: map[string]string{"foo.go": input},
+ }.processTest(t, "foo.go", nil, nil, want)
}