如何简化Moq框架中多参数异步方法的Setup调用？

我使用Moq模拟异步方法时，原本的写法如下：

_mockClass.Setup(d => d.TestAsync(It.IsAny<string>(), It.IsAny<long>()))
    .ReturnsAsync(1);

由于TestAsync有10多个参数，每次重复编写It.IsAny<>过于繁琐，希望简化为：

Setup(_mockClass.TestAsync, 1)

我尝试过以下伪代码，但不知道如何通过反射自动填充所有参数为It.IsAny<>：

// 伪代码
public void Setup(delegate del, object returnVal)
{
    _mockClass.Setup(d => d.del(It.IsAny<string>(), It.IsAny<long>()))// 如何用反射自动生成所有It.IsAny？
    .ReturnsAsync(returnVal);
}

最终目标是实现类似mock.Setup(d => d.TestAsync).ReturnsAsync(1)的简洁调用。

解决方案 #

核心思路 #

通过反射获取目标方法的参数类型，自动生成对应数量的It.IsAny<T>()表达式，再动态构建Moq所需的Setup表达式，以此避免手动编写大量参数匹配器。同时针对异步方法的Task<T>返回类型做适配。

具体实现 #

可以编写一个泛型扩展方法，结合表达式树与反射完成逻辑：

using Moq;
using System;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;

public static class MockExtensions
{
    // 适配返回Task<TResult>的异步方法
    public static void SetupAsync<TMock, TResult>(this Mock<TMock> mock,
        Func<TMock, Func<object[], Task<TResult>>> methodSelector, TResult returnValue)
        where TMock : class
    {
        // 获取目标方法的MethodInfo
        var method = methodSelector(mock.Object).Method.GetGenericMethodDefinition();
        var parameterTypes = method.GetParameters().Select(p => p.ParameterType).ToArray();

        // 构建所有参数的It.IsAny<T>()表达式
        var paramMatchers = parameterTypes.Select(type =>
            Expression.Call(typeof(It), nameof(It.IsAny), new[] { type })).ToArray();

        // 构建mock对象的参数表达式（即Setup中的d => ...里的d）
        var mockParam = Expression.Parameter(typeof(TMock), "d");

        // 构建方法调用表达式：d.TestAsync(It.IsAny<...>, ...)
        var methodCallExpr = Expression.Call(mockParam, method, paramMatchers);

        // 包装成Lambda表达式供Moq Setup使用
        var setupExpr = Expression.Lambda<Expression<Func<TMock, Task<TResult>>>>(methodCallExpr, mockParam);

        // 执行Setup并设置异步返回值
        mock.Setup(setupExpr.Compile()).ReturnsAsync(returnValue);
    }

    // 适配无返回值的异步方法（Task类型）
    public static void SetupAsync<TMock>(this Mock<TMock> mock,
        Func<TMock, Func<object[], Task>> methodSelector)
        where TMock : class
    {
        var method = methodSelector(mock.Object).Method.GetGenericMethodDefinition();
        var parameterTypes = method.GetParameters().Select(p => p.ParameterType).ToArray();

        var paramMatchers = parameterTypes.Select(type =>
            Expression.Call(typeof(It), nameof(It.IsAny), new[] { type })).ToArray();

        var mockParam = Expression.Parameter(typeof(TMock), "d");
        var methodCallExpr = Expression.Call(mockParam, method, paramMatchers);
        var setupExpr = Expression.Lambda<Expression<Func<TMock, Task>>>(methodCallExpr, mockParam);

        mock.Setup(setupExpr.Compile()).Returns(Task.CompletedTask);
    }
}

使用示例 #

对于有返回值的异步方法：

_mockClass.SetupAsync(m => m.TestAsync, 1);

对于无返回值的异步方法：

_mockClass.SetupAsync(m => m.SomeVoidAsyncMethod);

说明 #

• 该实现自动处理任意数量的参数，无需手动编写每个It.IsAny<>；
• 通过两个重载分别支持Task<T>和Task类型的异步方法；
• 方法选择器m => m.TestAsync会自动关联到目标方法，反射解析参数类型后生成匹配器。

内容的提问来源于stack exchange，提问作者Nitin S