photoprism/pkg/capture/output.go

36 lines
437 B
Go
Raw Normal View History

package capture
import (
"bytes"
"io"
"os"
)
// Output returns output to stdout and stderr for testing.
func Output(f func()) string {
r, w, err := os.Pipe()
if err != nil {
panic(err)
}
stdout := os.Stdout
os.Stdout = w
defer func() {
os.Stdout = stdout
}()
stderr := os.Stderr
os.Stderr = w
defer func() {
os.Stderr = stderr
}()
f()
w.Close()
var buf bytes.Buffer
io.Copy(&buf, r)
return buf.String()
}