From fad8780c0cdc891902993f172f099f56f79e6d1a Mon Sep 17 00:00:00 2001
From: Steven Allen <steven@stebalien.com>
Date: Thu, 8 Mar 2018 09:00:00 -0800
Subject: [PATCH] add test for netlistener wrapper/unwrapper

---
 net_test.go | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/net_test.go b/net_test.go
index e7299b2..a073e3b 100644
--- a/net_test.go
+++ b/net_test.go
@@ -514,3 +514,57 @@ func TestInterfaceAddressesWorks(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestNetListener(t *testing.T) {
+	listener, err := net.Listen("tcp", "127.0.0.1:1234")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer listener.Close()
+	malist, err := WrapNetListener(listener)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !malist.Multiaddr().Equal(newMultiaddr(t, "/ip4/127.0.0.1/tcp/1234")) {
+		t.Fatal("unexpected multiaddr")
+	}
+
+	go func() {
+		c, err := Dial(malist.Multiaddr())
+		if err != nil {
+			t.Fatal("failed to dial")
+		}
+		if !c.RemoteMultiaddr().Equal(malist.Multiaddr()) {
+			t.Fatal("dialed wrong target")
+		}
+		c.Close()
+
+		c, err = Dial(malist.Multiaddr())
+		if err != nil {
+			t.Fatal("failed to dial")
+		}
+		c.Close()
+	}()
+
+	c, err := malist.Accept()
+	if err != nil {
+		t.Fatal(err)
+	}
+	c.Close()
+	netList := NetListener(malist)
+	malist2, err := WrapNetListener(netList)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if malist2 != malist {
+		t.Fatal("expected WrapNetListener(NetListener(malist)) == malist")
+	}
+	nc, err := netList.Accept()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !nc.(Conn).LocalMultiaddr().Equal(malist.Multiaddr()) {
+		t.Fatal("wrong multiaddr on conn")
+	}
+	nc.Close()
+}