Fix tests

Author: ejsmith

src/Foundatio.Mediator.Abstractions/Mediator.cs

@@ -194,10 +194,7 @@ private PublishAsyncDelegate[] GetAllApplicableHandlers(object message)
     [DebuggerStepThrough]
     private IEnumerable<HandlerRegistration> GetHandlersForType(Type type)
     {
-        var handlers = _serviceProvider.GetKeyedServices<HandlerRegistration>(MessageTypeKey.Get(type));
-        if (handlers != null && handlers.Any()) return handlers;
-
-        return Array.Empty<HandlerRegistration>();
+        return _serviceProvider.GetKeyedServices<HandlerRegistration>(MessageTypeKey.Get(type)).ToArray();
     }
 
     private static readonly ConcurrentDictionary<Type, object> _middlewareCache = new();

src/Foundatio.Mediator.Abstractions/MediatorExtensions.cs

@@ -65,7 +65,7 @@ public static IServiceCollection AddMediator(this IServiceCollection services, A
     /// <returns></returns>
     public static IServiceCollection AddHandler(this IServiceCollection services, HandlerRegistration registration)
     {
-        services.AddKeyedSingleton<HandlerRegistration>(registration.MessageTypeName);
+        services.AddKeyedSingleton(registration.MessageTypeName, registration);
         services.AddSingleton(registration);
         return services;
     }

tests/Foundatio.Mediator.Tests/BasicHandlerGenerationTests.GeneratesWrapperForSimpleHandler.verified.txt

@@ -164,9 +164,9 @@ public static class Tests_MediatorHandlers
         // Register HandlerRegistration instances keyed by message type name
         // Optionally register handler classes into DI based on MediatorHandlerLifetime setting
 
-        services.AddKeyedSingleton<HandlerRegistration>(MessageTypeKey.Get(typeof(Ping)),
-            new HandlerRegistration(
+        services.AddHandler(new HandlerRegistration(
                 MessageTypeKey.Get(typeof(Ping)),
+                "PingHandler_Ping_Handler",
                 PingHandler_Ping_Handler.UntypedHandleAsync,
                 null,
                 true));

tests/Foundatio.Mediator.Tests/DIRegistrationTests.cs

@@ -17,8 +17,8 @@ public class BHandler { public void Handle(B m) { } }
 
         var (_, _, trees) = RunGenerator(src, [ new MediatorGenerator() ]);
         var di = trees.First(t => t.HintName.EndsWith("_MediatorHandlers.g.cs"));
-        Assert.Contains("AddKeyedSingleton<HandlerRegistration>(MessageTypeKey.Get(typeof(A))", di.Source);
-        Assert.Contains("AddKeyedSingleton<HandlerRegistration>(MessageTypeKey.Get(typeof(B))", di.Source);
+        Assert.Contains("MessageTypeKey.Get(typeof(A))", di.Source);
+        Assert.Contains("MessageTypeKey.Get(typeof(B))", di.Source);
         Assert.Contains("UntypedHandleAsync", di.Source);
         Assert.Contains("UntypedHandle(", di.Source);
     }