Function CaptureUserOut

pkg/util/capture.go:16–38

CaptureUserOut captures output written to UserOut to a string. Capturing starts when it is called. It returns an anonymous function that when called, will return a string containing the output during capture, and revert once again to the original value of UserOut.Out.

()

// when called, will return a string containing the output during capture, and
// revert once again to the original value of UserOut.Out.
func CaptureUserOut() func() string {
	old := output.UserOut.Out // keep backup of the real stdout
	r, w, _ := os.Pipe()
	output.UserOut.Out = w

	return func() string {
		outC := make(chan string)
		// copy the output in a separate goroutine so printing can't block indefinitely
		go func() {
			var buf bytes.Buffer
			_, err := io.Copy(&buf, r)
			CheckErr(err)
			outC <- buf.String()
		}()

		// back to normal state
		CheckClose(w)
		output.UserOut.Out = old // restoring the real stdout

		out := <-outC
		return out
	}
}

// CaptureUserErr captures output written to UserErr to a string.
// Capturing starts when it is called. It returns an anonymous function that