Issue with NeinLinq.EntityFramework, Linq.Count() method about nein-linq HOT 6 CLOSED

public class GetTodoItemsQueryProvider : IQueryProvider<ApplicationDbContext, int, TodoItem>
{
    [InjectLambda(nameof(GetQueryExpression))]
    public IQueryable<TodoItem> GetQuery(ApplicationDbContext context, int query)
    {
        throw new NotImplementedException();
    }

    public Expression<Func<ApplicationDbContext, int, IQueryable<TodoItem>>> GetQueryExpression()
    {
        return (context, query) => from todoItem in context.TodoItems
                                   where todoItem.ListId == query
                                   select todoItem;
    }
}

using System.Data.Entity;
using Xunit;

namespace NeinLinq.Tests;

public class SomeTest
{
    private readonly ApplicationDbContext _dbContext;
    private readonly IQueryProvider<ApplicationDbContext, int, TodoItem> _queryProvider;

    public SomeTest()
    {
        _dbContext = new ApplicationDbContext("Data Source=(localdb)\\MSSQLLocalDB;Initial Catalog=SomeTest;Integrated Security=true;");
        _dbContext.TodoLists.Add(
            new TodoList
            {
                Items =
                {
                    new TodoItem()
                }
            }
        );
        _dbContext.SaveChanges();

        _queryProvider = new GetTodoItemsQueryProvider();
    }

    [Fact]
    public void Test_WithLambdaInjection()
    {
        var query = from list in _dbContext.Set<TodoList>().ToDbInjectable()
                    let item = _queryProvider.GetQuery(_dbContext, list.Id).FirstOrDefault()
                    select item;

        var result = query.ToList();

        Assert.NotEmpty(result);

        int count = query.Count();

        Assert.NotEqual(0, count);
    }

    [Fact]
    public void Test_WithoutLambdaInjection()
    {
        var query = from list in _dbContext.Set<TodoList>()
                    let item = (from todoItem in _dbContext.Set<TodoItem>()
                                where todoItem.ListId == list.Id
                                select todoItem).FirstOrDefault()
                    select item;

        var result = query.ToList();

        Assert.NotEmpty(result);

        int count = query.Count();

        Assert.NotEqual(0, count);
    }
}

public class TodoList
{
    public int Id { get; set; }

    public virtual ICollection<TodoItem> Items { get; } = new List<TodoItem>();
}

public class TodoItem
{
    public int Id { get; set; }

    public int ListId { get; set; }

    public virtual TodoList? List { get; set; }
}

public class ApplicationDbContext : DbContext
{
    public DbSet<TodoList> TodoLists { get; }

    public DbSet<TodoItem> TodoItems { get; }

    public ApplicationDbContext(string nameOrConnectionString) : base(nameOrConnectionString)
    {
        TodoLists = Set<TodoList>();
        TodoItems = Set<TodoItem>();
    }
}

public interface IQueryProvider<DbContext, TParam, TResult>
{
    [InjectLambda]
    IQueryable<TResult> GetQuery(DbContext context, TParam query);
}

public class GetTodoItemsQueryProvider : IQueryProvider<ApplicationDbContext, int, TodoItem>
{
    [InjectLambda(nameof(GetQueryExpression))]
    public IQueryable<TodoItem> GetQuery(ApplicationDbContext context, int query)
    {
        throw new NotImplementedException();
    }

    public Expression<Func<ApplicationDbContext, int, IQueryable<TodoItem>>> GetQueryExpression()
    {
        return (context, query) => from todoItem in context.TodoItems
                                   where todoItem.ListId == query
                                   select todoItem;
    }
}