Share C# and Razor diagnostic tests

We don't support Html diagnostics in VS Code

Author: davidwengier

src/Razor/src/Microsoft.VisualStudioCode.RazorExtension/Endpoints/DocumentPullDiagnosticsEndpoint.cs

@@ -77,5 +77,13 @@ public ImmutableArray<Registration> GetRegistrations(VSInternalClientCapabilitie
             ResultId = Guid.NewGuid().ToString()
         };
     }
+
+    internal TestAccessor GetTestAccessor() => new(this);
+
+    internal readonly struct TestAccessor(DocumentPullDiagnosticsEndpoint instance)
+    {
+        public Task<LspDiagnostic[]?> HandleRequestAsync(TextDocument razorDocument, CancellationToken cancellationToken)
+            => instance.GetDiagnosticsAsync(razorDocument, cancellationToken);
+    }
 }
 

src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostDocumentPullDiagnosticsTest.cs

@@ -1,7 +1,6 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
-using System.Collections.Generic;
 using System.Linq;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Razor.LanguageServer.Test;
@@ -12,68 +11,11 @@
 using Microsoft.VisualStudio.Razor.Settings;
 using Roslyn.Test.Utilities;
 using Xunit;
-using Xunit.Abstractions;
 
 namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
 
-public class CohostDocumentPullDiagnosticsTest(ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper)
+public partial class CohostDocumentPullDiagnosticsTest
 {
-    [Fact]
-    public Task NoDiagnostics()
-        => VerifyDiagnosticsAsync("""
-            <div></div>
-
-            @code
-            {
-                public void IJustMetYou()
-                {
-                }
-            }
-            """);
-
-    [Fact]
-    public Task CSharp()
-        => VerifyDiagnosticsAsync("""
-            <div></div>
-
-            @code
-            {
-                public void IJustMetYou()
-                {
-                    {|CS0103:CallMeMaybe|}();
-                }
-            }
-            """);
-
-    [Fact]
-    public Task Razor()
-        => VerifyDiagnosticsAsync("""
-            <div>
-
-            {|RZ10012:<NonExistentComponent />|}
-
-            </div>
-            """);
-
-    [Fact]
-    public Task CSharpAndRazor_MiscellaneousFile()
-        => VerifyDiagnosticsAsync("""
-            <div>
-
-            {|RZ10012:<NonExistentComponent />|}
-
-            </div>
-
-            @code
-            {
-                public void IJustMetYou()
-                {
-                    {|CS0103:CallMeMaybe|}();
-                }
-            }
-            """,
-            miscellaneousFile: true);
-
     [Fact]
     public Task Html()
     {
@@ -449,34 +391,6 @@ Task Send() =>
             }]);
     }
 
-    [Fact]
-    public Task CombinedAndNestedDiagnostics()
-        => VerifyDiagnosticsAsync("""
-            @using System.Threading.Tasks;
-
-            <div>
-
-            {|RZ10012:<NonExistentComponent />|}
-
-            @code
-            {
-                public void IJustMetYou()
-                {
-                    {|CS0103:CallMeMaybe|}();
-                }
-            }
-
-            <div>
-                @{
-                    {|CS4033:await Task.{|CS1501:Delay|}()|};
-                }
-
-                {|RZ9980:<p>|}
-            </div>
-
-            </div>
-            """);
-
     [Fact]
     public Task TODOComments()
         => VerifyDiagnosticsAsync("""

src/Razor/test/Microsoft.VisualStudioCode.RazorExtension.Test/Endpoints/CohostDocumentPullDiagnosticsTest.cs

@@ -0,0 +1,44 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Linq;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Razor.Test.Common;
+using Microsoft.CodeAnalysis.Razor.Telemetry;
+using Microsoft.CodeAnalysis.Text;
+using Roslyn.Test.Utilities;
+using Xunit;
+
+namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
+
+public partial class CohostDocumentPullDiagnosticsTest
+{
+    private async Task VerifyDiagnosticsAsync(TestCode input, VSInternalDiagnosticReport[]? htmlResponse = null, bool miscellaneousFile = false)
+    {
+        var document = CreateProjectAndRazorDocument(input.Text, miscellaneousFile: miscellaneousFile);
+        var inputText = await document.GetTextAsync(DisposalToken);
+
+        var requestInvoker = new TestHtmlRequestInvoker([(VSInternalMethods.DocumentPullDiagnosticName, htmlResponse)]);
+
+        var endpoint = new DocumentPullDiagnosticsEndpoint(IncompatibleProjectService, RemoteServiceInvoker, requestInvoker, ClientCapabilitiesService, NoOpTelemetryReporter.Instance, LoggerFactory);
+
+        var result = await endpoint.GetTestAccessor().HandleRequestAsync(document, DisposalToken);
+
+        Assert.NotNull(result);
+
+        var markers = result.SelectMany(d =>
+            new[] {
+                (index: inputText.GetTextSpan(d.Range).Start, text: $"{{|{d.Code!.Value.Second}:"),
+                (index: inputText.GetTextSpan(d.Range).End, text:"|}")
+            });
+
+        var testOutput = input.Text;
+        // Ordering by text last means start tags get sorted before end tags, for zero width ranges
+        foreach (var (index, text) in markers.OrderByDescending(i => i.index).ThenByDescending(i => i.text))
+        {
+            testOutput = testOutput.Insert(index, text);
+        }
+
+        AssertEx.EqualOrDiff(input.OriginalInput, testOutput);
+    }
+}

src/Razor/test/Microsoft.VisualStudioCode.RazorExtension.Test/Endpoints/Shared/CohostDocumentPullDiagnosticsTest.cs

@@ -0,0 +1,95 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Threading.Tasks;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
+
+public partial class CohostDocumentPullDiagnosticsTest(ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper)
+{
+    [Fact]
+    public Task NoDiagnostics()
+        => VerifyDiagnosticsAsync("""
+            <div></div>
+
+            @code
+            {
+                public void IJustMetYou()
+                {
+                }
+            }
+            """);
+
+    [Fact]
+    public Task CSharp()
+        => VerifyDiagnosticsAsync("""
+            <div></div>
+
+            @code
+            {
+                public void IJustMetYou()
+                {
+                    {|CS0103:CallMeMaybe|}();
+                }
+            }
+            """);
+
+    [Fact]
+    public Task Razor()
+        => VerifyDiagnosticsAsync("""
+            <div>
+
+            {|RZ10012:<NonExistentComponent />|}
+
+            </div>
+            """);
+
+    [Fact]
+    public Task CSharpAndRazor_MiscellaneousFile()
+        => VerifyDiagnosticsAsync("""
+            <div>
+
+            {|RZ10012:<NonExistentComponent />|}
+
+            </div>
+
+            @code
+            {
+                public void IJustMetYou()
+                {
+                    {|CS0103:CallMeMaybe|}();
+                }
+            }
+            """,
+            miscellaneousFile: true);
+
+    [Fact]
+    public Task CombinedAndNestedDiagnostics()
+        => VerifyDiagnosticsAsync("""
+            @using System.Threading.Tasks;
+
+            <div>
+
+            {|RZ10012:<NonExistentComponent />|}
+
+            @code
+            {
+                public void IJustMetYou()
+                {
+                    {|CS0103:CallMeMaybe|}();
+                }
+            }
+
+            <div>
+                @{
+                    {|CS4033:await Task.{|CS1501:Delay|}()|};
+                }
+
+                {|RZ9980:<p>|}
+            </div>
+
+            </div>
+            """);
+}