From 6b76a30c04cb74dd9a759789f75bba2eea86a1fb Mon Sep 17 00:00:00 2001 From: Mikhail Osipov Date: Wed, 16 Dec 2020 18:03:08 +0300 Subject: add defer test --- pkg/test/auth_test.go | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) (limited to 'pkg/test/auth_test.go') diff --git a/pkg/test/auth_test.go b/pkg/test/auth_test.go index cf5f28b..41d1c2d 100644 --- a/pkg/test/auth_test.go +++ b/pkg/test/auth_test.go @@ -5,32 +5,32 @@ import ( ) func TestAuthHook(t *testing.T) { - const msg = "Hello, World!" - c, s := newClientServer(t) defer closeClientServer(c, s) - listen := xListen(t, "tcp", "127.0.0.1:0") - defer listen.Close() - - xaddr := c.AddListenTunnel("X", "-aes -auth dial,addr=%s", listen.Addr()) - taddr := c.AddListenTunnel("T", "auth aes dial,addr=%s", xaddr) + c.Exec("add name T listen,addr=-:0 auth aes dial,addr=@[tunnel.X.listen]") + c.Exec("add name X listen,addr=-:0 -aes -auth dial,addr=@[addr]") c.Exec("set tunnel.X.secret secret") c.Exec("set tunnel.T.secret secret") - out := xDial(t, "tcp", taddr) + listen := xListen(t, "tcp", "127.0.0.1:0") + defer listen.Close() + + c.Set("addr", listen.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, len(msg)) + buf := make([]byte, len(xData)) xReadFull(t, in, buf) - if r := string(buf); r != msg { - t.Fatalf("wrong reply: send '%s', recv '%s'", msg, r) + if r := string(buf); r != xData { + t.Fatalf("wrong reply: send '%s', recv '%s'", xData, r) } } -- cgit v1.2.3-70-g09d2