github.com/jhump/golang-x-tools@v0.0.0-20220218190644-4958d6d39439/cmd/getgo/main_test.go (about)

     1  // Copyright 2017 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  //go:build !plan9
     6  // +build !plan9
     7  
     8  package main
     9  
    10  import (
    11  	"bytes"
    12  	"fmt"
    13  	"io/ioutil"
    14  	"os"
    15  	"os/exec"
    16  	"runtime"
    17  	"testing"
    18  )
    19  
    20  const (
    21  	testbin = "testgetgo"
    22  )
    23  
    24  var (
    25  	exeSuffix string // ".exe" on Windows
    26  )
    27  
    28  func init() {
    29  	if runtime.GOOS == "windows" {
    30  		exeSuffix = ".exe"
    31  	}
    32  }
    33  
    34  // TestMain creates a getgo command for testing purposes and
    35  // deletes it after the tests have been run.
    36  func TestMain(m *testing.M) {
    37  	if os.Getenv("GOGET_INTEGRATION") == "" {
    38  		fmt.Fprintln(os.Stderr, "main_test: Skipping integration tests with GOGET_INTEGRATION unset")
    39  		return
    40  	}
    41  
    42  	args := []string{"build", "-tags", testbin, "-o", testbin + exeSuffix}
    43  	out, err := exec.Command("go", args...).CombinedOutput()
    44  	if err != nil {
    45  		fmt.Fprintf(os.Stderr, "building %s failed: %v\n%s", testbin, err, out)
    46  		os.Exit(2)
    47  	}
    48  
    49  	// Don't let these environment variables confuse the test.
    50  	os.Unsetenv("GOBIN")
    51  	os.Unsetenv("GOPATH")
    52  	os.Unsetenv("GIT_ALLOW_PROTOCOL")
    53  	os.Unsetenv("PATH")
    54  
    55  	r := m.Run()
    56  
    57  	os.Remove(testbin + exeSuffix)
    58  
    59  	os.Exit(r)
    60  }
    61  
    62  func createTmpHome(t *testing.T) string {
    63  	tmpd, err := ioutil.TempDir("", "testgetgo")
    64  	if err != nil {
    65  		t.Fatalf("creating test tempdir failed: %v", err)
    66  	}
    67  
    68  	os.Setenv("HOME", tmpd)
    69  	return tmpd
    70  }
    71  
    72  // doRun runs the test getgo command, recording stdout and stderr and
    73  // returning exit status.
    74  func doRun(t *testing.T, args ...string) error {
    75  	var stdout, stderr bytes.Buffer
    76  	t.Logf("running %s %v", testbin, args)
    77  	cmd := exec.Command("./"+testbin+exeSuffix, args...)
    78  	cmd.Stdout = &stdout
    79  	cmd.Stderr = &stderr
    80  	cmd.Env = os.Environ()
    81  	status := cmd.Run()
    82  	if stdout.Len() > 0 {
    83  		t.Log("standard output:")
    84  		t.Log(stdout.String())
    85  	}
    86  	if stderr.Len() > 0 {
    87  		t.Log("standard error:")
    88  		t.Log(stderr.String())
    89  	}
    90  	return status
    91  }
    92  
    93  func TestCommandVerbose(t *testing.T) {
    94  	tmpd := createTmpHome(t)
    95  	defer os.RemoveAll(tmpd)
    96  
    97  	err := doRun(t, "-v")
    98  	if err != nil {
    99  		t.Fatal(err)
   100  	}
   101  	// make sure things are in path
   102  	shellConfig, err := shellConfigFile()
   103  	if err != nil {
   104  		t.Fatal(err)
   105  	}
   106  	b, err := ioutil.ReadFile(shellConfig)
   107  	if err != nil {
   108  		t.Fatal(err)
   109  	}
   110  	home, err := getHomeDir()
   111  	if err != nil {
   112  		t.Fatal(err)
   113  	}
   114  
   115  	expected := fmt.Sprintf(`
   116  export PATH=$PATH:%s/.go/bin
   117  
   118  export GOPATH=%s/go
   119  
   120  export PATH=$PATH:%s/go/bin
   121  `, home, home, home)
   122  
   123  	if string(b) != expected {
   124  		t.Fatalf("%s expected %q, got %q", shellConfig, expected, string(b))
   125  	}
   126  }
   127  
   128  func TestCommandPathExists(t *testing.T) {
   129  	tmpd := createTmpHome(t)
   130  	defer os.RemoveAll(tmpd)
   131  
   132  	// run once
   133  	err := doRun(t, "-skip-dl")
   134  	if err != nil {
   135  		t.Fatal(err)
   136  	}
   137  	// make sure things are in path
   138  	shellConfig, err := shellConfigFile()
   139  	if err != nil {
   140  		t.Fatal(err)
   141  	}
   142  	b, err := ioutil.ReadFile(shellConfig)
   143  	if err != nil {
   144  		t.Fatal(err)
   145  	}
   146  	home, err := getHomeDir()
   147  	if err != nil {
   148  		t.Fatal(err)
   149  	}
   150  
   151  	expected := fmt.Sprintf(`
   152  export GOPATH=%s/go
   153  
   154  export PATH=$PATH:%s/go/bin
   155  `, home, home)
   156  
   157  	if string(b) != expected {
   158  		t.Fatalf("%s expected %q, got %q", shellConfig, expected, string(b))
   159  	}
   160  
   161  	// run twice
   162  	if err := doRun(t, "-skip-dl"); err != nil {
   163  		t.Fatal(err)
   164  	}
   165  
   166  	b, err = ioutil.ReadFile(shellConfig)
   167  	if err != nil {
   168  		t.Fatal(err)
   169  	}
   170  
   171  	if string(b) != expected {
   172  		t.Fatalf("%s expected %q, got %q", shellConfig, expected, string(b))
   173  	}
   174  }