From 095471c4a4e72f7eca4501bc43763e4b55d12177 Mon Sep 17 00:00:00 2001
From: Nhat Nguyen <nhat.nguyen@elastic.co>
Date: Tue, 8 Jul 2025 16:26:26 -0700
Subject: [PATCH] Clear mock transport service rules in DataNodeRequestSenderIT

---
 .../xpack/esql/plugin/DataNodeRequestSenderIT.java            | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSenderIT.java b/x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSenderIT.java
index 9b3c2278c1bb8..c457868dc1659 100644
--- a/x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSenderIT.java
+++ b/x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSenderIT.java
@@ -153,6 +153,10 @@ public void testRetryOnShardMovement() {
 
         try (EsqlQueryResponse resp = run("FROM " + randomFrom("index-1,index-2", "index-*"))) {
             assertThat(getValuesList(resp), hasSize(2));
+        } finally {
+            for (TransportService transportService : internalCluster().getInstances(TransportService.class)) {
+                as(transportService, MockTransportService.class).clearAllRules();
+            }
         }
     }
 }