smaller

Author: dnhatn

test/framework/src/main/java/org/elasticsearch/indices/CrankyCircuitBreakerService.java

@@ -29,7 +29,7 @@ public class CrankyCircuitBreakerService extends CircuitBreakerService {
      */
     public static final String ERROR_MESSAGE = "cranky breaker";
 
-    private final CircuitBreaker breaker = new CircuitBreaker() {
+    public static final class CrankyCircuitBreaker implements CircuitBreaker {
         private final AtomicLong used = new AtomicLong();
 
         @Override
@@ -82,7 +82,9 @@ public Durability getDurability() {
         public void setLimitAndOverhead(long limit, double overhead) {
 
         }
-    };
+    }
+
+    private final CrankyCircuitBreaker breaker = new CrankyCircuitBreaker();
 
     @Override
     public CircuitBreaker getBreaker(String name) {

x-pack/plugin/esql/compute/test/src/main/java/org/elasticsearch/compute/test/TestDriverFactory.java

@@ -18,6 +18,7 @@
 import org.elasticsearch.compute.operator.SourceOperator;
 import org.elasticsearch.core.Releasable;
 import org.elasticsearch.core.Releasables;
+import org.elasticsearch.indices.CrankyCircuitBreakerService;
 
 import java.util.List;
 
@@ -44,7 +45,8 @@ public static Driver create(
         Releasable releasable
     ) {
         // Do not wrap the local breaker for small local breakers, as the output mights not match expectations.
-        if (driverContext.breaker() instanceof LocalCircuitBreaker == false
+        if (driverContext.breaker() instanceof CrankyCircuitBreakerService.CrankyCircuitBreaker == false
+            && driverContext.breaker() instanceof LocalCircuitBreaker == false
             && driverContext.breaker().getLimit() >= ByteSizeValue.ofMb(100).getBytes()
             && Randomness.get().nextBoolean()) {
             final int overReservedBytes = Randomness.get().nextInt(1024 * 1024);