From 13f0aae8a93f91c8444d9f538f3cde4172a5788d Mon Sep 17 00:00:00 2001
From: Lari Hotari <lhotari@users.noreply.github.com>
Date: Mon, 6 May 2024 21:48:47 +0300
Subject: [PATCH] [fix][test] Clear MockedPulsarServiceBaseTest fields to
 prevent test runtime memory leak (#22659)

(cherry picked from commit 025354ef4e733d62eee0d332edacb0b33b787da2)
---
 .../pulsar/broker/auth/MockedPulsarServiceBaseTest.java    | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
index 7380912e4247a..2cdb8e11da7b7 100644
--- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
+++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
@@ -258,13 +258,20 @@ protected final void internalCleanup() throws Exception {
         }
         if (brokerGateway != null) {
             brokerGateway.close();
+            brokerGateway = null;
         }
         if (pulsarTestContext != null) {
             pulsarTestContext.close();
             pulsarTestContext = null;
         }
+
         resetConfig();
         onCleanup();
+
+        // clear fields to avoid test runtime memory leak, pulsarTestContext already handles closing of these instances
+        pulsar = null;
+        mockZooKeeper = null;
+        mockZooKeeperGlobal = null;
     }
 
     protected void onCleanup() {