diff options
Diffstat (limited to 'pkg/test/proxy_test.go')
| -rw-r--r-- | pkg/test/proxy_test.go | 31 |
1 files changed, 14 insertions, 17 deletions
diff --git a/pkg/test/proxy_test.go b/pkg/test/proxy_test.go index b2fb097..ae89b0a 100644 --- a/pkg/test/proxy_test.go +++ b/pkg/test/proxy_test.go @@ -5,32 +5,29 @@ import ( ) func TestProxyHook(t *testing.T) { - const msg = "Hello, World!" + e := newEnv(t) + defer e.Free() - c, s := newClientServer(t) - defer closeClientServer(c, s) + c := e.newInstance() - listen := xListen(t, "tcp", "127.0.0.1:0") - defer listen.Close() - - saddr := c.AddListenTunnel("S", "proxy") - caddr := c.AddListenTunnel("C", "proxy,addr=%s dial,addr=%s", listen.Addr(), saddr) + c.Exec("add name C listen,addr=-:0 proxy,addr=@[addr] dial,addr=@[tunnel.S.listen]") + c.Exec("add name S listen,addr=-:0 proxy") c.Exec("set tunnel.S.proxy.auth user:password") c.Exec("set tunnel.C.proxy.auth user:password") - out := xDial(t, "tcp", caddr) - defer out.Close() + listen := e.Listen("tcp", "127.0.0.1:0") + c.Set("addr", listen.Addr()) - in := xAccept(t, listen) - defer in.Close() + out := e.Dial("tcp", c.Get("tunnel.C.listen")) + in := e.Accept(listen) - xWrite(t, out, msg) + e.Write(out, dummy) - buf := make([]byte, len(msg)) - xReadFull(t, in, buf) + buf := make([]byte, len(dummy)) + e.ReadFull(in, buf) - if r := string(buf); r != msg { - t.Fatalf("wrong reply: send '%s', recv '%s'", msg, r) + if r := string(buf); r != dummy { + t.Fatalf("wrong reply: send '%s', recv '%s'", dummy, r) } } |