move fetching ProjectInfo into its own method

SpecFlowOSS · Apr 25, 2019 · a035844 · a035844
1 parent 2bbb629
commit a035844
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/...cFlow.VsIntegration.Implementation/SingleFileGenerator/SpecFlowSingleFileGeneratorBase.cs b/...cFlow.VsIntegration.Implementation/SingleFileGenerator/SpecFlowSingleFileGeneratorBase.cs
@@ -37,11 +37,7 @@ protected override bool GenerateInternal(string inputFilePath, string inputFileC
                 return true;
             }
 
-            var vsProject = project.Object as VSProject2;
-            var references = vsProject?.References.Cast<Reference3>() ?? Enumerable.Empty<Reference3>();
-            var specFlowReference = references.FirstOrDefault(r => r.Name == "TechTalk.SpecFlow");
-            var referencedSpecFlowVersion = specFlowReference is Reference3 reference ? Version.Parse(reference.Version) : null;
-            var projectInfo = new ProjectInfo(project.Name, referencedSpecFlowVersion);
+            var projectInfo = GetProjectInfoForProject(project);
 
             var ideSingleFileGenerator = new IdeSingleFileGenerator(projectInfo);
 
@@ -64,6 +60,16 @@ protected override bool GenerateInternal(string inputFilePath, string inputFileC
             return outputFilePath != null;
         }
 
+        protected ProjectInfo GetProjectInfoForProject(Project project)
+        {
+            var vsProject = project.Object as VSProject2;
+            var references = vsProject?.References.Cast<Reference3>() ?? Enumerable.Empty<Reference3>();
+            var specFlowReference = references.FirstOrDefault(r => r.Name == "TechTalk.SpecFlow");
+            var referencedSpecFlowVersion = specFlowReference is Reference3 reference ? Version.Parse(reference.Version) : null;
+            var projectInfo = new ProjectInfo(project.Name, referencedSpecFlowVersion);
+            return projectInfo;
+        }
+
         protected abstract Func<GeneratorServices> GeneratorServicesProvider(Project project);
 
         protected override void AfterCodeGenerated(bool error)