went async was for Razor

Author: davidwengier

src/EditorFeatures/Core/InlineRename/AbstractEditorInlineRenameService.InlineRenameLocationSet.cs

@@ -7,6 +7,7 @@
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.CodeAnalysis.Host;
 using Microsoft.CodeAnalysis.Rename;
 using Microsoft.CodeAnalysis.Shared.Extensions;
 using Roslyn.Utilities;
@@ -47,6 +48,8 @@ private InlineRenameLocationSet(
         private static async ValueTask<InlineRenameLocation> ConvertLocationAsync(Solution solution, RenameLocation location, CancellationToken cancellationToken)
         {
             var document = await solution.GetRequiredDocumentAsync(location.DocumentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
+
+            Contract.ThrowIfTrue(location.DocumentId.IsSourceGenerated && !document.IsRazorSourceGeneratedDocument());
             return new InlineRenameLocation(document, location.Location.SourceSpan);
         }
 

src/EditorFeatures/Core/InlineRename/InlineRenameSession.cs

@@ -939,7 +939,8 @@ private async Task<Solution> GetFinalSolutionAsync(ImmutableArray<(DocumentId, s
         {
             if (documentId.IsSourceGenerated)
             {
-                _ = await finalSolution.GetDocumentAsync(documentId, includeSourceGenerated: true, CancellationToken.None).ConfigureAwait(false);
+                var document = await finalSolution.GetDocumentAsync(documentId, includeSourceGenerated: true, CancellationToken.None).ConfigureAwait(false);
+                Contract.ThrowIfFalse(document.IsRazorSourceGeneratedDocument());
             }
 
             finalSolution = newRoot != null

src/VisualStudio/Core/Def/ProjectSystem/VisualStudioWorkspaceImpl.cs

@@ -672,7 +672,8 @@ internal override void ApplyMappedFileChanges(SolutionChanges solutionChanges)
                 // so the mapper has something to compare to.
                 foreach (var (docId, state) in solutionChanges.NewSolution.CompilationState.FrozenSourceGeneratedDocumentStates.States)
                 {
-                    _ = await solutionChanges.OldSolution.GetRequiredDocumentAsync(docId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
+                    var document = await solutionChanges.OldSolution.GetRequiredDocumentAsync(docId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
+                    Contract.ThrowIfFalse(document.IsRazorSourceGeneratedDocument());
                 }
 
                 foreach (var docId in solutionChanges.GetExplicitlyChangedSourceGeneratedDocuments())

src/Workspaces/Core/Portable/Remote/RemoteUtilities.cs

@@ -7,6 +7,7 @@
 using System.Collections.Immutable;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.CodeAnalysis.Host;
 using Microsoft.CodeAnalysis.PooledObjects;
 using Microsoft.CodeAnalysis.Shared.Extensions;
 using Microsoft.CodeAnalysis.Text;
@@ -65,6 +66,8 @@ public static async Task<Solution> UpdateSolutionAsync(
             .SelectAsArrayAsync(async (tuple, cancellationToken) =>
             {
                 var document = await oldSolution.GetDocumentAsync(tuple.documentId, includeSourceGenerated: true, cancellationToken).ConfigureAwait(false);
+                Contract.ThrowIfTrue(tuple.documentId.IsSourceGenerated && !document.IsRazorSourceGeneratedDocument());
+
                 var oldText = await document.GetValueTextAsync(cancellationToken).ConfigureAwait(false);
                 var newText = oldText.WithChanges(tuple.textChanges);
                 return (tuple.documentId, newText);

src/Workspaces/Core/Portable/Rename/IRemoteRenamerService.cs

@@ -7,6 +7,7 @@
 using System.Runtime.Serialization;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.CodeAnalysis.Host;
 using Microsoft.CodeAnalysis.Remote;
 using Microsoft.CodeAnalysis.Rename.ConflictEngine;
 using Microsoft.CodeAnalysis.Shared.Extensions;
@@ -86,6 +87,8 @@ public static SerializableRenameLocation Dehydrate(RenameLocation location)
     public async ValueTask<RenameLocation> RehydrateAsync(Solution solution, CancellationToken cancellation)
     {
         var document = await solution.GetRequiredDocumentAsync(DocumentId, includeSourceGenerated: true, cancellation).ConfigureAwait(false);
+        Contract.ThrowIfTrue(DocumentId.IsSourceGenerated && !document.IsRazorSourceGeneratedDocument());
+
         var tree = await document.GetRequiredSyntaxTreeAsync(cancellation).ConfigureAwait(false);
 
         return new RenameLocation(