chore: fix tests

internal/websocket/e2e_test.go

@@ -63,6 +63,10 @@ func TestEndToEnd(t *testing.T) {
 				c.Close(websocket.StatusNormalClosure, "")
 			}))
 	defer s.Close()
+	// Give our test server a few seconds to start.
+	// Otherwise, our websocket will get a "failed to WebSocket dial: failed to send handshake request: Get \"http://127.0.0.1:52644\":
+	// dial tcp6: address 127.0.0.1: no suitable address found" error
+	time.Sleep(2000 * time.Millisecond)
 
 	var msgCnt, connectCnt, disconnectCnt atomic.Int64
 
@@ -110,7 +114,7 @@ func TestEndToEnd(t *testing.T) {
 	// Allow multiple calls to start.
 	got.Start()
 
-	ctx, cancel := context.WithTimeout(context.Background(), 2000*time.Millisecond)
+	ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
 	defer cancel()
 
 	for {