如何为工作单元模式编写Mock？求指导GetAllAsync方法Mock实现

现有代码的局限性 #

你当前的Mock代码存在以下问题：

• 参数引用匹配问题：includes1是一个具体的List实例，只有当被测试代码调用GetAllAsync时传入的includes参数与includes1是同一个对象引用时，Mock才会生效。如果被测试代码创建了新的List（哪怕内容完全一致），Mock将不会触发。
• 表达式硬编码：r => r.CompanyId == companyId是一个具体的表达式实例，若被测试代码使用了等价但不同引用的表达式，同样无法匹配到Mock。

优化后的Mock方案 #

1. 匹配任意参数（最简方式） #

如果不需要针对特定参数做精准校验，直接使用It.IsAny<>匹配所有参数：

_unitOfWork.Setup(uow => uow.Order.GetAllAsync(
    It.IsAny<Expression<Func<Order, bool>>>(),
    It.IsAny<Func<IQueryable<Order>, IOrderedQueryable<Order>>>(),
    It.IsAny<List<Expression<Func<Order, object>>>>(),
    It.IsAny<bool>()
)).ReturnsAsync(GetAllOrders());

2. 精准匹配参数逻辑 #

如果需要校验参数的逻辑（而非引用），可以通过It.Is<>结合自定义判断实现：

匹配指定条件的Predicate

添加一个辅助方法判断表达式逻辑：

private bool MatchesCompanyIdPredicate(Expression<Func<Order, bool>> predicate, int targetCompanyId)
{
    // 解析表达式，判断是否为 r => r.CompanyId == targetCompanyId
    if (predicate.Body is BinaryExpression binaryExpr && binaryExpr.NodeType == ExpressionType.Equal)
    {
        if (binaryExpr.Left is MemberExpression leftMember && leftMember.Member.Name == nameof(Order.CompanyId))
        {
            if (binaryExpr.Right is ConstantExpression rightConst && (int)rightConst.Value == targetCompanyId)
            {
                return true;
            }
        }
    }
    return false;
}

然后在Setup中使用该方法匹配Predicate：

_unitOfWork.Setup(uow => uow.Order.GetAllAsync(
    It.Is<Expression<Func<Order, bool>>>(p => MatchesCompanyIdPredicate(p, companyId)),
    It.IsAny<Func<IQueryable<Order>, IOrderedQueryable<Order>>>(),
    It.Is<List<Expression<Func<Order, object>>>>(includes => 
        includes?.Any(expr => expr.ToString() == "x => x.OrderDetails") ?? false),
    It.IsAny<bool>()
)).ReturnsAsync(GetAllOrders());

匹配包含指定Include的列表

上面的代码中，通过expr.ToString()判断Include表达式是否匹配，这是一种简单的方式；如果需要更严谨的表达式比较，可以编写表达式树相等校验的辅助方法。

3. 动态返回结果（根据参数） #

如果需要根据传入的参数返回不同结果，可以使用ReturnsAsync的重载：

_unitOfWork.Setup(uow => uow.Order.GetAllAsync(
    It.IsAny<Expression<Func<Order, bool>>>(),
    It.IsAny<Func<IQueryable<Order>, IOrderedQueryable<Order>>>(),
    It.IsAny<List<Expression<Func<Order, object>>>>(),
    It.IsAny<bool>()
)).ReturnsAsync((
    Expression<Func<Order, bool>> predicate,
    Func<IQueryable<Order>, IOrderedQueryable<Order>> orderBy,
    List<Expression<Func<Order, object>>> includes,
    bool disableTracking) => 
{
    // 根据参数动态生成返回结果
    var orders = GetAllOrders();
    if (predicate != null)
    {
        orders = orders.AsQueryable().Where(predicate).ToList();
    }
    // 按需处理orderBy等逻辑
    return Task.FromResult(orders);
});

注意事项 #

• 确保GetAllOrders()返回的是List<Order>类型，Mock时使用ReturnsAsync包装（原方法返回Task<T>时必须这么做）。
• 对于泛型方法的Mock，Moq会自动处理类型参数，只要指定了具体的实体类型（如Order），就可以正常匹配。

内容的提问来源于stack exchange，提问作者Dhivya Bharathi