fixing CrossClustersCancellationIT

masseyke · Jan 3, 2024 · 7c9b19d · 7c9b19d
1 parent ece05a9
commit 7c9b19d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/...rnalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClustersCancellationIT.java b/...rnalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClustersCancellationIT.java
@@ -151,10 +151,14 @@ private void createRemoteIndex(int numDocs) throws Exception {
         mapping.endObject();
         client(REMOTE_CLUSTER).admin().indices().prepareCreate("test").setMapping(mapping).get();
         BulkRequestBuilder bulk = client(REMOTE_CLUSTER).prepareBulk("test").setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
-        for (int i = 0; i < numDocs; i++) {
-            bulk.add(new IndexRequest().source("foo", i));
+        try {
+            for (int i = 0; i < numDocs; i++) {
+                bulk.add(new IndexRequest().source("foo", i));
+            }
+            bulk.get();
+        } finally {
+            bulk.request().decRef();
         }
-        bulk.get();
     }
 
     public void testCancel() throws Exception {