diff options
Diffstat (limited to 'pkg/test/hook_test.go')
| -rw-r--r-- | pkg/test/hook_test.go | 33 |
1 files changed, 14 insertions, 19 deletions
diff --git a/pkg/test/hook_test.go b/pkg/test/hook_test.go index 06204bb..31e74f7 100644 --- a/pkg/test/hook_test.go +++ b/pkg/test/hook_test.go @@ -8,52 +8,47 @@ import ( ) func TestUpperHook(t *testing.T) { - const msg = "Hello, World!" - c, s := newClientServer(t) defer closeClientServer(c, s) - tunnel := "add name %s listen,addr=127.0.0.1:0 upper loop" - c.Exec(tunnel, t.Name()) - - addr := c.Send("get tunnel.%s.listen", t.Name()) + c.Exec("add name T listen,addr=-:0 upper loop") - conn := xDial(t, "tcp", addr) + conn := xDial(t, "tcp", c.Get("tunnel.T.listen")) defer conn.Close() - xWrite(t, conn, msg) + xWrite(t, conn, xData) - buf := make([]byte, len(msg)) + buf := make([]byte, len(xData)) xReadFull(t, conn, buf) - if r := string(buf); r != strings.ToUpper(msg) { - t.Fatalf("wrong reply: send '%s', recv '%s'", msg, r) + if r := string(buf); r != strings.ToUpper(xData) { + t.Fatalf("wrong reply: send '%s', recv '%s'", xData, r) } } func TestHexHook(t *testing.T) { - const msg = "Hello, World!" - c, s := newClientServer(t) defer closeClientServer(c, s) + c.Exec("add name T listen,addr=-:0 hex dial,addr=@[addr]") + listen := xListen(t, "tcp", "127.0.0.1:0") defer listen.Close() - addr := c.AddListenTunnel(t.Name(), "hex dial,addr=%s", listen.Addr()) + c.Set("addr", listen.Addr()) - out := xDial(t, "tcp", addr) + out := xDial(t, "tcp", c.Get("tunnel.T.listen")) defer out.Close() in := xAccept(t, listen) defer in.Close() - xWrite(t, out, msg) + xWrite(t, out, xData) - buf := make([]byte, 2*len(msg)) + buf := make([]byte, 2*len(xData)) xReadFull(t, in, buf) - if r := string(buf); r != hex.EncodeToString([]byte(msg)) { - t.Fatalf("wrong reply: send '%s', recv '%s'", msg, r) + if r := string(buf); r != hex.EncodeToString([]byte(xData)) { + t.Fatalf("wrong reply: send '%s', recv '%s'", xData, r) } } |