如何对Mongo ClusterBuilder的事件订阅进行单元测试？

嘿，这个问题我之前也碰到过，要确保Mongo集群事件的订阅一个都没少，没人不小心删了关键代码，单元测试可以从这两个实用的角度入手：

方法一：直接验证ClusterSettings中的订阅者 #

这种方法最直接，不需要额外抽象，直接检查配置后的集群设置里有没有我们需要的事件订阅。

假设你的事件订阅逻辑是封装在一个方法里的，比如：

public static ClusterBuilder SetupClusterEvents(ClusterBuilder builder)
{
    builder.Subscribe<CommandSucceededEvent>(ev => LogEvent(ev));
    builder.Subscribe<CommandFailedEvent>(ev => LogEvent(ev));
    builder.Subscribe<WhateverEvent>(ev => LogEvent(ev));
    return builder;
}

private static void LogEvent<T>(T ev) where T : ClusterEvent
{
    // 你的日志或处理逻辑
}

对应的单元测试可以这么写：

[Test]
public void AllRequiredClusterEvents_AreSubscribed()
{
    // 准备工作：创建ClusterBuilder实例
    var clusterBuilder = new ClusterBuilder();
    
    // 执行配置方法
    YourClass.SetupClusterEvents(clusterBuilder);
    var clusterSettings = clusterBuilder.Build();
    
    // 定义需要验证的事件类型列表
    var requiredEventTypes = new[]
    {
        typeof(CommandSucceededEvent),
        typeof(CommandFailedEvent),
        typeof(WhateverEvent)
    };
    
    // 逐个检查每个事件是否有对应的订阅者
    foreach (var eventType in requiredEventTypes)
    {
        var hasMatchingSubscriber = clusterSettings.Subscribers.Any(subscriber =>
        {
            // 调用订阅者的GetEventHandler方法，检查是否能处理当前事件类型
            var getHandlerMethod = subscriber.GetType()
                .GetMethod("GetEventHandler")?
                .MakeGenericMethod(eventType);
                
            return getHandlerMethod?.Invoke(subscriber, null) != null;
        });
        
        Assert.IsTrue(hasMatchingSubscriber, $"事件类型 {eventType.Name} 未被正确订阅！");
    }
}

这个测试的核心是通过反射检查每个订阅者是否能处理目标事件，确保所有需要的订阅都存在。

方法二：用抽象层+Mock简化测试（适合复杂场景） #

如果你的订阅逻辑比较复杂，或者想和Mongo驱动的具体实现解耦，可以把订阅逻辑抽象成一个接口，这样测试时用Mock框架就能轻松验证方法调用。

首先定义一个抽象的订阅服务接口：

public interface IClusterEventSubscriber
{
    void SubscribeToCommandSucceeded(Action<CommandSucceededEvent> handler);
    void SubscribeToCommandFailed(Action<CommandFailedEvent> handler);
    void SubscribeToWhatever(Action<WhateverEvent> handler);
}

然后实现这个接口，内部调用ClusterBuilder的订阅方法：

public class MongoClusterEventSubscriber : IClusterEventSubscriber
{
    private readonly ClusterBuilder _clusterBuilder;
    
    public MongoClusterEventSubscriber(ClusterBuilder clusterBuilder)
    {
        _clusterBuilder = clusterBuilder;
    }
    
    public void SubscribeToCommandSucceeded(Action<CommandSucceededEvent> handler)
    {
        _clusterBuilder.Subscribe(handler);
    }
    
    public void SubscribeToCommandFailed(Action<CommandFailedEvent> handler)
    {
        _clusterBuilder.Subscribe(handler);
    }
    
    public void SubscribeToWhatever(Action<WhateverEvent> handler)
    {
        _clusterBuilder.Subscribe(handler);
    }
}

现在你的业务配置代码就可以依赖这个接口，而不是直接操作ClusterBuilder：

public class ClusterConfigurator
{
    private readonly IClusterEventSubscriber _eventSubscriber;
    
    public ClusterConfigurator(IClusterEventSubscriber eventSubscriber)
    {
        _eventSubscriber = eventSubscriber;
    }
    
    public void ConfigureEvents()
    {
        _eventSubscriber.SubscribeToCommandSucceeded(ev => LogEvent(ev));
        _eventSubscriber.SubscribeToCommandFailed(ev => LogEvent(ev));
        _eventSubscriber.SubscribeToWhatever(ev => LogEvent(ev));
    }
    
    private void LogEvent<T>(T ev) where T : ClusterEvent
    {
        // 你的处理逻辑
    }
}

对应的单元测试用Moq来验证：

[Test]
public void EventSubscriptionMethods_AreAllCalled()
{
    // 准备Mock的订阅服务
    var mockSubscriber = new Mock<IClusterEventSubscriber>();
    var configurator = new ClusterConfigurator(mockSubscriber.Object);
    
    // 执行配置
    configurator.ConfigureEvents();
    
    // 验证每个订阅方法都被调用了一次
    mockSubscriber.Verify(s => s.SubscribeToCommandSucceeded(It.IsAny<Action<CommandSucceededEvent>>()), Times.Once);
    mockSubscriber.Verify(s => s.SubscribeToCommandFailed(It.IsAny<Action<CommandFailedEvent>>()), Times.Once);
    mockSubscriber.Verify(s => s.SubscribeToWhatever(It.IsAny<Action<WhateverEvent>>()), Times.Once);
}

这种方式的好处是测试不依赖Mongo驱动的内部细节，逻辑更清晰，也更容易扩展。

备注：内容来源于stack exchange，提问作者Leonardo