From 1001298f583252c19036692bfa41d57ab0c86340 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 +
 pom.xml                                                         | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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 0bba94f05f9c..be05a0fcc8a9 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();
   }
 
diff --git a/pom.xml b/pom.xml
index de7543867610..d8e96ed859dd 100644
--- a/pom.xml
+++ b/pom.xml
@@ -1765,7 +1765,7 @@
                             @{jacocoArgLine}
                             ${jdk.strong.encapsulation.argLine}
                             ${jdk.security.manager.allow.argLine}
-                            -Xmx2500m
+                            -Xmx2048m
                             -XX:MaxDirectMemorySize=2500m
                             -XX:+ExitOnOutOfMemoryError
                             -XX:+HeapDumpOnOutOfMemoryError