Use CTS from testContext to detect cancellation

microsoft · Dec 19, 2024 · e906ec5 · e906ec5
1 parent 4c74a53
commit e906ec5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Adapter/MSTest.TestAdapter/Execution/TestAssemblyInfo.cs b/src/Adapter/MSTest.TestAdapter/Execution/TestAssemblyInfo.cs
@@ -287,7 +287,7 @@ internal void ExecuteAssemblyCleanup(TestContext testContext)
                              AssemblyCleanupMethod.InvokeAsSynchronousTask(null, testContext);
                          }
                      },
-                     new CancellationTokenSource(),
+                     testContext.CancellationTokenSource,
                      AssemblyCleanupMethodTimeoutMilliseconds,
                      AssemblyCleanupMethod,
                      new AssemblyExecutionContextScope(isCleanup: true),

diff --git a/src/Adapter/MSTest.TestAdapter/Execution/TestClassInfo.cs b/src/Adapter/MSTest.TestAdapter/Execution/TestClassInfo.cs
@@ -765,7 +765,7 @@ void DoRun()
                     methodInfo.InvokeAsSynchronousTask(null, testContext);
                 }
             },
-            new CancellationTokenSource(),
+            testContext.CancellationTokenSource,
             timeout,
             methodInfo,
             new ClassExecutionContextScope(ClassType, remainingCleanupCount),