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 := "> "
|
2020-04-16 07:11:23 +08:00
|
|
|
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)
|
|
|
|
}
|
2020-09-03 13:48:39 +08:00
|
|
|
|
|
|
|
func mustPipe() (*os.File, *os.File) {
|
|
|
|
r, w, err := os.Pipe()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
return r, w
|
|
|
|
}
|