如何Mock IOrderedQueryable与IDocumentQuery并应用过滤表达式？

嘿，我刚好碰到过类似的Mock DocumentDB查询的问题，给你分享一下我是怎么搞定的！

你的核心问题是要在Mock Provider的CreateQuery方法里，从传入的Expression中提取出过滤条件，然后应用到你的模拟数据列表_list上对吧？直接调用Invoke不行是因为表达式树需要先编译成可执行的委托，而且你得先正确解析出Where子句里的lambda表达式。

第一步：提取Where过滤条件 #

首先你需要写一个工具类（用ExpressionVisitor最灵活），来从传入的Expression里抠出Expression<Func<T, bool>>类型的过滤条件，它能轻松处理链式查询（比如Where+OrderBy这种组合）：

public class WhereClauseExtractor<T> : ExpressionVisitor
{
    public Expression<Func<T, bool>> FilterPredicate { get; private set; }

    protected override Expression VisitMethodCall(MethodCallExpression methodCallExpr)
    {
        // 识别Queryable.Where方法
        if (methodCallExpr.Method.DeclaringType == typeof(Queryable) 
            && methodCallExpr.Method.Name == "Where"
            && methodCallExpr.Arguments.Count == 2)
        {
            // 第二个参数是Where的lambda表达式，转成Expression<Func<T, bool>>
            var lambdaUnary = methodCallExpr.Arguments[1] as UnaryExpression;
            FilterPredicate = lambdaUnary?.Operand as Expression<Func<T, bool>>;
        }

        // 递归处理链式调用的上层表达式
        return base.VisitMethodCall(methodCallExpr);
    }
}

第二步：在MyProvider中应用过滤 #

接下来在你的MyProvider的CreateQuery方法里，用上面的提取器拿到过滤条件，编译后过滤你的模拟列表：

public IQueryable<T> CreateQuery<T>(Expression expression)
{
    var extractor = new WhereClauseExtractor<T>();
    extractor.Visit(expression);

    // 默认用完整的模拟列表
    IEnumerable<T> resultData = _list;

    // 如果提取到过滤条件，编译后应用到列表上
    if (extractor.FilterPredicate != null)
    {
        var filterFunc = extractor.FilterPredicate.Compile();
        resultData = _list.Where(filterFunc);
    }

    // 返回你的Mock Queryable，传入过滤后的数据
    return new MyQueryable<T>(resultData.AsQueryable(), this);
}

为什么之前Invoke没成功？ #

你之前尝试调用Invoke失败，大概率是因为没正确解析表达式结构——传入的Expression通常是MethodCallExpression（对应Queryable.Where的调用），而不是直接的lambda表达式。你得先把Where方法里的lambda参数提取出来，编译成委托后才能用来过滤列表。

扩展：处理更多查询操作 #

如果你的查询里还有OrderBy、Skip、Take这类操作，可以继续扩展ExpressionVisitor，识别对应的方法并应用到模拟数据上，比如处理OrderBy时提取排序的key selector，再调用OrderBy方法排序。

这样你的Mock Queryable就能正确模拟DocumentDB的查询行为，单元测试里就能验证你的GetDocuments方法的条件查询逻辑啦！

内容的提问来源于stack exchange，提问作者Horia Toma