Allow add using to work on Razor generated documents (#79530)

davidwengier · web-flow · commit 3c013a34c3c7 · 2025-07-23T14:18:09.000+10:00
Completely missed this because Razor cohosting tests don't use the
Roslyn OOP.
diff --git a/src/Workspaces/Remote/ServiceHub/Services/MissingImportDiscovery/RemoteMissingImportDiscoveryService.cs b/src/Workspaces/Remote/ServiceHub/Services/MissingImportDiscovery/RemoteMissingImportDiscoveryService.cs
@@ -39,7 +39,7 @@ public ValueTask<ImmutableArray<AddImportFixData>> GetFixesAsync(
     {
         return RunServiceAsync(solutionChecksum, async solution =>
         {
-            var document = solution.GetDocument(documentId);
+            var document = await solution.GetDocumentAsync(documentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
             if (document is null)
                 return [];
 
@@ -68,7 +68,7 @@ public ValueTask<ImmutableArray<AddImportFixData>> GetUniqueFixesAsync(
     {
         return RunServiceAsync(solutionChecksum, async solution =>
         {
-            var document = solution.GetDocument(documentId);
+            var document = await solution.GetDocumentAsync(documentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
             if (document is null)
                 return [];
 

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ public ValueTask<ImmutableArray<AddImportFixData>> GetFixesAsync(`
`39`	`39`	`{`
`40`	`40`	`return RunServiceAsync(solutionChecksum, async solution =>`
`41`	`41`	`{`
`42`		`- var document = solution.GetDocument(documentId);`
	`42`	`+ var document = await solution.GetDocumentAsync(documentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);`
`43`	`43`	`if (document is null)`
`44`	`44`	`return [];`
`45`	`45`
`@@ -68,7 +68,7 @@ public ValueTask<ImmutableArray<AddImportFixData>> GetUniqueFixesAsync(`
`68`	`68`	`{`
`69`	`69`	`return RunServiceAsync(solutionChecksum, async solution =>`
`70`	`70`	`{`
`71`		`- var document = solution.GetDocument(documentId);`
	`71`	`+ var document = await solution.GetDocumentAsync(documentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);`
`72`	`72`	`if (document is null)`
`73`	`73`	`return [];`
`74`	`74`