From de4c9f1afb6b3a89cdaf9c8730f1e54f0f716ee8 Mon Sep 17 00:00:00 2001
From: Akshat Jain <akjn11@gmail.com>
Date: Sun, 15 Sep 2024 18:53:10 +0530
Subject: [PATCH] Handle memory leaks from Mockito inline mocks

---
 .../src/test/java/org/apache/druid/msq/test/MSQTestBase.java     | 1 +
 1 file changed, 1 insertion(+)

diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java
index c17238d4b683..df9a00925cd4 100644
--- a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java
+++ b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java
@@ -381,6 +381,7 @@ public List<? extends com.fasterxml.jackson.databind.Module> getJacksonModules()
   @AfterEach
   public void tearDown2()
   {
+    Mockito.framework().clearInlineMocks();
     groupByBuffers.close();
   }