Remove a first chance exception for normal cancellation

This exception only occurs in integration test scenarios, but the
internal performance test harness is not able to eliminate this false
positive automatically.

Author: sharwell

src/Workspaces/Core/Portable/Shared/TestHooks/AsynchronousOperationListener.cs

@@ -40,33 +40,50 @@ public AsynchronousOperationListener(string featureName, bool enableDiagnosticTo
             TrackActiveTokens = Debugger.IsAttached || enableDiagnosticTokens;
         }
 
-        public async Task<bool> Delay(TimeSpan delay, CancellationToken cancellationToken)
+        public Task<bool> Delay(TimeSpan delay, CancellationToken cancellationToken)
         {
-            cancellationToken.ThrowIfCancellationRequested();
+            if (cancellationToken.IsCancellationRequested)
+                return Task.FromCanceled<bool>(cancellationToken);
 
             var expeditedDelayCancellationToken = _expeditedDelayCancellationTokenSource.Token;
             if (expeditedDelayCancellationToken.IsCancellationRequested)
             {
                 // The operation is already being expedited
-                return false;
+                return SpecializedTasks.False;
             }
 
-            using var cancellationTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, expeditedDelayCancellationToken);
+            var cancellationTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, expeditedDelayCancellationToken);
 
             var delayTask = Task.Delay(delay, cancellationTokenSource.Token);
-            await delayTask.NoThrowAwaitableInternal(false);
-
-            if (delayTask.Status != TaskStatus.RanToCompletion)
+            if (delayTask.IsCompleted)
             {
-                cancellationToken.ThrowIfCancellationRequested();
-
-                // The cancellation only occurred due to a request to expedite the operation.
-                // Don't use try-catch to reduce the number of first chance exceptions.
-                if (expeditedDelayCancellationToken.IsCancellationRequested)
-                    return false;
+                cancellationTokenSource.Dispose();
+                if (delayTask.Status == TaskStatus.RanToCompletion)
+                    return SpecializedTasks.True;
+                else if (cancellationToken.IsCancellationRequested)
+                    return Task.FromCanceled<bool>(cancellationToken);
+                else
+                    return SpecializedTasks.False;
             }
 
-            return true;
+            return DelaySlow(delayTask, cancellationTokenSource, cancellationToken);
+
+            static Task<bool> DelaySlow(Task delayTask, CancellationTokenSource cancellationTokenSourceToDispose, CancellationToken cancellationToken)
+            {
+                return delayTask.ContinueWith(
+                    task =>
+                    {
+                        if (task.Status == TaskStatus.RanToCompletion)
+                            return SpecializedTasks.True;
+                        else if (cancellationToken.IsCancellationRequested)
+                            return Task.FromCanceled<bool>(cancellationToken);
+                        else
+                            return SpecializedTasks.False;
+                    },
+                    CancellationToken.None,
+                    TaskContinuationOptions.ExecuteSynchronously,
+                    TaskScheduler.Default).Unwrap();
+            }
         }
 
         public IAsyncToken BeginAsyncOperation(string name, object? tag = null, [CallerFilePath] string filePath = "", [CallerLineNumber] int lineNumber = 0)