elvish/pkg/eval/port_helper_test.go

54 lines
1.0 KiB
Go
Raw Normal View History

2017-12-28 05:04:07 +08:00
package eval
import (
"io"
"io/ioutil"
"os"
"testing"
)
func TestEvalerPorts(t *testing.T) {
stdoutReader, stdout := mustPipe()
defer stdoutReader.Close()
stderrReader, stderr := mustPipe()
defer stderrReader.Close()
prefix := "> "
ports, cleanup := portsFromFiles([3]*os.File{DevNull, stdout, stderr}, prefix)
ports[1].Chan <- "x"
ports[1].Chan <- "y"
ports[2].Chan <- "bad"
ports[2].Chan <- "err"
cleanup()
2017-12-28 05:04:07 +08:00
stdout.Close()
stderr.Close()
stdoutAll := mustReadAllString(stdoutReader)
wantStdoutAll := "> x\n> y\n"
if stdoutAll != wantStdoutAll {
t.Errorf("stdout is %q, want %q", stdoutAll, wantStdoutAll)
}
stderrAll := mustReadAllString(stderrReader)
wantStderrAll := "> bad\n> err\n"
if stderrAll != wantStderrAll {
t.Errorf("stderr is %q, want %q", stderrAll, wantStderrAll)
}
}
func mustReadAllString(r io.Reader) string {
b, err := ioutil.ReadAll(r)
if err != nil {
panic(err)
}
return string(b)
}
func mustPipe() (*os.File, *os.File) {
r, w, err := os.Pipe()
if err != nil {
panic(err)
}
return r, w
}