Separate TestThreadContext from ITestRunner and handle ITestRunner.Te…

…stWorkerId test framework independently

CHANGELOG.md

@@ -1,5 +1,7 @@
 # [vNext]
 
+* Separate `TestThreadContext` from `ITestRunner` and handle `ITestRunner.TestWorkerId` test framework independently
+
 # v2.0.1 - 2024-05-29
 
 ## Bug fixes:

Plugins/Reqnroll.MSTest.Generator.ReqnrollPlugin/build/MSTest.AssemblyHooks.template.vb

@@ -16,7 +16,7 @@ Public NotInheritable Class PROJECT_ROOT_NAMESPACE_MSTestAssemblyHooks
     Public Shared Async Function AssemblyInitializeAsync(testContext As TestContext) As Task
         Dim currentAssembly As Assembly = GetType(PROJECT_ROOT_NAMESPACE_MSTestAssemblyHooks).Assembly
         Dim containerBuilder As New MsTestContainerBuilder(testContext)
-        Await Global.Reqnroll.TestRunnerManager.OnTestRunStartAsync(currentAssembly, Nothing, containerBuilder)
+        Await Global.Reqnroll.TestRunnerManager.OnTestRunStartAsync(currentAssembly, containerBuilder)
     End Function
 
     <AssemblyCleanup>

Reqnroll.Generator/Generation/UnitTestFeatureGenerator.cs

@@ -177,15 +177,9 @@ private void SetupTestClassInitializeMethod(TestClassGenerationContext generatio
             //testRunner = TestRunnerManager.GetTestRunnerForAssembly(null, [test_worker_id]);
             var testRunnerField = _scenarioPartHelper.GetTestRunnerExpression();
 
-            var testRunnerParameters = new[]
-            {
-                new CodePrimitiveExpression(null),
-                _testGeneratorProvider.GetTestWorkerIdExpression()
-            };
-
             var getTestRunnerExpression = new CodeMethodInvokeExpression(
                 new CodeTypeReferenceExpression(_codeDomHelper.GetGlobalizedTypeName(typeof(TestRunnerManager))),
-                nameof(TestRunnerManager.GetTestRunnerForAssembly), testRunnerParameters);
+                nameof(TestRunnerManager.GetTestRunnerForAssembly));
 
             testClassInitializeMethod.Statements.Add(
                 new CodeAssignStatement(

Reqnroll.Generator/UnitTestProvider/IUnitTestGeneratorProvider.cs

@@ -28,7 +28,5 @@ public interface IUnitTestGeneratorProvider
         void SetTestMethodAsRow(TestClassGenerationContext generationContext, CodeMemberMethod testMethod, string scenarioTitle, string exampleSetName, string variantName, IEnumerable<KeyValuePair<string, string>> arguments);
 
         void MarkCodeMethodInvokeExpressionAsAwait(CodeMethodInvokeExpression expression);
-
-        CodeExpression GetTestWorkerIdExpression();
     }
 }

Reqnroll.Generator/UnitTestProvider/MsTestGeneratorProvider.cs

@@ -228,14 +228,5 @@ public void MarkCodeMethodInvokeExpressionAsAwait(CodeMethodInvokeExpression exp
         {
             CodeDomHelper.MarkCodeMethodInvokeExpressionAsAwait(expression);
         }
-
-        public CodeExpression GetTestWorkerIdExpression()
-        {
-            // System.Threading.Thread.CurrentThread.ManagedThreadId.ToString()
-            return new CodeMethodInvokeExpression(
-                new CodeVariableReferenceExpression("System.Threading.Thread.CurrentThread.ManagedThreadId"),
-                nameof(ToString)
-            );
-        }
     }
 }

Reqnroll.Generator/UnitTestProvider/NUnit3TestGeneratorProvider.cs

@@ -154,8 +154,5 @@ public void MarkCodeMethodInvokeExpressionAsAwait(CodeMethodInvokeExpression exp
         {
             CodeDomHelper.MarkCodeMethodInvokeExpressionAsAwait(expression);
         }
-
-        public CodeExpression GetTestWorkerIdExpression() 
-            => new CodePropertyReferenceExpression(GetTestContextExpression(), TESTCONTEXT_WORKERID_PROPERTY);
     }
 }

Reqnroll.Generator/UnitTestProvider/XUnit2TestGeneratorProvider.cs

@@ -33,7 +33,6 @@ public class XUnit2TestGeneratorProvider : IUnitTestGeneratorProvider
         protected internal const string IGNORE_TEST_CLASS = "IgnoreTestClass";
         protected internal const string NONPARALLELIZABLE_COLLECTION_NAME = "ReqnrollNonParallelizableFeatures";
         protected internal const string IASYNCLIFETIME_INTERFACE = "Xunit.IAsyncLifetime";
-        protected internal const string XUNITPARALLELWORKERTRACKER_INSTANCE = "Reqnroll.xUnit.ReqnrollPlugin.XUnitParallelWorkerTracker.Instance";
 
         public XUnit2TestGeneratorProvider(CodeDomHelper codeDomHelper)
         {
@@ -178,19 +177,6 @@ public virtual void FinalizeTestClass(TestClassGenerationContext generationConte
                         nameof(IObjectContainer.RegisterInstanceAs),
                         new CodeTypeReference(OUTPUT_INTERFACE)),
                     new CodeVariableReferenceExpression(OUTPUT_INTERFACE_FIELD_NAME)));
