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); }