#2608: Await the call to FeatureBackgroundAsync (#2654)

SpecFlowOSS · Oct 24, 2022 · 631a2d5 · 631a2d5
1 parent 8d58ac0
commit 631a2d5
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));
                 }
             }