Allow LimitOperator in Driver assertion

Author: dnhatn

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/Driver.java

@@ -267,7 +267,7 @@ private IsBlockedResult runSingleLoopIteration() {
 
             if (op.isFinished() == false && nextOp.needsInput()) {
                 driverContext.checkForEarlyTermination();
-                assert nextOp.isFinished() == false || nextOp instanceof ExchangeSinkOperator
+                assert nextOp.isFinished() == false || nextOp instanceof ExchangeSinkOperator || nextOp instanceof LimitOperator
                     : "next operator should not be finished yet: " + nextOp;
                 Page page = op.getOutput();
                 if (page == null) {