1// Copyright 2019 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 5package stdio_test 6 7import ( 8 "bytes" 9 "cmd/cgo/internal/cgotest" 10 "internal/testenv" 11 "log" 12 "os" 13 "os/exec" 14 "path/filepath" 15 "strings" 16 "testing" 17) 18 19func TestMain(m *testing.M) { 20 log.SetFlags(log.Lshortfile) 21 os.Exit(testMain(m)) 22} 23 24func testMain(m *testing.M) int { 25 GOPATH, err := os.MkdirTemp("", "cgostdio") 26 if err != nil { 27 log.Panic(err) 28 } 29 defer os.RemoveAll(GOPATH) 30 os.Setenv("GOPATH", GOPATH) 31 32 // Copy testdata into GOPATH/src/cgostdio, along with a go.mod file 33 // declaring the same path. 34 modRoot := filepath.Join(GOPATH, "src", "cgostdio") 35 if err := cgotest.OverlayDir(modRoot, "testdata"); err != nil { 36 log.Panic(err) 37 } 38 if err := os.Chdir(modRoot); err != nil { 39 log.Panic(err) 40 } 41 os.Setenv("PWD", modRoot) 42 if err := os.WriteFile("go.mod", []byte("module cgostdio\n"), 0666); err != nil { 43 log.Panic(err) 44 } 45 46 return m.Run() 47} 48 49// TestTestRun runs a cgo test that doesn't depend on non-standard libraries. 50func TestTestRun(t *testing.T) { 51 testenv.MustHaveGoRun(t) 52 testenv.MustHaveCGO(t) 53 54 for _, file := range [...]string{ 55 "chain.go", 56 "fib.go", 57 "hello.go", 58 } { 59 file := file 60 wantFile := strings.Replace(file, ".go", ".out", 1) 61 t.Run(file, func(t *testing.T) { 62 cmd := exec.Command("go", "run", file) 63 got, err := cmd.CombinedOutput() 64 if err != nil { 65 t.Fatalf("%v: %s\n%s", cmd, err, got) 66 } 67 got = bytes.ReplaceAll(got, []byte("\r\n"), []byte("\n")) 68 want, err := os.ReadFile(wantFile) 69 if err != nil { 70 t.Fatal("reading golden output:", err) 71 } 72 if !bytes.Equal(got, want) { 73 t.Errorf("'%v' output does not match expected in %s. Instead saw:\n%s", cmd, wantFile, got) 74 } 75 }) 76 } 77} 78