func ParseNewAddr(addr string) (siphon.Addr, error) {
addrParts := strings.SplitN(addr, "://", 2)
switch addrParts[0] {
case "unix":
return siphon.NewAddr(addr, "unix", addrParts[1]), nil
// case "tcp":
// // lib siphon supports this, but it's so very likely to be a bad idea that i'm making you compile a program for it yourself if you want this.
// return siphon.NewAddr(addr, "tcp", addrParts[1]), nil
default:
return siphon.Addr{}, fmt.Errorf("invalid protocol format. \"%s\"", addr)
}
}
func TestDoubleClient(t *testing.T) {
assert := assrt.NewAssert(t)
addr := siphon.NewAddr("test", "unix", "test.sock")
cmd := exec.Command("cat", "-")
host := siphon.NewHost(cmd, addr)
host.Serve()
defer host.UnServe()
client1 := siphon.NewClient(addr)
client1.Connect()
client2 := siphon.NewClient(addr)
client2.Connect()
host.Start()
go func() {
// timed pauses are a hack required to make sure the host side flushes the echos in an order we can test.
client1.Stdin().Write([]byte("foo\n"))
time.Sleep(50 * time.Millisecond)
client2.Stdin().Write([]byte("bar\n"))
time.Sleep(50 * time.Millisecond)
client1.Stdin().Write([]byte("baz\n"))
time.Sleep(50 * time.Millisecond)
client2.Stdin().Write([]byte{4}) // EOT
}()
outBuffer1 := new(bytes.Buffer)
io.Copy(outBuffer1, client1.Stdout())
out1 := string(outBuffer1.Bytes())
outBuffer2 := new(bytes.Buffer)
io.Copy(outBuffer2, client2.Stdout())
out2 := string(outBuffer2.Bytes())
expected :=
"foo\r\nfoo\r\n" +
"bar\r\nbar\r\n" +
"baz\r\nbaz\r\n"
assert.Equal(
expected,
out1,
)
assert.Equal(
expected,
out2,
)
}
func TestUnixSocket(t *testing.T) {
assert := assrt.NewAssert(t)
addr := siphon.NewAddr("test", "unix", "test.sock")
cmd := exec.Command("cat", "-")
host := siphon.NewHost(cmd, addr)
host.Serve()
defer host.UnServe()
client := siphon.NewClient(addr)
client.Connect()
//client.Attach() // you can't do this in a test. there's no tty.
//FIXME: there's really no guarantee that the host finished processing the client connect request by now
host.Start()
go func() {
stdin := client.Stdin()
stdin.Write([]byte("foo\nbar\nbaz\n"))
stdin.Write([]byte{4}) // EOT
}()
outBuffer := new(bytes.Buffer)
io.Copy(outBuffer, client.Stdout())
out := string(outBuffer.Bytes())
expected :=
"foo\r\n" +
"bar\r\n" +
"baz\r\n"
expected = expected + expected
assert.Equal(
expected,
out,
)
}