Added a few assertions for prepare rename (#413)

Author: david-driscoll

test/Lsp.Tests/Integration/RenameTests.cs

@@ -4,10 +4,12 @@
 using System.Threading.Tasks;
 using FluentAssertions;
 using NSubstitute;
+using NSubstitute.Callbacks;
 using OmniSharp.Extensions.JsonRpc.Testing;
 using OmniSharp.Extensions.LanguageProtocol.Testing;
 using OmniSharp.Extensions.LanguageServer.Client;
 using OmniSharp.Extensions.LanguageServer.Protocol;
+using OmniSharp.Extensions.LanguageServer.Protocol.Client.Capabilities;
 using OmniSharp.Extensions.LanguageServer.Protocol.Document;
 using OmniSharp.Extensions.LanguageServer.Protocol.Models;
 using OmniSharp.Extensions.LanguageServer.Server;
@@ -20,19 +22,19 @@ namespace Lsp.Tests.Integration
 {
     public class RenameTests : LanguageProtocolTestBase
     {
-        private readonly Func<PrepareRenameParams, CancellationToken, Task<RangeOrPlaceholderRange?>> _prepareRename;
-        private readonly Func<RenameParams, CancellationToken, Task<WorkspaceEdit?>> _rename;
+        private readonly Func<PrepareRenameParams, RenameCapability, CancellationToken, Task<RangeOrPlaceholderRange?>> _prepareRename;
+        private readonly Func<RenameParams, RenameCapability, CancellationToken, Task<WorkspaceEdit?>> _rename;
 
         public RenameTests(ITestOutputHelper outputHelper) : base(new JsonRpcTestOptions().ConfigureForXUnit(outputHelper, LogEventLevel.Verbose))
         {
-            _prepareRename = Substitute.For<Func<PrepareRenameParams, CancellationToken, Task<RangeOrPlaceholderRange?>>>();
-            _rename = Substitute.For<Func<RenameParams, CancellationToken, Task<WorkspaceEdit?>>>();
+            _prepareRename = Substitute.For<Func<PrepareRenameParams, RenameCapability, CancellationToken, Task<RangeOrPlaceholderRange?>>>();
+            _rename = Substitute.For<Func<RenameParams, RenameCapability, CancellationToken, Task<WorkspaceEdit?>>>();
         }
 
         [Fact]
         public async Task Should_Handle_Rename_With_No_Value()
         {
-            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<CancellationToken>())
+            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                           .Returns(
                                call => {
                                    var pos = call.Arg<PrepareRenameParams>().Position;
@@ -45,7 +47,7 @@ public async Task Should_Handle_Rename_With_No_Value()
                                }
                            );
 
-            _rename.Invoke(Arg.Any<RenameParams>(), Arg.Any<CancellationToken>())
+            _rename.Invoke(Arg.Any<RenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                    .Returns(
                         new WorkspaceEdit() {
                             DocumentChanges = new Container<WorkspaceEditDocumentChange>(new WorkspaceEditDocumentChange(new CreateFile() {
@@ -76,12 +78,19 @@ public async Task Should_Handle_Rename_With_No_Value()
 
             renameResponse!.DocumentChanges.Should().HaveCount(1);
             renameResponse.DocumentChanges.Should().Match(z => z.Any(x => x.IsCreateFile));
+
+            // Ensure capability was provided to both sides as needed
+            _prepareRename.Received(1).Invoke(Arg.Any<PrepareRenameParams>(), Arg.Is<RenameCapability>(z => z.PrepareSupport), Arg.Any<CancellationToken>());
+            _rename.Received(1).Invoke(Arg.Any<RenameParams>(), Arg.Is<RenameCapability>(z => z.PrepareSupport), Arg.Any<CancellationToken>());
+            var capability1 = _prepareRename.ReceivedCalls().Select(z => z.GetArguments()[1] as RenameCapability).FirstOrDefault();
+            var capability2 = _rename.ReceivedCalls().Select(z => z.GetArguments()[1] as RenameCapability).FirstOrDefault();
+            capability1.Should().BeSameAs(capability2);
         }
 
         [Fact]
         public async Task Should_Handle_Prepare_Rename_With_No_Value()
         {
-            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<CancellationToken>())
+            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                           .Returns(Task.FromResult<RangeOrPlaceholderRange?>(null)!);
             var (client, _) = await Initialize(ClientOptionsAction, ServerOptionsAction);
 
@@ -99,7 +108,7 @@ public async Task Should_Handle_Prepare_Rename_With_No_Value()
         [Fact]
         public async Task Should_Handle_Prepare_Rename_With_Range()
         {
-            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<CancellationToken>())
+            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                           .Returns(
                                call => {
                                    var pos = call.Arg<PrepareRenameParams>().Position;
@@ -128,7 +137,7 @@ public async Task Should_Handle_Prepare_Rename_With_Range()
         [Fact]
         public async Task Should_Handle_Prepare_Rename_With_PlaceholderRange()
         {
-            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<CancellationToken>())
+            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                           .Returns(
                                call => {
                                    var pos = call.Arg<PrepareRenameParams>().Position;
@@ -161,7 +170,7 @@ public async Task Should_Handle_Prepare_Rename_With_PlaceholderRange()
         [Fact]
         public async Task Should_Handle_Prepare_Rename_With_DefaultBehavior()
         {
-            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<CancellationToken>())
+            _prepareRename.Invoke(Arg.Any<PrepareRenameParams>(), Arg.Any<RenameCapability>(), Arg.Any<CancellationToken>())
                           .Returns(
                                call => new RangeOrPlaceholderRange(
                                    new RenameDefaultBehavior() {
@@ -193,13 +202,19 @@ private void ServerOptionsAction(LanguageServerOptions obj)
             obj.OnRename(
                 _rename, new RenameRegistrationOptions() {
                     DocumentSelector = DocumentSelector.ForLanguage("csharp"),
-                    PrepareProvider = true
+                    PrepareProvider = true,
                 }
             );
         }
 
         private void ClientOptionsAction(LanguageClientOptions obj)
         {
+            obj.WithCapability(
+                new RenameCapability() {
+                    PrepareSupport = true,
+                    PrepareSupportDefaultBehavior = true
+                }
+            );
         }
     }
 }