diff options
Diffstat (limited to 'src/pkg/websocket/websocket_test.go')
-rw-r--r-- | src/pkg/websocket/websocket_test.go | 48 |
1 files changed, 24 insertions, 24 deletions
diff --git a/src/pkg/websocket/websocket_test.go b/src/pkg/websocket/websocket_test.go index ed2505313..c62604621 100644 --- a/src/pkg/websocket/websocket_test.go +++ b/src/pkg/websocket/websocket_test.go @@ -5,57 +5,57 @@ package websocket import ( - "bytes"; - "http"; - "io"; - "log"; - "net"; - "once"; - "strings"; - "testing"; + "bytes" + "http" + "io" + "log" + "net" + "once" + "strings" + "testing" ) var serverAddr string -func echoServer(ws *Conn) { io.Copy(ws, ws) } +func echoServer(ws *Conn) { io.Copy(ws, ws) } func startServer() { - l, e := net.Listen("tcp", ":0"); // any available address + l, e := net.Listen("tcp", ":0") // any available address if e != nil { log.Exitf("net.Listen tcp :0 %v", e) } - serverAddr = l.Addr().String(); - log.Stderr("Test WebSocket server listening on ", serverAddr); - http.Handle("/echo", Handler(echoServer)); - go http.Serve(l, nil); + serverAddr = l.Addr().String() + log.Stderr("Test WebSocket server listening on ", serverAddr) + http.Handle("/echo", Handler(echoServer)) + go http.Serve(l, nil) } func TestEcho(t *testing.T) { - once.Do(startServer); + once.Do(startServer) - client, err := net.Dial("tcp", "", serverAddr); + client, err := net.Dial("tcp", "", serverAddr) if err != nil { t.Fatal("dialing", err) } ws, err := newClient("/echo", "localhost", "http://localhost", - "ws://localhost/echo", "", client); + "ws://localhost/echo", "", client) if err != nil { - t.Errorf("WebSocket handshake error", err); - return; + t.Errorf("WebSocket handshake error", err) + return } - msg := strings.Bytes("hello, world\n"); + msg := strings.Bytes("hello, world\n") if _, err := ws.Write(msg); err != nil { t.Errorf("Write: error %v", err) } - var actual_msg = make([]byte, 512); - n, err := ws.Read(actual_msg); + var actual_msg = make([]byte, 512) + n, err := ws.Read(actual_msg) if err != nil { t.Errorf("Read: error %v", err) } - actual_msg = actual_msg[0:n]; + actual_msg = actual_msg[0:n] if !bytes.Equal(msg, actual_msg) { t.Errorf("Echo: expected %q got %q", msg, actual_msg) } - ws.Close(); + ws.Close() } |