commit 37b5dbf0ea461b7570afddd0504b4c2319a7e83e
author Timothy Bish <tabish121@gmail.com> Wed Apr 27 14:50:29 2016 -0400
committer Christopher L. Shannon (cshannon) <christopher.l.shannon@gmail.com> Thu Apr 28 11:56:52 2016 +0000
tree 38234080d6549ca90dda93dc3908d4b2a89b2493
parent d38c5906f6eadbd20aee97ce37c665b3356dbfd1

Add some additional checks for valid state.
(cherry picked from commit 3c342ffce43546667f59bb16db373061724e1b0f)

activemq-unit-tests/src/test/java/org/apache/activemq/bugs/AMQ6254Test.java

diff --git a/activemq-unit-tests/src/test/java/org/apache/activemq/bugs/AMQ6254Test.java b/activemq-unit-tests/src/test/java/org/apache/activemq/bugs/AMQ6254Test.java
index 9c9a35b..f88fca9 100644
--- a/activemq-unit-tests/src/test/java/org/apache/activemq/bugs/AMQ6254Test.java
+++ b/activemq-unit-tests/src/test/java/org/apache/activemq/bugs/AMQ6254Test.java
@@ -27,14 +27,18 @@
 import java.util.Set;
 
 import javax.jms.Connection;
+import javax.jms.JMSException;
 import javax.jms.MessageProducer;
 import javax.jms.Session;
 import javax.jms.TextMessage;
 import javax.jms.TopicSubscriber;
+import javax.management.MalformedObjectNameException;
+import javax.management.ObjectName;
 
 import org.apache.activemq.ActiveMQConnectionFactory;
 import org.apache.activemq.broker.BrokerPlugin;
 import org.apache.activemq.broker.BrokerService;
+import org.apache.activemq.broker.jmx.TopicViewMBean;
 import org.apache.activemq.broker.region.Destination;
 import org.apache.activemq.broker.region.RegionBroker;
 import org.apache.activemq.broker.region.TopicRegion;
@@ -111,8 +115,26 @@
         assertNotNull("Message not received.", message);
         assertEquals("Hello A", message.getText());
 
+        assertTrue("Should have only one consumer", Wait.waitFor(new Wait.Condition() {
+
+            @Override
+            public boolean isSatisified() throws Exception {
+                return getProxyToTopic(topicA).getConsumerCount() == 1;
+            }
+        }));
+
         subscriber.close();
 
+        assertTrue("Should have one message consumed", Wait.waitFor(new Wait.Condition() {
+
+            @Override
+            public boolean isSatisified() throws Exception {
+                return getProxyToTopic(topicA).getDequeueCount() == 1;
+            }
+        }));
+
+        connection.close();
+
         assertTrue("Should have only one destination", Wait.waitFor(new Wait.Condition() {
 
             @Override
@@ -122,6 +144,14 @@
             }
         }));
 
+        assertTrue("Should have only one inactive subscription", Wait.waitFor(new Wait.Condition() {
+
+            @Override
+            public boolean isSatisified() throws Exception {
+                return brokerService.getAdminView().getInactiveDurableTopicSubscribers().length == 1;
+            }
+        }));
+
         // Restart broker
         brokerService.stop();
         brokerService.waitUntilStopped();
@@ -201,7 +231,7 @@
         BrokerService answer = new BrokerService();
 
         answer.setKeepDurableSubsActive(true);
-        answer.setUseJmx(false);
+        answer.setUseJmx(true);
         answer.setPersistent(true);
         answer.setDeleteAllMessagesOnStartup(deleteAllMessages);
         answer.setAdvisorySupport(false);
@@ -308,4 +338,11 @@
 
         return authorizationPlugin;
     }
+
+    protected TopicViewMBean getProxyToTopic(String name) throws MalformedObjectNameException, JMSException {
+        ObjectName queueViewMBeanName = new ObjectName("org.apache.activemq:type=Broker,brokerName=localhost,destinationType=Topic,destinationName="+name);
+        TopicViewMBean proxy = (TopicViewMBean) brokerService.getManagementContext()
+                .newProxyInstance(queueViewMBeanName, TopicViewMBean.class, true);
+        return proxy;
+    }
 }