SpecFlowOSS#2608: Await the call to FeatureBackgroundAsync

laingsimon · Oct 15, 2022 · 0f90e76 · 0f90e76
1 parent b3a131b
commit 0f90e76
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/TechTalk.SpecFlow.Generator/Generation/UnitTestMethodGenerator.cs b/TechTalk.SpecFlow.Generator/Generation/UnitTestMethodGenerator.cs
@@ -256,10 +256,12 @@ internal void GenerateTestMethodBody(TestClassGenerationContext generationContex
             {
                 using (new SourceLineScope(_specFlowConfiguration, _codeDomHelper, statementsWhenScenarioIsExecuted, generationContext.Document.SourceFilePath, generationContext.Feature.Background.Location))
                 {
-                    statementsWhenScenarioIsExecuted.Add(new CodeExpressionStatement(
-                        new CodeMethodInvokeExpression(
-                            new CodeThisReferenceExpression(),
-                            generationContext.FeatureBackgroundMethod.Name)));
+                    var backgroundMethodCallExpression = new CodeMethodInvokeExpression(
+                        new CodeThisReferenceExpression(),
+                        generationContext.FeatureBackgroundMethod.Name);
+
+                    _codeDomHelper.MarkCodeMethodInvokeExpressionAsAwait(backgroundMethodCallExpression);
+                    statementsWhenScenarioIsExecuted.Add(new CodeExpressionStatement(backgroundMethodCallExpression));
                 }
             }