fix

src/libraries/System.Net.WebSockets.Client/src/System/Net/WebSockets/BrowserWebSockets/BrowserWebSocket.cs

@@ -542,18 +542,14 @@ private async Task CloseAsyncCore(WebSocketCloseStatus closeStatus, string? stat
 
         private async Task CancellationHelper(Task promise, CancellationToken cancellationToken, WebSocketState previousState, IDisposable? disposable = null)
         {
-            if (cancellationToken.IsCancellationRequested)
-            {
-                disposable?.Dispose();
-                cancellationToken.ThrowIfCancellationRequested();
-            }
-            if (promise.IsCompletedSuccessfully)
-            {
-                disposable?.Dispose();
-                return;
-            }
             try
             {
+                cancellationToken.ThrowIfCancellationRequested();
+                if (promise.IsCompletedSuccessfully)
+                {
+                    disposable?.Dispose();
+                    return;
+                }
                 if (promise.IsCompleted)
                 {
                     // don't have to register for cancelation
@@ -570,7 +566,7 @@ private async Task CancellationHelper(Task promise, CancellationToken cancellati
                     return;
                 }
             }
-            catch (JSException ex)
+            catch (Exception ex)
             {
                 lock (_lockObject)
                 {
@@ -580,7 +576,15 @@ private async Task CancellationHelper(Task promise, CancellationToken cancellati
                         ForceReadCloseStatusLocked();
                         throw new OperationCanceledException(nameof(WebSocketState.Aborted), ex);
                     }
-
+                    if (ex is OperationCanceledException)
+                    {
+                        if(state != WebSocketState.Closed)
+                        {
+                            FastState = WebSocketState.Aborted;
+                        }
+                        _cancelled = true;
+                        throw;
+                    }
                     if (state != WebSocketState.Closed && cancellationToken.IsCancellationRequested)
                     {
                         FastState = WebSocketState.Aborted;