entity-framework-4 – 通过扩展方法的IDbSet和公开包含方法

我使用EF的Code-First方法，我想使用IDbSet而不是DbSet，所以我可以用嘲讽单元测试。我的问题是在需要时使用Include()方法来加载，但是不能通过IDbSet来显示Include()。我看到一个使用扩展方法来显示Include()的示例代码，但它似乎不适用于我;此示例中的objectQuery对象始终为空。请让我知道如何解决这个问题。

public static class IQueryableExtension
{
    public static IQueryable<T> Include<T>(this IQueryable<T> source,string path)
        where T : class
    {
        ObjectQuery<T> objectQuery = source as ObjectQuery<T>;
        if (objectQuery != null)
        {
            return objectQuery.Include(path);
        }
        return source;
    }

    public static IQueryable<T> Include<T,TProperty>(this IQueryable<T> source,System.Linq.Expressions.Expression<Func<T,TProperty>> path)
        where T : class
    {
        ObjectQuery<T> objectQuery = source as ObjectQuery<T>;
        if (objectQuery != null)
        {
            return source.Include(path);
        }
        return source;
    }
}

using System.Data.Entity;