Add benchmark

Author: ascandella

internal/gps/pkgtree/pkgtree_benchmark_test.go

@@ -0,0 +1,28 @@
+// Copyright 2017 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 pkgtree
+
+import "testing"
+
+func BenchmarkListPackages(b *testing.B) {
+	b.StopTimer()
+
+	j := func(s ...string) string {
+		return testDir(b, s...)
+	}
+
+	table := []string{
+		"dotgodir",
+		"buildtag",
+		"varied",
+	}
+
+	b.StartTimer()
+
+	for _, name := range table {
+		for n := 0; n < b.N; n++ {
+			ListPackages(j(name), name)
+		}
+	}
+}

internal/gps/pkgtree/pkgtree_test.go

@@ -472,10 +472,14 @@ func TestListPackagesNoDir(t *testing.T) {
 	}
 }
 
-func TestListPackages(t *testing.T) {
+func testDir(t testing.TB, s ...string) string {
 	srcdir := filepath.Join(getTestdataRootDir(t), "src")
+	return filepath.Join(srcdir, filepath.Join(s...))
+}
+
+func TestListPackages(t *testing.T) {
 	j := func(s ...string) string {
-		return filepath.Join(srcdir, filepath.Join(s...))
+		return testDir(t, s...)
 	}
 
 	table := map[string]struct {
@@ -1976,7 +1980,7 @@ func TestToReachMapFilterDot(t *testing.T) {
 	}
 }
 
-func getTestdataRootDir(t *testing.T) string {
+func getTestdataRootDir(t testing.TB) string {
 	cwd, err := os.Getwd()
 	if err != nil {
 		t.Fatal(err)