| // 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 build |
| |
| import ( |
| "io" |
| "os" |
| "path/filepath" |
| "reflect" |
| "runtime" |
| "strings" |
| "testing" |
| ) |
| |
| func TestMatch(t *testing.T) { |
| ctxt := Default |
| what := "default" |
| match := func(tag string, want map[string]bool) { |
| m := make(map[string]bool) |
| if !ctxt.match(tag, m) { |
| t.Errorf("%s context should match %s, does not", what, tag) |
| } |
| if !reflect.DeepEqual(m, want) { |
| t.Errorf("%s tags = %v, want %v", tag, m, want) |
| } |
| } |
| nomatch := func(tag string, want map[string]bool) { |
| m := make(map[string]bool) |
| if ctxt.match(tag, m) { |
| t.Errorf("%s context should NOT match %s, does", what, tag) |
| } |
| if !reflect.DeepEqual(m, want) { |
| t.Errorf("%s tags = %v, want %v", tag, m, want) |
| } |
| } |
| |
| match(runtime.GOOS+","+runtime.GOARCH, map[string]bool{runtime.GOOS: true, runtime.GOARCH: true}) |
| match(runtime.GOOS+","+runtime.GOARCH+",!foo", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "foo": true}) |
| nomatch(runtime.GOOS+","+runtime.GOARCH+",foo", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "foo": true}) |
| |
| what = "modified" |
| ctxt.BuildTags = []string{"foo"} |
| match(runtime.GOOS+","+runtime.GOARCH, map[string]bool{runtime.GOOS: true, runtime.GOARCH: true}) |
| match(runtime.GOOS+","+runtime.GOARCH+",foo", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "foo": true}) |
| nomatch(runtime.GOOS+","+runtime.GOARCH+",!foo", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "foo": true}) |
| match(runtime.GOOS+","+runtime.GOARCH+",!bar", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "bar": true}) |
| nomatch(runtime.GOOS+","+runtime.GOARCH+",bar", map[string]bool{runtime.GOOS: true, runtime.GOARCH: true, "bar": true}) |
| nomatch("!", map[string]bool{}) |
| } |
| |
| func TestDotSlashImport(t *testing.T) { |
| p, err := ImportDir("testdata/other", 0) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if len(p.Imports) != 1 || p.Imports[0] != "./file" { |
| t.Fatalf("testdata/other: Imports=%v, want [./file]", p.Imports) |
| } |
| |
| p1, err := Import("./file", "testdata/other", 0) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if p1.Name != "file" { |
| t.Fatalf("./file: Name=%q, want %q", p1.Name, "file") |
| } |
| dir := filepath.Clean("testdata/other/file") // Clean to use \ on Windows |
| if p1.Dir != dir { |
| t.Fatalf("./file: Dir=%q, want %q", p1.Name, dir) |
| } |
| } |
| |
| func TestEmptyImport(t *testing.T) { |
| p, err := Import("", Default.GOROOT, FindOnly) |
| if err == nil { |
| t.Fatal(`Import("") returned nil error.`) |
| } |
| if p == nil { |
| t.Fatal(`Import("") returned nil package.`) |
| } |
| if p.ImportPath != "" { |
| t.Fatalf("ImportPath=%q, want %q.", p.ImportPath, "") |
| } |
| } |
| |
| func TestLocalDirectory(t *testing.T) { |
| cwd, err := os.Getwd() |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| p, err := ImportDir(cwd, 0) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if p.ImportPath != "go/build" { |
| t.Fatalf("ImportPath=%q, want %q", p.ImportPath, "go/build") |
| } |
| } |
| |
| func TestShouldBuild(t *testing.T) { |
| const file1 = "// +build tag1\n\n" + |
| "package main\n" |
| want1 := map[string]bool{"tag1": true} |
| |
| const file2 = "// +build cgo\n\n" + |
| "// This package implements parsing of tags like\n" + |
| "// +build tag1\n" + |
| "package build" |
| want2 := map[string]bool{"cgo": true} |
| |
| const file3 = "// Copyright The Go Authors.\n\n" + |
| "package build\n\n" + |
| "// shouldBuild checks tags given by lines of the form\n" + |
| "// +build tag\n" + |
| "func shouldBuild(content []byte)\n" |
| want3 := map[string]bool{} |
| |
| ctx := &Context{BuildTags: []string{"tag1"}} |
| m := map[string]bool{} |
| if !ctx.shouldBuild([]byte(file1), m) { |
| t.Errorf("shouldBuild(file1) = false, want true") |
| } |
| if !reflect.DeepEqual(m, want1) { |
| t.Errorf("shoudBuild(file1) tags = %v, want %v", m, want1) |
| } |
| |
| m = map[string]bool{} |
| if ctx.shouldBuild([]byte(file2), m) { |
| t.Errorf("shouldBuild(file2) = true, want fakse") |
| } |
| if !reflect.DeepEqual(m, want2) { |
| t.Errorf("shoudBuild(file2) tags = %v, want %v", m, want2) |
| } |
| |
| m = map[string]bool{} |
| ctx = &Context{BuildTags: nil} |
| if !ctx.shouldBuild([]byte(file3), m) { |
| t.Errorf("shouldBuild(file3) = false, want true") |
| } |
| if !reflect.DeepEqual(m, want3) { |
| t.Errorf("shoudBuild(file3) tags = %v, want %v", m, want3) |
| } |
| } |
| |
| type readNopCloser struct { |
| io.Reader |
| } |
| |
| func (r readNopCloser) Close() error { |
| return nil |
| } |
| |
| var matchFileTests = []struct { |
| name string |
| data string |
| match bool |
| }{ |
| {"foo_arm.go", "", true}, |
| {"foo1_arm.go", "// +build linux\n\npackage main\n", false}, |
| {"foo_darwin.go", "", false}, |
| {"foo.go", "", true}, |
| {"foo1.go", "// +build linux\n\npackage main\n", false}, |
| {"foo.badsuffix", "", false}, |
| } |
| |
| func TestMatchFile(t *testing.T) { |
| for _, tt := range matchFileTests { |
| ctxt := Context{GOARCH: "arm", GOOS: "plan9"} |
| ctxt.OpenFile = func(path string) (r io.ReadCloser, err error) { |
| if path != "x+"+tt.name { |
| t.Fatalf("OpenFile asked for %q, expected %q", path, "x+"+tt.name) |
| } |
| return &readNopCloser{strings.NewReader(tt.data)}, nil |
| } |
| ctxt.JoinPath = func(elem ...string) string { |
| return strings.Join(elem, "+") |
| } |
| match, err := ctxt.MatchFile("x", tt.name) |
| if match != tt.match || err != nil { |
| t.Fatalf("MatchFile(%q) = %v, %v, want %v, nil", tt.name, match, err, tt.match) |
| } |
| } |
| } |