add defer test

1 files changed, 32 insertions, 0 deletions

diff --git a/pkg/test/defer_test.go b/pkg/test/defer_test.go
new file mode 100644
index 0000000..215e0b1
--- /dev/null
+++ b/pkg/test/defer_test.go
@@ -0,0 +1,32 @@
+package test
+
+import (
+	"testing"
+)
+
+func TestDeferSocket(t *testing.T) {
+	c, s := newClientServer(t)
+	defer closeClientServer(c, s)
+
+	c.Exec("add name T listen,addr=-:0 defer,addr=@[addr]")
+
+	out := xDial(t, "tcp", c.Get("tunnel.T.listen"))
+	defer out.Close()
+
+	listen := xListen(t, "tcp", "127.0.0.1:0")
+	defer listen.Close()
+
+	c.Set("addr", listen.Addr())
+
+	xWrite(t, out, xData)
+
+	in := xAccept(t, listen)
+	defer in.Close()
+
+	buf := make([]byte, len(xData))
+	xReadFull(t, in, buf)
+
+	if r := string(buf); r != xData {
+		t.Fatalf("wrong reply: send '%s', recv '%s'", xData, r)
+	}
+}