websocket: use URL.RawPath to construct WebSocket-Location: header

R=rsc
CC=golang-dev
https://golang.org/cl/651041

src/pkg/websocket/server.go

@@ -66,7 +66,7 @@ func (f Handler) ServeHTTP(c *http.Conn, req *http.Request) {
 		return
 	}
 	defer rwc.Close()
-	location := "ws://" + req.Host + req.URL.Path
+	location := "ws://" + req.Host + req.URL.RawPath
 
 	// TODO(ukai): verify origin,location,protocol.
 

src/pkg/websocket/websocket_test.go

@@ -60,6 +60,23 @@ func TestEcho(t *testing.T) {
 	ws.Close()
 }
 
+func TestWithQuery(t *testing.T) {
+	once.Do(startServer)
+
+	client, err := net.Dial("tcp", "", serverAddr)
+	if err != nil {
+		t.Fatal("dialing", err)
+	}
+
+	ws, err := newClient("/echo?q=v", "localhost", "http://localhost",
+		"ws://localhost/echo?q=v", "", client)
+	if err != nil {
+		t.Errorf("WebSocket handshake error", err)
+		return
+	}
+	ws.Close()
+}
+
 func TestHTTP(t *testing.T) {
 	once.Do(startServer)