func TestWalkSimple(t *testing.T) {
dir, err := tempDir()
if err != nil {
t.Fatalf("tempDir() failed with %v; want success", err)
}
defer os.RemoveAll(dir)
fname := filepath.Join(dir, "lib.go")
if err := ioutil.WriteFile(fname, []byte("package lib"), 0600); err != nil {
t.Fatalf(`ioutil.WriteFile(%q, "package lib", 0600) failed with %v; want success`, fname, err)
}
var n int
err = generator.Walk(build.Default, dir, func(pkg *build.Package) error {
if got, want := pkg.Name, "lib"; got != want {
t.Errorf("pkg.Name = %q; want %q", got, want)
}
n++
return nil
})
if err != nil {
t.Errorf("generator.Walk(build.Default, %q, func) failed with %v; want success", dir, err)
}
if got, want := n, 1; got != want {
t.Errorf("n = %d; want %d", got, want)
}
}
func TestWalkNested(t *testing.T) {
dir, err := tempDir()
if err != nil {
t.Fatalf("tempDir() failed with %v; want success", err)
}
defer os.RemoveAll(dir)
for _, p := range []struct {
path, content string
}{
{path: "a/foo.go", content: "package a"},
{path: "b/c/bar.go", content: "package c"},
{path: "b/d/baz.go", content: "package main"},
} {
path := filepath.Join(dir, p.path)
if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil {
t.Fatalf("os.MkdirAll(%q, 0700) failed with %v; want success", filepath.Dir(path), err)
}
if err := ioutil.WriteFile(path, []byte(p.content), 0600); err != nil {
t.Fatalf("ioutil.WriteFile(%q, %q, 0600) failed with %v; want success", path, p.content, err)
}
}
var dirs, pkgs []string
err = generator.Walk(build.Default, dir, func(pkg *build.Package) error {
rel, err := filepath.Rel(dir, pkg.Dir)
if err != nil {
t.Errorf("filepath.Rel(%q, %q) failed with %v; want success", dir, pkg.Dir, err)
return err
}
dirs = append(dirs, rel)
pkgs = append(pkgs, pkg.Name)
return nil
})
if err != nil {
t.Errorf("generator.Walk(build.Default, %q, func) failed with %v; want success", dir, err)
}
sort.Strings(dirs)
if got, want := dirs, []string{"a", "b/c", "b/d"}; !reflect.DeepEqual(got, want) {
t.Errorf("pkgs = %q; want %q", got, want)
}
sort.Strings(pkgs)
if got, want := pkgs, []string{"a", "c", "main"}; !reflect.DeepEqual(got, want) {
t.Errorf("pkgs = %q; want %q", got, want)
}
}