Fix for GH#6 - Create Step Definitions generates Reqnroll using State…

…ments for SpecFlow projects (#13) Co-authored-by: Gáspár Nagy <[email protected]>
reqnroll · Mar 11, 2024 · 9f95c1e · 9f95c1e
1 parent 86872fa
commit 9f95c1e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,4 +1,6 @@
 # [vNext]
+
+* FIX for Create Step Definition Snippets Generates Reqnroll Using Statements for SpecFlow Projects #6
 * Fix for GH7 - "Find step definitions usages command not visible for SpecFlow projects
 
 # v2024.1.49 - 2024-02-08

diff --git a/Reqnroll.VisualStudio/Editor/Commands/DefineStepsCommand.cs b/Reqnroll.VisualStudio/Editor/Commands/DefineStepsCommand.cs
@@ -126,9 +126,12 @@ private void SaveAsStepDefinitionClass(IProjectScope projectScope, string combin
             targetFolder = stepDefinitionsFolder;
             fileNamespace = fileNamespace + ".StepDefinitions";
         }
+        var projectTraits = projectScope.GetProjectSettings().ReqnrollProjectTraits;
+        var isSpecFlow = projectTraits.HasFlag(ReqnrollProjectTraits.LegacySpecFlow) || projectTraits.HasFlag(ReqnrollProjectTraits.SpecFlowCompatibility);
+        var libraryNameSpace = isSpecFlow ? "SpecFlow" : "Reqnroll";
 
         var template = "using System;" + newLine +
-                       "using Reqnroll;" + newLine +
+                       $"using {libraryNameSpace};" + newLine +
                        newLine +
                        $"namespace {fileNamespace}" + newLine +
                        "{" + newLine +