returns real remote and local address instead mocked

Author: photostorm

conn_test.go

@@ -163,8 +163,8 @@ func TestConn(t *testing.T) {
 		n1.SetDeadline(time.Time{})
 
 		assert.Equal(t, "remote addr", n1.RemoteAddr(), n1.LocalAddr())
-		assert.Equal(t, "remote addr string", "websocket/unknown-addr", n1.RemoteAddr().String())
-		assert.Equal(t, "remote addr network", "websocket", n1.RemoteAddr().Network())
+		assert.Equal(t, "remote addr string", "pipe", n1.RemoteAddr().String())
+		assert.Equal(t, "remote addr network", "pipe", n1.RemoteAddr().Network())
 
 		errs := xsync.Go(func() error {
 			_, err := n2.Write([]byte("hello"))

netconn.go

@@ -33,7 +33,7 @@ import (
 // reading/writing goroutines are interrupted but the connection is kept alive.
 //
 // The Addr methods will return a mock net.Addr that returns "websocket" for Network
-// and "websocket/unknown-addr" for String.
+// and "websocket/unknown-addr" for String when RemoteAddr and LocalAddr methods do not exist.
 //
 // A received StatusNormalClosure or StatusGoingAway close frame will be translated to
 // io.EOF when reading.
@@ -133,14 +133,6 @@ func (a websocketAddr) String() string {
 	return "websocket/unknown-addr"
 }
 
-func (c *netConn) RemoteAddr() net.Addr {
-	return websocketAddr{}
-}
-
-func (c *netConn) LocalAddr() net.Addr {
-	return websocketAddr{}
-}
-
 func (c *netConn) SetDeadline(t time.Time) error {
 	c.SetWriteDeadline(t)
 	c.SetReadDeadline(t)

netconn_js.go

@@ -0,0 +1,11 @@
+package websocket
+
+import "net"
+
+func (c *netConn) RemoteAddr() net.Addr {
+	return websocketAddr{}
+}
+
+func (c *netConn) LocalAddr() net.Addr {
+	return websocketAddr{}
+}

netconn_notjs.go

@@ -0,0 +1,26 @@
+// +build !js
+
+package websocket
+
+import "net"
+
+type LocalRemoteAddr interface {
+	RemoteAddr() net.Addr
+	LocalAddr() net.Addr
+}
+
+func (c *netConn) RemoteAddr() net.Addr {
+	if ra, ok := c.c.rwc.(LocalRemoteAddr); ok {
+		return ra.RemoteAddr()
+	} else {
+		return websocketAddr{}
+	}
+}
+
+func (c *netConn) LocalAddr() net.Addr {
+	if la, ok := c.c.rwc.(LocalRemoteAddr); ok {
+		return la.RemoteAddr()
+	} else {
+		return websocketAddr{}
+	}
+}