Update document mapping to work with the source generator (#11558)

davidwengier · web-flow · commit d84fe4c03e25 · 2025-03-03T13:58:45.000+11:00
Fixes a bunch of cohost functionality post source generator hookup Part of #10693
diff --git a/src/Compiler/Microsoft.CodeAnalysis.Razor.Compiler/src/SourceGenerators/RazorGeneratorResult.cs b/src/Compiler/Microsoft.CodeAnalysis.Razor.Compiler/src/SourceGenerators/RazorGeneratorResult.cs
@@ -7,12 +7,14 @@
 
 namespace Microsoft.NET.Sdk.Razor.SourceGenerators
 {
-    internal sealed class RazorGeneratorResult(IReadOnlyList<TagHelperDescriptor> tagHelpers, ImmutableDictionary<string, (string hintName, RazorCodeDocument document)> documents)
+    internal sealed class RazorGeneratorResult(IReadOnlyList<TagHelperDescriptor> tagHelpers, ImmutableDictionary<string, (string hintName, RazorCodeDocument document)> filePathToDocument, ImmutableDictionary<string, string> hintNameToFilePath)
     {
         public IReadOnlyList<TagHelperDescriptor> TagHelpers => tagHelpers;
 
-        public RazorCodeDocument? GetCodeDocument(string physicalPath) => documents.TryGetValue(physicalPath, out var pair) ? pair.document : null;
+        public RazorCodeDocument? GetCodeDocument(string physicalPath) => filePathToDocument.TryGetValue(physicalPath, out var pair) ? pair.document : null;
 
-        public string? GetHintName(string physicalPath) => documents.TryGetValue(physicalPath, out var pair) ? pair.hintName : null;
+        public string? GetHintName(string physicalPath) => filePathToDocument.TryGetValue(physicalPath, out var pair) ? pair.hintName : null;
+
+        public string? GetFilePath(string hintName) => hintNameToFilePath.TryGetValue(hintName, out var filePath) ? filePath : null;
     }
 }
diff --git a/src/Compiler/Microsoft.CodeAnalysis.Razor.Compiler/src/SourceGenerators/RazorSourceGenerator.cs b/src/Compiler/Microsoft.CodeAnalysis.Razor.Compiler/src/SourceGenerators/RazorSourceGenerator.cs
@@ -3,10 +3,11 @@
 
 using System;
 using System.Collections.Generic;
-using System.Collections.Immutable;
 using System.IO;
 using System.Linq;
+using Microsoft.AspNetCore.Razor;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.AspNetCore.Razor.PooledObjects;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 
@@ -346,8 +347,16 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 
                 if (!isGeneratorSuppressed)
                 {
-                    var documentDictionary = documents.Select(p => KeyValuePair.Create(p.codeDocument.Source.FilePath!, (p.hintName, p.codeDocument))).ToImmutableDictionary();
-                    context.AddOutput(nameof(RazorGeneratorResult), new RazorGeneratorResult(tagHelpers, documentDictionary));
+                    using var filePathToDocument = new PooledDictionaryBuilder<string, (string, RazorCodeDocument)>();
+                    using var hintNameToFilePath = new PooledDictionaryBuilder<string, string>();
+
+                    foreach (var (hintName, codeDocument, _) in documents)
+                    {
+                        filePathToDocument.Add(codeDocument.Source.FilePath!, (hintName, codeDocument));
+                        hintNameToFilePath.Add(hintName, codeDocument.Source.FilePath!);
+                    }
+
+                    context.AddOutput(nameof(RazorGeneratorResult), new RazorGeneratorResult(tagHelpers, filePathToDocument.ToImmutable(), hintNameToFilePath.ToImmutable()));
                 }
             });
         }
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Razor.Workspaces/Extensions/ProjectExtensions.cs b/src/Razor/src/Microsoft.CodeAnalysis.Razor.Workspaces/Extensions/ProjectExtensions.cs
@@ -7,6 +7,7 @@
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Razor;
+using Microsoft.CodeAnalysis.ExternalAccess.Razor;
 
 namespace Microsoft.CodeAnalysis;
 