-
-            // Wrap FeatureTearDown:
-            // var testWorkerId = <testRunner>.TestWorkerId;
-            // <FeatureTearDown>
-            // XUnitParallelWorkerTracker.Instance.ReleaseWorker(testWorkerId);
-            generationContext.TestClassCleanupMethod.Statements.Insert(0, 
-                new CodeVariableDeclarationStatement(typeof(string), "testWorkerId", 
-                    new CodePropertyReferenceExpression(new CodeVariableReferenceExpression(generationContext.TestRunnerField.Name), "TestWorkerId")));
-            generationContext.TestClassCleanupMethod.Statements.Add(
-                new CodeMethodInvokeExpression(
-                    new CodeVariableReferenceExpression(GlobalNamespaceIfCSharp(XUNITPARALLELWORKERTRACKER_INSTANCE)),
-                    "ReleaseWorker",
-                    new CodeVariableReferenceExpression("testWorkerId")));
         }
 
         protected virtual void IgnoreFeature(TestClassGenerationContext generationContext)
@@ -414,14 +400,6 @@ public void MarkCodeMethodInvokeExpressionAsAwait(CodeMethodInvokeExpression exp
             CodeDomHelper.MarkCodeMethodInvokeExpressionAsAwait(expression);
         }
 
-        public CodeExpression GetTestWorkerIdExpression()
-        {
-            // XUnitParallelWorkerTracker.Instance.GetWorkerId()
-            return new CodeMethodInvokeExpression(
-                new CodeVariableReferenceExpression(GlobalNamespaceIfCSharp(XUNITPARALLELWORKERTRACKER_INSTANCE)),
-                "GetWorkerId");
-        }
-
         private string GlobalNamespaceIfCSharp(string typeName)
         {
             return CodeDomHelper.TargetLanguage == CodeDomProviderLanguage.CSharp ? "global::" + typeName : typeName;

Reqnroll/ISyncTestRunner.cs

@@ -5,7 +5,7 @@ namespace Reqnroll
     public interface ISyncTestRunner
     {
         /// <summary>
-        /// The ID of the parallel test worker processing the current scenario. How the worker ID is obtained is dependent on the test execution framework.
+        /// The ID of the parallel test worker processing the current scenario.
         /// </summary>
         string TestWorkerId { get; }
 

Reqnroll/ITestRunner.cs

@@ -1,17 +1,19 @@
+using System;
 using System.Threading.Tasks;
 
 namespace Reqnroll
 {
     public interface ITestRunner
     {
         /// <summary>
-        /// The ID of the parallel test worker processing the current scenario. How the worker ID is obtained is dependent on the test execution framework.
+        /// The ID of the parallel test worker processing the current scenario.
         /// </summary>
         string TestWorkerId { get; }
         FeatureContext FeatureContext { get; }
         ScenarioContext ScenarioContext { get; }
         ITestThreadContext TestThreadContext { get; }
 
+        [Obsolete("TestWorkerId is now managed by Reqnroll internally - Method will be removed in v3")]
         void InitializeTestRunner(string testWorkerId);
 
         Task OnTestRunStartAsync();

Reqnroll/ITestRunnerManager.cs

@@ -10,7 +10,8 @@ public interface ITestRunnerManager
         Assembly TestAssembly { get; }
         Assembly[] BindingAssemblies { get; }
         bool IsMultiThreaded { get; }
-        ITestRunner GetTestRunner(string workerId);
+        ITestRunner GetTestRunner();
+        void ReleaseTestThreadContext(ITestThreadContext testThreadContext);
         void Initialize(Assembly testAssembly);
         Task FireTestRunEndAsync();
         Task FireTestRunStartAsync();

Reqnroll/Infrastructure/TestExecutionEngine.cs

@@ -162,6 +162,7 @@ public virtual async Task OnFeatureEndAsync()
             _testThreadExecutionEventPublisher.PublishEvent(new FeatureFinishedEvent(FeatureContext));
 
             _contextManager.CleanupFeatureContext();
+            _testRunnerManager.ReleaseTestThreadContext(TestThreadContext);
         }
 
         public virtual void OnScenarioInitialize(ScenarioInfo scenarioInfo)

Reqnroll/TestRunner.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Threading.Tasks;
 using Reqnroll.Bindings;
 using Reqnroll.Infrastructure;
@@ -8,7 +9,7 @@ public class TestRunner : ITestRunner
     {
         private readonly ITestExecutionEngine _executionEngine;
 
-        public string TestWorkerId { get; private set; }
+        public string TestWorkerId => TestThreadContainerInfo.GetId(TestThreadContext.TestThreadContainer);
 
         public TestRunner(ITestExecutionEngine executionEngine)
         {
@@ -26,9 +27,10 @@ public async Task OnTestRunStartAsync()
             await _executionEngine.OnTestRunStartAsync();
         }
 
+        [Obsolete("TestWorkerId is now managed by Reqnroll internally - Method will be removed in v3")]
         public void InitializeTestRunner(string testWorkerId)
         {
-            TestWorkerId = testWorkerId;
+            // do nothing method will be removed
         }
 
         public async Task OnFeatureStartAsync(FeatureInfo featureInfo)