Fix generate file

src/OmniSharp.Roslyn.CSharp/Services/Refactoring/V2/RunCodeActionService.cs

@@ -73,7 +73,7 @@ public override async Task<RunCodeActionResponse> Handle(RunCodeActionRequest re
                     var fileChanges = await GetFileChangesAsync(applyChangesOperation.ChangedSolution, solution, directory, request.WantsTextChanges);
 
                     changes.AddRange(fileChanges);
-                    solution = applyChangesOperation.ChangedSolution;
+                    solution = this.Workspace.CurrentSolution;
                 }
 
                 if (request.WantsAllCodeActionOperations)
@@ -179,7 +179,7 @@ private async Task<IEnumerable<ModifiedFileResponse>> GetFileChangesAsync(Soluti
                             }
                         }
 
-                        this.Workspace.AddDocument(projectChange.ProjectId, newFilePath, newDocument.SourceCodeKind);
+                        this.Workspace.AddDocument(documentId, projectChange.ProjectId, newFilePath, newDocument.SourceCodeKind);
                     }
                     else
                     {

src/OmniSharp.Roslyn/OmniSharpWorkspace.cs

@@ -86,6 +86,12 @@ public void AddDocument(DocumentInfo documentInfo)
         public DocumentId AddDocument(ProjectId projectId, string filePath, SourceCodeKind sourceCodeKind = SourceCodeKind.Regular)
         {
             var documentId = DocumentId.CreateNewId(projectId);
+            this.AddDocument(documentId, projectId, filePath, sourceCodeKind);
+            return documentId;
+        }
+
+        public DocumentId AddDocument(DocumentId documentId, ProjectId projectId, string filePath, SourceCodeKind sourceCodeKind = SourceCodeKind.Regular)
+        {
             var loader = new OmniSharpTextLoader(filePath);
             var documentInfo = DocumentInfo.Create(documentId, filePath, filePath: filePath, loader: loader, sourceCodeKind: sourceCodeKind);
 

test-assets/test-projects/ProjectWithMissingType/HelloWorld.csproj

@@ -0,0 +1,21 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <OutputType>Exe</OutputType>
+    <TargetFramework>netcoreapp1.0</TargetFramework>
+    <EnableDefaultCompileItems>false</EnableDefaultCompileItems>
+    <LangVersion>7.1</LangVersion>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <Compile Include="**\*.cs" />
+    <EmbeddedResource Include="**\*.resx" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <PackageReference Include="Microsoft.NETCore.App">
+      <Version>1.0.1</Version>
+    </PackageReference>
+  </ItemGroup>
+
+</Project>

test-assets/test-projects/ProjectWithMissingType/Program.cs

@@ -0,0 +1,12 @@
+using System;
+
+namespace ConsoleApplication
+{
+    public class Program
+    {
+        public static void Main(string[] args)
+        {
+            Z x = null;
+        }
+    }
+}

tests/OmniSharp.Roslyn.CSharp.Tests/CodeActionsV2Facts.cs

@@ -139,7 +139,6 @@ public void Whatever()
                         [|Console.Write(""should be using System;"");|]
                     }
                 }";
-
             const string expected =
                 @"public class Class1
                 {
@@ -153,11 +152,40 @@ private static void NewMethod()
                         Console.Write(""should be using System;"");
                     }
                 }";
-
             var response = await RunRefactoringAsync(code, "Extract Method");
             AssertIgnoringIndent(expected, ((ModifiedFileResponse)response.Changes.First()).Buffer);
         }
 
+        [Fact]
+        public async Task Can_generate_type_and_return_name_of_new_file()
+        {
+            using (var testProject = await TestAssets.Instance.GetTestProjectAsync("ProjectWithMissingType"))
+            using (var host = CreateOmniSharpHost(testProject.Directory))
+            {
+                var requestHandler = host.GetRequestHandler<RunCodeActionService>(OmniSharpEndpoints.V2.RunCodeAction);
+                var document = host.Workspace.CurrentSolution.Projects.First().Documents.First();
+                var buffer = await document.GetTextAsync();
+                var path = document.FilePath;
+
+                var request = new RunCodeActionRequest
+                {
+                    Line = 8,
+                    Column = 12,
+                    FileName = path,
+                    Buffer = buffer.ToString(),
+                    Identifier = "Generate class 'Z' in new file",
+                    WantsTextChanges = true,
+                    WantsAllCodeActionOperations = true
+                };
+
+                var response = await requestHandler.Handle(request);
+                var changes = response.Changes.ToArray();
+                Assert.Equal(2, changes.Length);
+                Assert.NotNull(changes[0].FileName);
+                Assert.NotNull(changes[1].FileName);
+            }
+        }
+
         [Fact]
         public async Task Can_send_rename_and_fileOpen_responses_when_codeAction_renames_file()
         {