@@ -40,7 +41,7 @@ public static bool TryGetCSharpDocument(this Project project, Uri csharpDocument
     /// <summary>
     /// Finds source generated documents by iterating through all of them. In OOP there are better options!
     /// </summary>
-    public static async Task<Document?> TryGetSourceGeneratedDocumentFromHintNameAsync(this Project project, string? hintName, CancellationToken cancellationToken)
+    public static async Task<SourceGeneratedDocument?> TryGetSourceGeneratedDocumentFromHintNameAsync(this Project project, string? hintName, CancellationToken cancellationToken)
     {
         // TODO: use this when the location is case-insensitive on windows (https://github.com/dotnet/roslyn/issues/76869)
         //var generator = typeof(RazorSourceGenerator);
@@ -51,4 +52,16 @@ public static bool TryGetCSharpDocument(this Project project, Uri csharpDocument
         var generatedDocuments = await project.GetSourceGeneratedDocumentsAsync(cancellationToken).ConfigureAwait(false);
         return generatedDocuments.SingleOrDefault(d => d.HintName == hintName);
     }
+
+    /// <summary>
+    /// Finds source generated documents by iterating through all of them. In OOP there are better options!
+    /// </summary>
+    public static async Task<string?> TryGetHintNameFromGeneratedDocumentUriAsync(this Project project, Uri generatedDocumentUri, CancellationToken cancellationToken)
+    {
+        // TODO: Call SourceGeneratedDocumentUri.DeserializeIdentity: https://github.com/dotnet/razor/issues/11557
+
+        var generatedDocuments = await project.GetSourceGeneratedDocumentsAsync(cancellationToken).ConfigureAwait(false);
+        var document = generatedDocuments.SingleOrDefault(d => generatedDocumentUri.Equals(d.CreateUri()));
+        return document?.HintName;
+    }
 }
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/Completion/RemoteCompletionService.cs b/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/Completion/RemoteCompletionService.cs
@@ -171,7 +171,7 @@ private async ValueTask<Response> GetCompletionAsync(
             var generatedText = await generatedDocument.GetTextAsync(cancellationToken).ConfigureAwait(false);
             var change = generatedText.GetTextChange(provisionalTextEdit);
             generatedText = generatedText.WithChanges([change]);
-            generatedDocument = generatedDocument.WithText(generatedText);
+            generatedDocument = (SourceGeneratedDocument)generatedDocument.WithText(generatedText);
         }
 
         // This is, to say the least, not ideal. In future we're going to normalize on to Roslyn LSP types, and this can go.
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/DocumentMapping/RemoteDocumentMappingService.cs b/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/DocumentMapping/RemoteDocumentMappingService.cs
@@ -2,11 +2,13 @@
 // Licensed under the MIT license. See License.txt in the project root for license information.
 
 using System;
+using System.Collections.Generic;
 using System.Composition;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Razor;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.CodeAnalysis.ExternalAccess.Razor;
 using Microsoft.CodeAnalysis.Razor.DocumentMapping;
 using Microsoft.CodeAnalysis.Razor.Logging;
 using Microsoft.CodeAnalysis.Razor.Workspaces;
@@ -32,11 +34,10 @@ internal sealed class RemoteDocumentMappingService(
         LinePositionSpan generatedDocumentRange,
         CancellationToken cancellationToken)
     {
-        var razorDocumentUri = _filePathService.GetRazorDocumentUri(generatedDocumentUri);
-
         // For Html we just map the Uri, the range will be the same
         if (_filePathService.IsVirtualHtmlFile(generatedDocumentUri))
         {
+            var razorDocumentUri = _filePathService.GetRazorDocumentUri(generatedDocumentUri);
             return (razorDocumentUri, generatedDocumentRange);
         }
 
@@ -46,31 +47,20 @@ internal sealed class RemoteDocumentMappingService(
             return (generatedDocumentUri, generatedDocumentRange);
         }
 
-        var solution = originSnapshot.TextDocument.Project.Solution;
-        if (!solution.TryGetRazorDocument(razorDocumentUri, out var razorDocument))
-        {
-            return (generatedDocumentUri, generatedDocumentRange);
-        }
-
-        var razorDocumentSnapshot = _snapshotManager.GetSnapshot(razorDocument);
-
-        var razorCodeDocument = await razorDocumentSnapshot
-            .GetGeneratedOutputAsync(cancellationToken)
-            .ConfigureAwait(false);
-
+        var project = originSnapshot.TextDocument.Project;
+        var razorCodeDocument = await _snapshotManager.GetSnapshot(project).TryGetCodeDocumentFromGeneratedDocumentUriAsync(generatedDocumentUri, cancellationToken).ConfigureAwait(false);
         if (razorCodeDocument is null)
         {
             return (generatedDocumentUri, generatedDocumentRange);
         }
 
-        if (!razorCodeDocument.TryGetGeneratedDocument(generatedDocumentUri, _filePathService, out var generatedDocument))
-        {
-            return Assumed.Unreachable<(Uri, LinePositionSpan)>();
-        }
-
-        if (TryMapToHostDocumentRange(generatedDocument, generatedDocumentRange, MappingBehavior.Strict, out var mappedRange))
+        if (TryMapToHostDocumentRange(razorCodeDocument.GetCSharpDocument(), generatedDocumentRange, MappingBehavior.Strict, out var mappedRange))
         {
-            return (razorDocumentUri, mappedRange);
+            var solution = project.Solution;
+            var filePath = razorCodeDocument.Source.FilePath;
+            var documentId = solution.GetDocumentIdsWithFilePath(filePath).First();
+            var document = solution.GetAdditionalDocument(documentId).AssumeNotNull();
+            return (document.CreateUri(), mappedRange);
         }
 
         return (generatedDocumentUri, generatedDocumentRange);
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/ProjectSystem/RemoteDocumentSnapshot.cs b/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/ProjectSystem/RemoteDocumentSnapshot.cs
@@ -22,7 +22,7 @@ internal sealed class RemoteDocumentSnapshot : IDocumentSnapshot
     public RemoteProjectSnapshot ProjectSnapshot { get; }
 
     private RazorCodeDocument? _codeDocument;
-    private Document? _generatedDocument;
+    private SourceGeneratedDocument? _generatedDocument;
 
     public RemoteDocumentSnapshot(TextDocument textDocument, RemoteProjectSnapshot projectSnapshot)
     {
@@ -103,7 +103,7 @@ public IDocumentSnapshot WithText(SourceText text)
         return snapshotManager.GetSnapshot(newDocument);
     }
 
-    public async ValueTask<Document> GetGeneratedDocumentAsync(CancellationToken cancellationToken)
+    public async ValueTask<SourceGeneratedDocument> GetGeneratedDocumentAsync(CancellationToken cancellationToken)
     {
         if (_generatedDocument is not null)
         {
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/ProjectSystem/RemoteProjectSnapshot.cs b/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/ProjectSystem/RemoteProjectSnapshot.cs
@@ -169,7 +169,7 @@ public bool TryGetDocument(string filePath, [NotNullWhen(true)] out IDocumentSna
         return generatorResult.GetCodeDocument(documentSnapshot.FilePath);
     }
 
-    internal async Task<Document?> GetGeneratedDocumentAsync(IDocumentSnapshot documentSnapshot, CancellationToken cancellationToken)
+    internal async Task<SourceGeneratedDocument?> GetGeneratedDocumentAsync(IDocumentSnapshot documentSnapshot, CancellationToken cancellationToken)
     {
         var generatorResult = await GetRazorGeneratorResultAsync(cancellationToken).ConfigureAwait(false);
         if (generatorResult is null)
@@ -184,6 +184,29 @@ public bool TryGetDocument(string filePath, [NotNullWhen(true)] out IDocumentSna
         return generatedDocument ?? throw new InvalidOperationException("Couldn't get the source generated document for a hint name that we got from the generator?");
     }
 
+    public async Task<RazorCodeDocument?> TryGetCodeDocumentFromGeneratedDocumentUriAsync(Uri generatedDocumentUri, CancellationToken cancellationToken)
+    {
+        if (await _project.TryGetHintNameFromGeneratedDocumentUriAsync(generatedDocumentUri, cancellationToken).ConfigureAwait(false) is not { } hintName)
+        {
+            return null;
+        }
+
+        return await TryGetCodeDocumentFromGeneratedHintNameAsync(hintName, cancellationToken).ConfigureAwait(false);
+    }
+
+    public async Task<RazorCodeDocument?> TryGetCodeDocumentFromGeneratedHintNameAsync(string generatedDocumentHintName, CancellationToken cancellationToken)
+    {
+        var runResult = await GetRazorGeneratorResultAsync(cancellationToken).ConfigureAwait(false);
+        if (runResult is null)
+        {
+            return null;
+        }
+
+        return runResult.GetFilePath(generatedDocumentHintName) is { } razorFilePath
+            ? runResult.GetCodeDocument(razorFilePath)
+            : null;
+    }
+
     private async Task<RazorGeneratorResult?> GetRazorGeneratorResultAsync(CancellationToken cancellationToken)
     {
         var result = await _project.GetSourceGeneratorRunResultAsync(cancellationToken).ConfigureAwait(false);
diff --git a/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/RemoteFilePathService.cs b/src/Razor/src/Microsoft.CodeAnalysis.Remote.Razor/RemoteFilePathService.cs
@@ -23,6 +23,6 @@ public override Uri GetRazorDocumentUri(Uri virtualDocumentUri)
 
     public override bool IsVirtualCSharpFile(Uri uri)
     {
-        return uri.Scheme == "source-generated";
+        return uri.Scheme == "roslyn-source-generated";
     }
 }
diff --git a/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CodeActions/ExtractToCodeBehindTests.cs b/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CodeActions/ExtractToCodeBehindTests.cs
@@ -1,7 +1,6 @@
 ﻿// Copyright (c) .NET Foundation. All rights reserved.
 // Licensed under the MIT license. See License.txt in the project root for license information.
 
-using System;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Razor.Test.Common;
 using Xunit.Abstractions;
@@ -11,7 +10,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost.CodeActions;
 
 public class ExtractToCodeBehindTests(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostCodeActionsEndpointTestBase(context, testOutputHelper)
 {
-    [FuseFact(Skip = "Need to map uri back to source generated document")]
+    [FuseFact]
     public async Task ExtractToCodeBehind()
     {
         await VerifyCodeActionAsync(
diff --git a/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostFindAllReferencesEndpointTest.cs b/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostFindAllReferencesEndpointTest.cs
@@ -17,7 +17,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
 
 public class CohostFindAllReferencesEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>
 {
-    [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseTheory]
     [CombinatorialData]
     public Task FindCSharpMember(bool supportsVSExtensions)
         => VerifyFindAllReferencesAsync("""
@@ -36,7 +36,7 @@ string M()
             """,
             supportsVSExtensions);
 
-    [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseTheory]
     [CombinatorialData]
     public async Task ComponentAttribute(bool supportsVSExtensions)
     {
@@ -60,7 +60,7 @@ await VerifyFindAllReferencesAsync(input, supportsVSExtensions,
             (FilePath("SurveyPrompt.razor"), surveyPrompt));
     }
 
-    [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseTheory]
     [CombinatorialData]
     public async Task OtherCSharpFile(bool supportsVSExtensions)
     {
diff --git a/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostGoToDefinitionEndpointTest.cs b/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostGoToDefinitionEndpointTest.cs
@@ -21,7 +21,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
 
 public class CohostGoToDefinitionEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>
 {
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task CSharp_Method()
     {
         var input = """
@@ -40,7 +40,7 @@ public async Task CSharp_Method()
         await VerifyGoToDefinitionAsync(input);
     }
 
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task CSharp_Local()
     {
         var input = """
@@ -87,7 +87,7 @@ public async Task CSharp_MetadataReference()
         Assert.Contains("public sealed class String", line);
     }
 
-    [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseTheory]
     [InlineData("$$IncrementCount")]
     [InlineData("In$$crementCount")]
     [InlineData("IncrementCount$$")]
@@ -107,7 +107,7 @@ public async Task Attribute_SameFile(string method)
         await VerifyGoToDefinitionAsync(input, FileKinds.Component);
     }
 
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task AttributeValue_BindAfter()
     {
         var input = """
@@ -157,7 +157,7 @@ public async Task Component()
         Assert.Equal(range, location.Range);
     }
 
-    [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseTheory]
     [InlineData("Ti$$tle")]
     [InlineData("$$@bind-Title")]
     [InlineData("@$$bind-Title")]
diff --git a/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostGoToImplementationEndpointTest.cs b/src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostGoToImplementationEndpointTest.cs
@@ -19,7 +19,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
 
 public class CohostGoToImplementationEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>
 {
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task CSharp_Method()
     {
         var input = """
@@ -40,7 +40,7 @@ public async Task CSharp_Method()
         await VerifyCSharpGoToImplementationAsync(input);
     }
 
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task CSharp_Field()
     {
         var input = """
@@ -64,7 +64,7 @@ string GetX()
         await VerifyCSharpGoToImplementationAsync(input);
     }
 
-    [FuseFact(Skip = "IFilePathService does not yet map generated documents")]
+    [FuseFact]
     public async Task CSharp_Multiple()
     {
         var input = """

Original file line number	Diff line number	Diff line change
`@@ -7,12 +7,14 @@`
`7`	`7`
`8`	`8`	`namespace Microsoft.NET.Sdk.Razor.SourceGenerators`
`9`	`9`	`{`
`10`		`- internal sealed class RazorGeneratorResult(IReadOnlyList<TagHelperDescriptor> tagHelpers, ImmutableDictionary<string, (string hintName, RazorCodeDocument document)> documents)`
	`10`	`+ internal sealed class RazorGeneratorResult(IReadOnlyList<TagHelperDescriptor> tagHelpers, ImmutableDictionary<string, (string hintName, RazorCodeDocument document)> filePathToDocument, ImmutableDictionary<string, string> hintNameToFilePath)`
`11`	`11`	`{`
`12`	`12`	`public IReadOnlyList<TagHelperDescriptor> TagHelpers => tagHelpers;`
`13`	`13`
`14`		`- public RazorCodeDocument? GetCodeDocument(string physicalPath) => documents.TryGetValue(physicalPath, out var pair) ? pair.document : null;`
	`14`	`+ public RazorCodeDocument? GetCodeDocument(string physicalPath) => filePathToDocument.TryGetValue(physicalPath, out var pair) ? pair.document : null;`
`15`	`15`
`16`		`- public string? GetHintName(string physicalPath) => documents.TryGetValue(physicalPath, out var pair) ? pair.hintName : null;`
	`16`	`+ public string? GetHintName(string physicalPath) => filePathToDocument.TryGetValue(physicalPath, out var pair) ? pair.hintName : null;`
	`17`	`+`
	`18`	`+ public string? GetFilePath(string hintName) => hintNameToFilePath.TryGetValue(hintName, out var filePath) ? filePath : null;`
`17`	`19`	`}`
`18`	`20`	`}`
Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ private async ValueTask<Response> GetCompletionAsync(`
`171`	`171`	`var generatedText = await generatedDocument.GetTextAsync(cancellationToken).ConfigureAwait(false);`
`172`	`172`	`var change = generatedText.GetTextChange(provisionalTextEdit);`
`173`	`173`	`generatedText = generatedText.WithChanges([change]);`
`174`		`- generatedDocument = generatedDocument.WithText(generatedText);`
	`174`	`+ generatedDocument = (SourceGeneratedDocument)generatedDocument.WithText(generatedText);`
`175`	`175`	`}`
`176`	`176`
`177`	`177`	`// This is, to say the least, not ideal. In future we're going to normalize on to Roslyn LSP types, and this can go.`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ internal sealed class RemoteDocumentSnapshot : IDocumentSnapshot`
`22`	`22`	`public RemoteProjectSnapshot ProjectSnapshot { get; }`
`23`	`23`
`24`	`24`	`private RazorCodeDocument? _codeDocument;`
`25`		`- private Document? _generatedDocument;`
	`25`	`+ private SourceGeneratedDocument? _generatedDocument;`
`26`	`26`
`27`	`27`	`public RemoteDocumentSnapshot(TextDocument textDocument, RemoteProjectSnapshot projectSnapshot)`
`28`	`28`	`{`
`@@ -103,7 +103,7 @@ public IDocumentSnapshot WithText(SourceText text)`
`103`	`103`	`return snapshotManager.GetSnapshot(newDocument);`
`104`	`104`	`}`
`105`	`105`
`106`		`- public async ValueTask<Document> GetGeneratedDocumentAsync(CancellationToken cancellationToken)`
	`106`	`+ public async ValueTask<SourceGeneratedDocument> GetGeneratedDocumentAsync(CancellationToken cancellationToken)`
`107`	`107`	`{`
`108`	`108`	`if (_generatedDocument is not null)`
`109`	`109`	`{`
Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,6 @@ public override Uri GetRazorDocumentUri(Uri virtualDocumentUri)`
`23`	`23`
`24`	`24`	`public override bool IsVirtualCSharpFile(Uri uri)`
`25`	`25`	`{`
`26`		`- return uri.Scheme == "source-generated";`
	`26`	`+ return uri.Scheme == "roslyn-source-generated";`
`27`	`27`	`}`
`28`	`28`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,6 @@`
`1`	`1`	`// Copyright (c) .NET Foundation. All rights reserved.`
`2`	`2`	`// Licensed under the MIT license. See License.txt in the project root for license information.`
`3`	`3`
`4`		`-using System;`
`5`	`4`	`using System.Threading.Tasks;`
`6`	`5`	`using Microsoft.AspNetCore.Razor.Test.Common;`
`7`	`6`	`using Xunit.Abstractions;`
`@@ -11,7 +10,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost.CodeActions;`
`11`	`10`
`12`	`11`	`public class ExtractToCodeBehindTests(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostCodeActionsEndpointTestBase(context, testOutputHelper)`
`13`	`12`	`{`
`14`		`- [FuseFact(Skip = "Need to map uri back to source generated document")]`
	`13`	`+ [FuseFact]`
`15`	`14`	`public async Task ExtractToCodeBehind()`
`16`	`15`	`{`
`17`	`16`	`await VerifyCodeActionAsync(`
Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;`
`17`	`17`
`18`	`18`	`public class CohostFindAllReferencesEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>`
`19`	`19`	`{`
`20`		`- [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]`
	`20`	`+ [FuseTheory]`
`21`	`21`	`[CombinatorialData]`
`22`	`22`	`public Task FindCSharpMember(bool supportsVSExtensions)`
`23`	`23`	`=> VerifyFindAllReferencesAsync("""`
`@@ -36,7 +36,7 @@ string M()`
`36`	`36`	`""",`
`37`	`37`	`supportsVSExtensions);`
`38`	`38`
`39`		`- [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]`
	`39`	`+ [FuseTheory]`
`40`	`40`	`[CombinatorialData]`
`41`	`41`	`public async Task ComponentAttribute(bool supportsVSExtensions)`
`42`	`42`	`{`
`@@ -60,7 +60,7 @@ await VerifyFindAllReferencesAsync(input, supportsVSExtensions,`
`60`	`60`	`(FilePath("SurveyPrompt.razor"), surveyPrompt));`
`61`	`61`	`}`
`62`	`62`
`63`		`- [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]`
	`63`	`+ [FuseTheory]`
`64`	`64`	`[CombinatorialData]`
`65`	`65`	`public async Task OtherCSharpFile(bool supportsVSExtensions)`
`66`	`66`	`{`
Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;`
`21`	`21`
`22`	`22`	`public class CohostGoToDefinitionEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>`
`23`	`23`	`{`
`24`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`24`	`+ [FuseFact]`
`25`	`25`	`public async Task CSharp_Method()`
`26`	`26`	`{`
`27`	`27`	`var input = """`
`@@ -40,7 +40,7 @@ public async Task CSharp_Method()`
`40`	`40`	`await VerifyGoToDefinitionAsync(input);`
`41`	`41`	`}`
`42`	`42`
`43`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`43`	`+ [FuseFact]`
`44`	`44`	`public async Task CSharp_Local()`
`45`	`45`	`{`
`46`	`46`	`var input = """`
`@@ -87,7 +87,7 @@ public async Task CSharp_MetadataReference()`
`87`	`87`	`Assert.Contains("public sealed class String", line);`
`88`	`88`	`}`
`89`	`89`
`90`		`- [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]`
	`90`	`+ [FuseTheory]`
`91`	`91`	`[InlineData("$$IncrementCount")]`
`92`	`92`	`[InlineData("In$$crementCount")]`
`93`	`93`	`[InlineData("IncrementCount$$")]`
`@@ -107,7 +107,7 @@ public async Task Attribute_SameFile(string method)`
`107`	`107`	`await VerifyGoToDefinitionAsync(input, FileKinds.Component);`
`108`	`108`	`}`
`109`	`109`
`110`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`110`	`+ [FuseFact]`
`111`	`111`	`public async Task AttributeValue_BindAfter()`
`112`	`112`	`{`
`113`	`113`	`var input = """`
`@@ -157,7 +157,7 @@ public async Task Component()`
`157`	`157`	`Assert.Equal(range, location.Range);`
`158`	`158`	`}`
`159`	`159`
`160`		`- [FuseTheory(Skip = "IFilePathService does not yet map generated documents")]`
	`160`	`+ [FuseTheory]`
`161`	`161`	`[InlineData("Ti$$tle")]`
`162`	`162`	`[InlineData("$$@bind-Title")]`
`163`	`163`	`[InlineData("@$$bind-Title")]`
Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;`
`19`	`19`
`20`	`20`	`public class CohostGoToImplementationEndpointTest(FuseTestContext context, ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper), IClassFixture<FuseTestContext>`
`21`	`21`	`{`
`22`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`22`	`+ [FuseFact]`
`23`	`23`	`public async Task CSharp_Method()`
`24`	`24`	`{`
`25`	`25`	`var input = """`
`@@ -40,7 +40,7 @@ public async Task CSharp_Method()`
`40`	`40`	`await VerifyCSharpGoToImplementationAsync(input);`
`41`	`41`	`}`
`42`	`42`
`43`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`43`	`+ [FuseFact]`
`44`	`44`	`public async Task CSharp_Field()`
`45`	`45`	`{`
`46`	`46`	`var input = """`
`@@ -64,7 +64,7 @@ string GetX()`
`64`	`64`	`await VerifyCSharpGoToImplementationAsync(input);`
`65`	`65`	`}`
`66`	`66`
`67`		`- [FuseFact(Skip = "IFilePathService does not yet map generated documents")]`
	`67`	`+ [FuseFact]`
`68`	`68`	`public async Task CSharp_Multiple()`
`69`	`69`	`{`
`70`	`70`	`var input = """`