misc: Format and compile #360

nhooyr · Oct 19, 2023 · 305eab9 · 305eab9
1 parent 309e088
commit 305eab9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/accept_test.go b/accept_test.go
@@ -9,11 +9,11 @@ import (
 	"net"
 	"net/http"
 	"net/http/httptest"
-	"nhooyr.io/websocket/internal/test/xrand"
 	"strings"
 	"testing"
 
 	"nhooyr.io/websocket/internal/test/assert"
+	"nhooyr.io/websocket/internal/test/xrand"
 )
 
 func TestAccept(t *testing.T) {
@@ -68,7 +68,7 @@ func TestAccept(t *testing.T) {
 			r.Header.Set("Connection", "Upgrade")
 			r.Header.Set("Upgrade", "websocket")
 			r.Header.Set("Sec-WebSocket-Version", "13")
-			r.Header.Set("Sec-WebSocket-Key", "meow123")
+			r.Header.Set("Sec-WebSocket-Key", xrand.Base64(16))
 			r.Header.Set("Sec-WebSocket-Extensions", extensions)
 			return r
 		}

diff --git a/internal/test/xrand/xrand.go b/internal/test/xrand/xrand.go
@@ -47,6 +47,7 @@ func Int(max int) int {
 	return int(x.Int64())
 }
 
+// Base64 returns a randomly generated base64 string of length n.
 func Base64(n int) string {
 	return base64.StdEncoding.EncodeToString(Bytes(n))
 }