| // Copyright 2011 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package main |
| |
| import ( |
| "bytes" |
| "flag" |
| "fmt" |
| "go/build" |
| "io/ioutil" |
| "os" |
| "os/exec" |
| "path/filepath" |
| "sort" |
| "strings" |
| "testing" |
| ) |
| |
| var ( |
| updateGolden = flag.Bool("updategolden", false, "update golden files") |
| ) |
| |
| func TestGolden(t *testing.T) { |
| td, err := os.Open("testdata/src/pkg") |
| if err != nil { |
| t.Fatal(err) |
| } |
| fis, err := td.Readdir(0) |
| if err != nil { |
| t.Fatal(err) |
| } |
| for _, fi := range fis { |
| if !fi.IsDir() { |
| continue |
| } |
| |
| // TODO(gri) remove extra pkg directory eventually |
| goldenFile := filepath.Join("testdata", "src", "pkg", fi.Name(), "golden.txt") |
| w := NewWalker(nil, "testdata/src/pkg") |
| pkg, _ := w.Import(fi.Name()) |
| w.export(pkg) |
| |
| if *updateGolden { |
| os.Remove(goldenFile) |
| f, err := os.Create(goldenFile) |
| if err != nil { |
| t.Fatal(err) |
| } |
| for _, feat := range w.Features() { |
| fmt.Fprintf(f, "%s\n", feat) |
| } |
| f.Close() |
| } |
| |
| bs, err := ioutil.ReadFile(goldenFile) |
| if err != nil { |
| t.Fatalf("opening golden.txt for package %q: %v", fi.Name(), err) |
| } |
| wanted := strings.Split(string(bs), "\n") |
| sort.Strings(wanted) |
| for _, feature := range wanted { |
| if feature == "" { |
| continue |
| } |
| _, ok := w.features[feature] |
| if !ok { |
| t.Errorf("package %s: missing feature %q", fi.Name(), feature) |
| } |
| delete(w.features, feature) |
| } |
| |
| for _, feature := range w.Features() { |
| t.Errorf("package %s: extra feature not in golden file: %q", fi.Name(), feature) |
| } |
| } |
| } |
| |
| func TestCompareAPI(t *testing.T) { |
| tests := []struct { |
| name string |
| features, required, optional, exception []string |
| ok bool // want |
| out string // want |
| }{ |
| { |
| name: "feature added", |
| features: []string{"A", "B", "C", "D", "E", "F"}, |
| required: []string{"B", "D"}, |
| ok: true, |
| out: "+A\n+C\n+E\n+F\n", |
| }, |
| { |
| name: "feature removed", |
| features: []string{"C", "A"}, |
| required: []string{"A", "B", "C"}, |
| ok: false, |
| out: "-B\n", |
| }, |
| { |
| name: "feature added then removed", |
| features: []string{"A", "C"}, |
| optional: []string{"B"}, |
| required: []string{"A", "C"}, |
| ok: true, |
| out: "±B\n", |
| }, |
| { |
| name: "exception removal", |
| required: []string{"A", "B", "C"}, |
| features: []string{"A", "C"}, |
| exception: []string{"B"}, |
| ok: true, |
| out: "", |
| }, |
| { |
| // https://golang.org/issue/4303 |
| name: "contexts reconverging", |
| required: []string{ |
| "A", |
| "pkg syscall (darwin-386), type RawSockaddrInet6 struct", |
| "pkg syscall (darwin-amd64), type RawSockaddrInet6 struct", |
| }, |
| features: []string{ |
| "A", |
| "pkg syscall, type RawSockaddrInet6 struct", |
| }, |
| ok: true, |
| out: "+pkg syscall, type RawSockaddrInet6 struct\n", |
| }, |
| } |
| for _, tt := range tests { |
| buf := new(bytes.Buffer) |
| gotok := compareAPI(buf, tt.features, tt.required, tt.optional, tt.exception, true) |
| if gotok != tt.ok { |
| t.Errorf("%s: ok = %v; want %v", tt.name, gotok, tt.ok) |
| } |
| if got := buf.String(); got != tt.out { |
| t.Errorf("%s: output differs\nGOT:\n%s\nWANT:\n%s", tt.name, got, tt.out) |
| } |
| } |
| } |
| |
| func TestSkipInternal(t *testing.T) { |
| tests := []struct { |
| pkg string |
| want bool |
| }{ |
| {"net/http", true}, |
| {"net/http/internal-foo", true}, |
| {"net/http/internal", false}, |
| {"net/http/internal/bar", false}, |
| {"internal/foo", false}, |
| {"internal", false}, |
| } |
| for _, tt := range tests { |
| got := !internalPkg.MatchString(tt.pkg) |
| if got != tt.want { |
| t.Errorf("%s is internal = %v; want %v", tt.pkg, got, tt.want) |
| } |
| } |
| } |
| |
| func BenchmarkAll(b *testing.B) { |
| stds, err := exec.Command("go", "list", "std").Output() |
| if err != nil { |
| b.Fatal(err) |
| } |
| b.ResetTimer() |
| pkgNames := strings.Fields(string(stds)) |
| |
| for _, c := range contexts { |
| c.Compiler = build.Default.Compiler |
| } |
| |
| for i := 0; i < b.N; i++ { |
| for _, context := range contexts { |
| w := NewWalker(context, filepath.Join(build.Default.GOROOT, "src")) |
| for _, name := range pkgNames { |
| if name != "unsafe" && !strings.HasPrefix(name, "cmd/") && !internalPkg.MatchString(name) { |
| pkg, _ := w.Import(name) |
| w.export(pkg) |
| } |
| } |
| w.Features() |
| } |
| } |
| } |