websocket: fix socket leak in test

Possibly fixes issue 1694.

R=bradfitzgo
CC=golang-dev
https://golang.org/cl/4427049

diff --git a/src/pkg/websocket/websocket_test.go b/src/pkg/websocket/websocket_test.go
index 8b3cf8925a99dfa86ca165caab7ceb03769deacf..10f88dfd1a019f65474b4f5f33efbc9b4e8756ae 100644 (file)
--- a/src/pkg/websocket/websocket_test.go
+++ b/src/pkg/websocket/websocket_test.go
@@ -186,11 +186,12 @@ func TestTrailingSpaces(t *testing.T) {
        once.Do(startServer)
        for i := 0; i < 30; i++ {
                // body
-               _, err := Dial(fmt.Sprintf("ws://%s/echo", serverAddr), "",
-                       "http://localhost/")
+               ws, err := Dial(fmt.Sprintf("ws://%s/echo", serverAddr), "", "http://localhost/")
                if err != nil {
-                       panic("Dial failed: " + err.String())
+                       t.Error("Dial failed:", err.String())
+                       break
                }
+               ws.Close()
        }
 }