From df354fa0237dba309592277e3cfecbe1d55690be Mon Sep 17 00:00:00 2001
From: "Christopher L. Shannon (cshannon)" <christopher.l.shannon@gmail.com>
Date: Wed, 23 Feb 2022 07:40:11 -0500
Subject: [PATCH] AMQ-8509: Update TempStore Usage test to verify cleanup

Add a unit test to the previous tests added in AMQ-7085 to verify
TempUsage is cleaned up and stopped on Queue.stop()
---
 .../org/apache/activemq/bugs/AMQ7085Test.java | 22 ++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/activemq-broker/src/test/java/org/apache/activemq/bugs/AMQ7085Test.java b/activemq-broker/src/test/java/org/apache/activemq/bugs/AMQ7085Test.java
index d8e25169a7..e818b27b0a 100644
--- a/activemq-broker/src/test/java/org/apache/activemq/bugs/AMQ7085Test.java
+++ b/activemq-broker/src/test/java/org/apache/activemq/bugs/AMQ7085Test.java
@@ -16,6 +16,15 @@
  */
 package org.apache.activemq.bugs;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+
+import javax.jms.Connection;
+import javax.jms.Destination;
+import javax.jms.Message;
+import javax.jms.MessageProducer;
+import javax.jms.Session;
+import javax.management.ObjectName;
 import org.apache.activemq.ActiveMQConnectionFactory;
 import org.apache.activemq.broker.BrokerService;
 import org.apache.activemq.broker.jmx.BrokerView;
@@ -25,11 +34,6 @@ import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
-import javax.jms.*;
-import javax.management.ObjectName;
-
-import static org.junit.Assert.assertEquals;
-
 /**
  * Tests to ensure when the temp usage limit is updated on the broker the queues also have their
  * temp usage limits automatically updated.
@@ -94,6 +98,10 @@ public class AMQ7085Test
         brokerView.setTempLimit(newBrokerTempLimit);
         assertEquals(brokerView.getTempLimit(), newBrokerTempLimit);
         assertEquals(queueViewMBean.getTempUsageLimit(), queueTempLimit);
+
+        //Verify that TempUsage is cleaned up on Queue stop
+        brokerService.getDestination(queue).stop();
+        assertFalse(brokerService.getDestination(queue).getTempUsage().isStarted());
     }
 
     @Test
@@ -112,5 +120,9 @@ public class AMQ7085Test
         brokerView.setTempLimit(newBrokerTempLimit);
         assertEquals(brokerView.getTempLimit(), newBrokerTempLimit);
         assertEquals(queueViewMBean.getTempUsageLimit(), newBrokerTempLimit);
+
+        //Verify that TempUsage is cleaned up on Queue stop
+        brokerService.getDestination(queue).stop();
+        assertFalse(brokerService.getDestination(queue).getTempUsage().isStarted());
     }
 }