Test updates

dotnet · Jul 30, 2024 · e0db8d1 · e0db8d1
1 parent 10095a5
commit e0db8d1
Showing 1 changed file with 21 additions and 6 deletions.
diff --git a/...azor.LanguageServer.Test/CodeActions/Razor/ExtractToNewComponentCodeActionProviderTest.cs b/...azor.LanguageServer.Test/CodeActions/Razor/ExtractToNewComponentCodeActionProviderTest.cs
@@ -2,10 +2,6 @@
 // Licensed under the MIT license. See License.txt in the project root for license information.
 
 using System;
-using System.Collections.Generic;
-using System.Collections.Immutable;
-using System.Linq;
-using System.Text;
 using System.Text.Json;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Razor.Language;
@@ -179,8 +175,8 @@ public async Task Handle_MultiPointSelectionWithEndAfterElement_ReturnsCurrentEl
             <PageTitle>Home</PageTitle>
 
             <div id="parent">
-                [|<div>
-                    $$<h1>Div a title</h1>
+                [|$$<div>
+                    <h1>Div a title</h1>
                     <p>Div a par</p>
                 </div>
                 <div>
@@ -205,6 +201,8 @@ public async Task Handle_MultiPointSelectionWithEndAfterElement_ReturnsCurrentEl
         var location = new SourceLocation(cursorPosition, -1, -1);
         var context = CreateRazorCodeActionContext(request, location, documentPath, contents);
 
+        AddMultiPointSelectionToContext(ref context, selectionSpan);
+
         var provider = new ExtractToNewComponentCodeActionProvider(LoggerFactory);
 
         // Act
@@ -250,4 +248,21 @@ private static RazorCodeActionContext CreateRazorCodeActionContext(VSCodeActionP
 
         return context;
     }
+
+
+    private static void AddMultiPointSelectionToContext(ref RazorCodeActionContext context, TextSpan selectionSpan)
+    {
+        var sourceText = context.CodeDocument.GetSourceText();
+        var startLinePosition = sourceText.Lines.GetLinePosition(selectionSpan.Start);
+        var startPosition = new Position(startLinePosition.Line, startLinePosition.Character);
+
+        var endLinePosition = sourceText.Lines.GetLinePosition(selectionSpan.End);
+        var endPosition = new Position(endLinePosition.Line, endLinePosition.Character);
+
+        context.Request.Range = new Range
+        {
+            Start = startPosition,
+            End = endPosition
+        };
+    }
 }