Flaky test investigation

Author: Youssef1313

src/Adapter/MSTest.TestAdapter/Helpers/FixtureMethodRunner.cs

@@ -1,6 +1,8 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT license. See LICENSE file in the project root for full license information.
 
+using System.Threading;
+
 using Microsoft.VisualStudio.TestPlatform.MSTest.TestAdapter.Execution;
 using Microsoft.VisualStudio.TestPlatform.MSTest.TestAdapter.Extensions;
 using Microsoft.VisualStudio.TestPlatform.MSTest.TestAdapter.ObjectModel;
@@ -125,7 +127,7 @@ internal static class FixtureMethodRunner
         Action action, CancellationTokenSource cancellationTokenSource, int timeout, MethodInfo methodInfo,
         IExecutionContextScope executionContextScope, string methodCanceledMessageFormat, string methodTimedOutMessageFormat)
     {
-        Exception? realException = null;
+       Exception? realException = null;
         Task? executionTask;
         try
         {
@@ -174,8 +176,15 @@ internal static class FixtureMethodRunner
                 UnitTestOutcome.Timeout,
                 string.Format(CultureInfo.InvariantCulture, methodCanceledMessageFormat, methodInfo.DeclaringType!.FullName, methodInfo.Name));
         }
-        catch (Exception)
+        catch (Exception ex)
         {
+            if (ex is AggregateException aggregateEx && aggregateEx.InnerExceptions.OfType<TaskCanceledException>().Any(tce => tce.CancellationToken == cancellationTokenSource.Token))
+            {
+                return new(
+                    UnitTestOutcome.Timeout,
+                    "Are we going to hit this??");
+            }
+
             // We throw the real exception to have the original stack trace to elaborate up the chain.
             if (realException is not null)
             {