Updated test

Author: david-driscoll

test/Lsp.Tests/HandlerResolverTests.cs

@@ -13,6 +13,7 @@
 using HandlerCollection = OmniSharp.Extensions.LanguageServer.HandlerCollection;
 using OmniSharp.Extensions.LanguageServer.Models;
 using OmniSharp.Extensions.LanguageServer.Abstractions;
+using OmniSharp.Extensions.LanguageServer.Handlers;
 
 namespace Lsp.Tests
 {
@@ -21,19 +22,35 @@ public class HandlerResolverTests
         [Theory]
         [InlineData(typeof(IInitializeHandler), "initialize", 1)]
         [InlineData(typeof(IInitializedHandler), "initialized", 1)]
+        [InlineData(typeof(IShutdownHandler), "shutdown", 1)]
+        [InlineData(typeof(IExitHandler), "exit", 1)]
         public void Should_Contain_AllDefinedMethods(Type requestHandler, string key, int count)
         {
             var handler = new HandlerCollection();
             var sub = (IJsonRpcHandler)Substitute.For(new Type[] { requestHandler }, new object[0]);
-            if (sub is IRegistration<TextDocumentRegistrationOptions> reg)
-                reg.GetRegistrationOptions()
-                    .Returns(new TextDocumentRegistrationOptions());
 
             handler.Add(sub);
             handler._handlers.Should().Contain(x => x.Method == key);
             handler._handlers.Count.Should().Be(count);
         }
 
+        [Fact]
+        public void Should_Contain_AllConcreteDefinedMethods()
+        {
+            var handler = new HandlerCollection();
+
+            handler.Add(
+                Substitute.For<IExitHandler>(),
+                Substitute.For<IInitializeHandler>(),
+                Substitute.For<IInitializedHandler>(),
+                Substitute.For<IShutdownHandler>()
+            );
+
+            handler._handlers.Should().Contain(x => x.Method == "exit");
+            handler._handlers.Should().Contain(x => x.Method == "shutdown");
+            handler._handlers.Count.Should().Be(4);
+        }
+
         [Theory]
         [InlineData("textDocument/didOpen", 4)]
         [InlineData("textDocument/didChange", 4)]
@@ -151,22 +168,5 @@ public void Should_Contain_AllDefinedMethods_OnLanguageServer_WithDifferentKeys(
             handler._handlers.Should().Contain(x => x.Method == key2);
             handler._handlers.Count.Should().Be(count);
         }
-
-        [Fact]
-        public void Should_BeAwesome()
-        {
-            var handler = new HandlerCollection();
-
-            handler.Add(
-                Substitute.For<IExitHandler>(),
-                Substitute.For<IInitializeHandler>(),
-                Substitute.For<IInitializedHandler>(),
-                Substitute.For<IShutdownHandler>()
-            );
-
-            handler._handlers.Should().Contain(x => x.Method == "exit");
-            handler._handlers.Should().Contain(x => x.Method == "shutdown");
-            handler._handlers.Count.Should().Be(4);
-        }
     }
 }