281 lines
		
	
	
		
			7.5 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			281 lines
		
	
	
		
			7.5 KiB
		
	
	
	
		
			Go
		
	
	
	
| // Copyright 2013 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 vcs
 | |
| 
 | |
| import (
 | |
| 	"errors"
 | |
| 	"io/ioutil"
 | |
| 	"os"
 | |
| 	"path"
 | |
| 	"path/filepath"
 | |
| 	"reflect"
 | |
| 	"runtime"
 | |
| 	"strings"
 | |
| 	"testing"
 | |
| )
 | |
| 
 | |
| // Test that RepoRootForImportPath creates the correct RepoRoot for a given importPath.
 | |
| // TODO(cmang): Add tests for SVN and BZR.
 | |
| func TestRepoRootForImportPath(t *testing.T) {
 | |
| 	if runtime.GOOS == "android" {
 | |
| 		t.Skipf("incomplete source tree on %s", runtime.GOOS)
 | |
| 	}
 | |
| 
 | |
| 	tests := []struct {
 | |
| 		path string
 | |
| 		want *RepoRoot
 | |
| 	}{
 | |
| 		{
 | |
| 			"github.com/golang/groupcache",
 | |
| 			&RepoRoot{
 | |
| 				VCS:  vcsGit,
 | |
| 				Repo: "https://github.com/golang/groupcache",
 | |
| 			},
 | |
| 		},
 | |
| 		// Unicode letters in directories (issue 18660).
 | |
| 		{
 | |
| 			"github.com/user/unicode/испытание",
 | |
| 			&RepoRoot{
 | |
| 				VCS:  vcsGit,
 | |
| 				Repo: "https://github.com/user/unicode",
 | |
| 			},
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	for _, test := range tests {
 | |
| 		got, err := RepoRootForImportPath(test.path, false)
 | |
| 		if err != nil {
 | |
| 			t.Errorf("RepoRootForImportPath(%q): %v", test.path, err)
 | |
| 			continue
 | |
| 		}
 | |
| 		want := test.want
 | |
| 		if got.VCS.Name != want.VCS.Name || got.Repo != want.Repo {
 | |
| 			t.Errorf("RepoRootForImportPath(%q) = VCS(%s) Repo(%s), want VCS(%s) Repo(%s)", test.path, got.VCS, got.Repo, want.VCS, want.Repo)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Test that FromDir correctly inspects a given directory and returns the right VCS and root.
 | |
| func TestFromDir(t *testing.T) {
 | |
| 	tempDir, err := ioutil.TempDir("", "vcstest")
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	defer os.RemoveAll(tempDir)
 | |
| 
 | |
| 	for j, vcs := range vcsList {
 | |
| 		dir := filepath.Join(tempDir, "example.com", vcs.Name, "."+vcs.Cmd)
 | |
| 		if j&1 == 0 {
 | |
| 			err := os.MkdirAll(dir, 0755)
 | |
| 			if err != nil {
 | |
| 				t.Fatal(err)
 | |
| 			}
 | |
| 		} else {
 | |
| 			err := os.MkdirAll(filepath.Dir(dir), 0755)
 | |
| 			if err != nil {
 | |
| 				t.Fatal(err)
 | |
| 			}
 | |
| 			f, err := os.Create(dir)
 | |
| 			if err != nil {
 | |
| 				t.Fatal(err)
 | |
| 			}
 | |
| 			f.Close()
 | |
| 		}
 | |
| 
 | |
| 		want := RepoRoot{
 | |
| 			VCS:  vcs,
 | |
| 			Root: path.Join("example.com", vcs.Name),
 | |
| 		}
 | |
| 		var got RepoRoot
 | |
| 		got.VCS, got.Root, err = FromDir(dir, tempDir)
 | |
| 		if err != nil {
 | |
| 			t.Errorf("FromDir(%q, %q): %v", dir, tempDir, err)
 | |
| 			continue
 | |
| 		}
 | |
| 		if got.VCS.Name != want.VCS.Name || got.Root != want.Root {
 | |
| 			t.Errorf("FromDir(%q, %q) = VCS(%s) Root(%s), want VCS(%s) Root(%s)", dir, tempDir, got.VCS, got.Root, want.VCS, want.Root)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| var parseMetaGoImportsTests = []struct {
 | |
| 	in  string
 | |
| 	out []metaImport
 | |
| }{
 | |
| 	{
 | |
| 		`<meta name="go-import" content="foo/bar git https://github.com/rsc/foo/bar">`,
 | |
| 		[]metaImport{{"foo/bar", "git", "https://github.com/rsc/foo/bar"}},
 | |
| 	},
 | |
| 	{
 | |
| 		`<meta name="go-import" content="foo/bar git https://github.com/rsc/foo/bar">
 | |
| 		<meta name="go-import" content="baz/quux git http://github.com/rsc/baz/quux">`,
 | |
| 		[]metaImport{
 | |
| 			{"foo/bar", "git", "https://github.com/rsc/foo/bar"},
 | |
| 			{"baz/quux", "git", "http://github.com/rsc/baz/quux"},
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		`<head>
 | |
| 		<meta name="go-import" content="foo/bar git https://github.com/rsc/foo/bar">
 | |
| 		</head>`,
 | |
| 		[]metaImport{{"foo/bar", "git", "https://github.com/rsc/foo/bar"}},
 | |
| 	},
 | |
| 	{
 | |
| 		`<meta name="go-import" content="foo/bar git https://github.com/rsc/foo/bar">
 | |
| 		<body>`,
 | |
| 		[]metaImport{{"foo/bar", "git", "https://github.com/rsc/foo/bar"}},
 | |
| 	},
 | |
| }
 | |
| 
 | |
| func TestParseMetaGoImports(t *testing.T) {
 | |
| 	for i, tt := range parseMetaGoImportsTests {
 | |
| 		out, err := parseMetaGoImports(strings.NewReader(tt.in))
 | |
| 		if err != nil {
 | |
| 			t.Errorf("test#%d: %v", i, err)
 | |
| 			continue
 | |
| 		}
 | |
| 		if !reflect.DeepEqual(out, tt.out) {
 | |
| 			t.Errorf("test#%d:\n\thave %q\n\twant %q", i, out, tt.out)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestValidateRepoRoot(t *testing.T) {
 | |
| 	tests := []struct {
 | |
| 		root string
 | |
| 		ok   bool
 | |
| 	}{
 | |
| 		{
 | |
| 			root: "",
 | |
| 			ok:   false,
 | |
| 		},
 | |
| 		{
 | |
| 			root: "http://",
 | |
| 			ok:   true,
 | |
| 		},
 | |
| 		{
 | |
| 			root: "git+ssh://",
 | |
| 			ok:   true,
 | |
| 		},
 | |
| 		{
 | |
| 			root: "http#://",
 | |
| 			ok:   false,
 | |
| 		},
 | |
| 		{
 | |
| 			root: "-config",
 | |
| 			ok:   false,
 | |
| 		},
 | |
| 		{
 | |
| 			root: "-config://",
 | |
| 			ok:   false,
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	for _, test := range tests {
 | |
| 		err := validateRepoRoot(test.root)
 | |
| 		ok := err == nil
 | |
| 		if ok != test.ok {
 | |
| 			want := "error"
 | |
| 			if test.ok {
 | |
| 				want = "nil"
 | |
| 			}
 | |
| 			t.Errorf("validateRepoRoot(%q) = %q, want %s", test.root, err, want)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestMatchGoImport(t *testing.T) {
 | |
| 	tests := []struct {
 | |
| 		imports []metaImport
 | |
| 		path    string
 | |
| 		mi      metaImport
 | |
| 		err     error
 | |
| 	}{
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo",
 | |
| 			mi:   metaImport{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo/",
 | |
| 			mi:   metaImport{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/fooa", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo",
 | |
| 			mi:   metaImport{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/fooa", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/fooa",
 | |
| 			mi:   metaImport{Prefix: "example.com/user/fooa", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/foo/bar", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo/bar",
 | |
| 			err:  errors.New("should not be allowed to create nested repo"),
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/foo/bar", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo/bar/baz",
 | |
| 			err:  errors.New("should not be allowed to create nested repo"),
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/foo/bar", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo/bar/baz/qux",
 | |
| 			err:  errors.New("should not be allowed to create nested repo"),
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/foo/bar", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com/user/foo/bar/baz/",
 | |
| 			err:  errors.New("should not be allowed to create nested repo"),
 | |
| 		},
 | |
| 		{
 | |
| 			imports: []metaImport{
 | |
| 				{Prefix: "example.com/user/foo", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 				{Prefix: "example.com/user/foo/bar", VCS: "git", RepoRoot: "https://example.com/repo/target"},
 | |
| 			},
 | |
| 			path: "example.com",
 | |
| 			err:  errors.New("pathologically short path"),
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	for _, test := range tests {
 | |
| 		mi, err := matchGoImport(test.imports, test.path)
 | |
| 		if mi != test.mi {
 | |
| 			t.Errorf("unexpected metaImport; got %v, want %v", mi, test.mi)
 | |
| 		}
 | |
| 
 | |
| 		got := err
 | |
| 		want := test.err
 | |
| 		if (got == nil) != (want == nil) {
 | |
| 			t.Errorf("unexpected error; got %v, want %v", got, want)
 | |
| 		}
 | |
| 	}
 | |
| }
 |