add stderr to tests
This commit is contained in:
parent
c76dca0456
commit
33d0dd0bfc
|
@ -1,2 +1,3 @@
|
||||||
|
*.swp
|
||||||
vendor
|
vendor
|
||||||
.glide
|
.glide
|
||||||
|
|
|
@ -57,28 +57,39 @@ func RunWithArgs(cmd Executable, args []string, env map[string]string) error {
|
||||||
// RunCaptureWithArgs executes the given command with the specified command line args
|
// RunCaptureWithArgs executes the given command with the specified command line args
|
||||||
// and environmental variables set. It returns whatever was writen to
|
// and environmental variables set. It returns whatever was writen to
|
||||||
// stdout along with any error returned from cmd.Execute()
|
// stdout along with any error returned from cmd.Execute()
|
||||||
func RunCaptureWithArgs(cmd Executable, args []string, env map[string]string) (output string, err error) {
|
func RunCaptureWithArgs(cmd Executable, args []string, env map[string]string) (stdout, stderr string, err error) {
|
||||||
old := os.Stdout // keep backup of the real stdout
|
oldout, olderr := os.Stdout, os.Stderr // keep backup of the real stdout
|
||||||
r, w, _ := os.Pipe()
|
rOut, wOut, _ := os.Pipe()
|
||||||
os.Stdout = w
|
rErr, wErr, _ := os.Pipe()
|
||||||
|
os.Stdout, os.Stderr = wOut, wErr
|
||||||
defer func() {
|
defer func() {
|
||||||
os.Stdout = old // restoring the real stdout
|
os.Stdout, os.Stderr = oldout, olderr // restoring the real stdout
|
||||||
}()
|
}()
|
||||||
|
|
||||||
outC := make(chan string)
|
|
||||||
// copy the output in a separate goroutine so printing can't block indefinitely
|
// copy the output in a separate goroutine so printing can't block indefinitely
|
||||||
|
outC := make(chan string)
|
||||||
go func() {
|
go func() {
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
// io.Copy will end when we call w.Close() below
|
// io.Copy will end when we call wOut.Close() below
|
||||||
io.Copy(&buf, r)
|
io.Copy(&buf, rOut)
|
||||||
outC <- buf.String()
|
outC <- buf.String()
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
errC := make(chan string)
|
||||||
|
go func() {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
// io.Copy will end when we call wErr.Close() below
|
||||||
|
io.Copy(&buf, rErr)
|
||||||
|
errC <- buf.String()
|
||||||
|
}()
|
||||||
|
|
||||||
// now run the command
|
// now run the command
|
||||||
err = RunWithArgs(cmd, args, env)
|
err = RunWithArgs(cmd, args, env)
|
||||||
|
|
||||||
// and grab the stdout to return
|
// and grab the stdout to return
|
||||||
w.Close()
|
wOut.Close()
|
||||||
output = <-outC
|
wErr.Close()
|
||||||
return output, err
|
stdout = <-outC
|
||||||
|
stderr = <-errC
|
||||||
|
return stdout, stderr, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -212,9 +212,11 @@ func TestSetupTrace(t *testing.T) {
|
||||||
|
|
||||||
viper.Reset()
|
viper.Reset()
|
||||||
args := append([]string{cmd.Use}, tc.args...)
|
args := append([]string{cmd.Use}, tc.args...)
|
||||||
out, err := RunCaptureWithArgs(cmd, args, tc.env)
|
stdout, stderr, err := RunCaptureWithArgs(cmd, args, tc.env)
|
||||||
require.NotNil(err, i)
|
require.NotNil(err, i)
|
||||||
msg := strings.Split(out, "\n")
|
require.Equal("", stdout, i)
|
||||||
|
require.NotEqual("", stderr, i)
|
||||||
|
msg := strings.Split(stderr, "\n")
|
||||||
desired := fmt.Sprintf("ERROR: %s", tc.expected)
|
desired := fmt.Sprintf("ERROR: %s", tc.expected)
|
||||||
assert.Equal(desired, msg[0], i)
|
assert.Equal(desired, msg[0], i)
|
||||||
if tc.long && assert.True(len(msg) > 2, i) {
|
if tc.long && assert.True(len(msg) > 2, i) {
|
||||||
|
|
Loading…
Reference in New Issue