public Expression Apply(Session session, Expression query)
            {
                var visitor = new TestVisitor();
                var result  = visitor.Visit(query);

                if (visitor.ClosureRegistered())
                {
                    session.Extensions.Set(new ClosureMarker());
                }
                return(result);
            }