[jboss-cvs] JBoss Messaging SVN: r5479 - in trunk: tests/src/org/jboss/messaging/tests/integration/cluster/management and 1 other directories.

jboss-cvs-commits at lists.jboss.org jboss-cvs-commits at lists.jboss.org
Mon Dec 8 11:15:01 EST 2008


Author: jmesnil
Date: 2008-12-08 11:15:01 -0500 (Mon, 08 Dec 2008)
New Revision: 5479

Added:
   trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareMessagingServerControlWrapperTest.java
Modified:
   trunk/src/main/org/jboss/messaging/core/management/jmx/impl/ReplicationAwareMessagingServerControlWrapper.java
   trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareTestBase.java
   trunk/tests/src/org/jboss/messaging/tests/util/RandomUtil.java
Log:
added integration tests for ReplicationAwareMessagingServerControlWrapper

Modified: trunk/src/main/org/jboss/messaging/core/management/jmx/impl/ReplicationAwareMessagingServerControlWrapper.java
===================================================================
--- trunk/src/main/org/jboss/messaging/core/management/jmx/impl/ReplicationAwareMessagingServerControlWrapper.java	2008-12-08 15:54:01 UTC (rev 5478)
+++ trunk/src/main/org/jboss/messaging/core/management/jmx/impl/ReplicationAwareMessagingServerControlWrapper.java	2008-12-08 16:15:01 UTC (rev 5479)
@@ -245,7 +245,7 @@
 
    public boolean closeConnectionsForAddress(final String ipAddress) throws Exception
    {
-      return (Boolean)replicationAwareInvoke("closeConnectionsForAddress", ipAddress);
+      return localControl.closeConnectionsForAddress(ipAddress);
    }
 
    public boolean commitPreparedTransaction(final String transactionAsBase64) throws Exception

Added: trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareMessagingServerControlWrapperTest.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareMessagingServerControlWrapperTest.java	                        (rev 0)
+++ trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareMessagingServerControlWrapperTest.java	2008-12-08 16:15:01 UTC (rev 5479)
@@ -0,0 +1,299 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
+ * by the @authors tag. See the copyright.txt in the distribution for a
+ * full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+
+package org.jboss.messaging.tests.integration.cluster.management;
+
+import static org.jboss.messaging.tests.util.RandomUtil.randomLong;
+import static org.jboss.messaging.tests.util.RandomUtil.randomPositiveLong;
+import static org.jboss.messaging.tests.util.RandomUtil.randomSimpleString;
+
+import javax.management.MBeanServer;
+import javax.management.MBeanServerInvocationHandler;
+import javax.management.ObjectName;
+import javax.management.openmbean.CompositeData;
+
+import org.jboss.messaging.core.client.ClientMessage;
+import org.jboss.messaging.core.client.ClientProducer;
+import org.jboss.messaging.core.client.ClientSession;
+import org.jboss.messaging.core.client.impl.ClientSessionFactoryImpl;
+import org.jboss.messaging.core.client.impl.ClientSessionFactoryInternal;
+import org.jboss.messaging.core.config.TransportConfiguration;
+import org.jboss.messaging.core.management.MessagingServerControlMBean;
+import org.jboss.messaging.core.management.QueueControlMBean;
+import org.jboss.messaging.core.management.impl.ManagementServiceImpl;
+import org.jboss.messaging.core.remoting.impl.invm.InVMConnectorFactory;
+import org.jboss.messaging.tests.util.RandomUtil;
+import org.jboss.messaging.util.SimpleString;
+
+/**
+ * A ReplicationAwareQueueControlWrapperTest
+ *
+ * @author <a href="jmesnil at redhat.com">Jeff Mesnil</a>
+ */
+public class ReplicationAwareMessagingServerControlWrapperTest extends ReplicationAwareTestBase
+{
+   // Constants -----------------------------------------------------
+
+   // Attributes ----------------------------------------------------
+
+   // Static --------------------------------------------------------
+
+   private static MessagingServerControlMBean createMessagingServerControl(MBeanServer mbeanServer) throws Exception
+   {
+      MessagingServerControlMBean serverControl = (MessagingServerControlMBean)MBeanServerInvocationHandler.newProxyInstance(mbeanServer,
+                                                                                                                             ManagementServiceImpl.getMessagingServerObjectName(),
+                                                                                                                             MessagingServerControlMBean.class,
+                                                                                                                             false);
+      return serverControl;
+   }
+   
+   private static QueueControlMBean createQueueControl(SimpleString address, SimpleString name, MBeanServer mbeanServer) throws Exception
+   {
+      QueueControlMBean queueControl = (QueueControlMBean)MBeanServerInvocationHandler.newProxyInstance(mbeanServer,
+                                                                                                        ManagementServiceImpl.getQueueObjectName(address,
+                                                                                                                                                 name),
+                                                                                                        QueueControlMBean.class,
+                                                                                                        false);
+      return queueControl;
+   }
+
+   private SimpleString address;
+   private ClientSession session;
+
+   // Constructors --------------------------------------------------
+
+   // Public --------------------------------------------------------
+
+   public void testAddAddress() throws Exception
+   {
+      SimpleString address = randomSimpleString();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      ObjectName addressON = ManagementServiceImpl.getAddressObjectName(address);
+
+      assertResourceNotExists(liveMBeanServer, addressON);
+      assertResourceNotExists(backupMBeanServer, addressON);
+
+      assertTrue(liveServerControl.addAddress(address.toString()));
+
+      assertResourceExists(liveMBeanServer, addressON);
+      assertResourceExists(backupMBeanServer, addressON);
+   }
+
+   public void testRemoveAddress() throws Exception
+   {
+      SimpleString address = randomSimpleString();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      ObjectName addressON = ManagementServiceImpl.getAddressObjectName(address);
+
+      assertResourceNotExists(liveMBeanServer, addressON);
+      assertResourceNotExists(backupMBeanServer, addressON);
+
+      // add the address...
+      assertTrue(liveServerControl.addAddress(address.toString()));
+
+      assertResourceExists(liveMBeanServer, addressON);
+      assertResourceExists(backupMBeanServer, addressON);
+
+      // ... and remove it
+      assertTrue(liveServerControl.removeAddress(address.toString()));
+
+      assertResourceNotExists(liveMBeanServer, addressON);
+      assertResourceNotExists(backupMBeanServer, addressON);
+   }
+
+   public void testCreateQueue() throws Exception
+   {
+      SimpleString address = randomSimpleString();
+      SimpleString name = randomSimpleString();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      ObjectName queueON = ManagementServiceImpl.getQueueObjectName(address, name);
+
+      assertResourceNotExists(liveMBeanServer, queueON);
+      assertResourceNotExists(backupMBeanServer, queueON);
+
+      liveServerControl.createQueue(address.toString(), name.toString());
+
+      assertResourceExists(liveMBeanServer, queueON);
+      assertResourceExists(backupMBeanServer, queueON);
+   }
+
+   public void testDestroyQueue() throws Exception
+   {
+      SimpleString address = randomSimpleString();
+      SimpleString name = randomSimpleString();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      ObjectName queueON = ManagementServiceImpl.getQueueObjectName(address, name);
+
+      assertResourceNotExists(liveMBeanServer, queueON);
+      assertResourceNotExists(backupMBeanServer, queueON);
+
+      // create the queue...
+      liveServerControl.createQueue(address.toString(), name.toString());
+
+      assertResourceExists(liveMBeanServer, queueON);
+      assertResourceExists(backupMBeanServer, queueON);
+
+      // ... and destroy it
+      liveServerControl.destroyQueue(name.toString());
+
+      assertResourceNotExists(liveMBeanServer, queueON);
+      assertResourceNotExists(backupMBeanServer, queueON);
+   }
+
+   public void testEnableMessageCounters() throws Exception
+   {
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      MessagingServerControlMBean backupServerControl = createMessagingServerControl(backupMBeanServer);
+
+      assertFalse(liveServerControl.isMessageCounterEnabled());
+      assertFalse(backupServerControl.isMessageCounterEnabled());
+
+      liveServerControl.enableMessageCounters();
+
+      assertTrue(liveServerControl.isMessageCounterEnabled());
+      assertTrue(backupServerControl.isMessageCounterEnabled());
+   }
+
+   public void testDisableMessageCounters() throws Exception
+   {
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      MessagingServerControlMBean backupServerControl = createMessagingServerControl(backupMBeanServer);
+
+      assertFalse(liveServerControl.isMessageCounterEnabled());
+      assertFalse(backupServerControl.isMessageCounterEnabled());
+
+      // enable the counters...
+      liveServerControl.enableMessageCounters();
+
+      assertTrue(liveServerControl.isMessageCounterEnabled());
+      assertTrue(backupServerControl.isMessageCounterEnabled());
+
+      // and disable them
+      liveServerControl.disableMessageCounters();
+
+      assertFalse(liveServerControl.isMessageCounterEnabled());
+      assertFalse(backupServerControl.isMessageCounterEnabled());
+   }
+
+   public void testResetAllMessageCounters() throws Exception
+   {      
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      liveServerControl.enableMessageCounters();
+      liveServerControl.setMessageCounterSamplePeriod(2000);
+
+      QueueControlMBean liveQueueControl = createQueueControl(address, address, liveMBeanServer);
+      QueueControlMBean backupQueueControl = createQueueControl(address, address, backupMBeanServer);
+
+      // send on queue
+      ClientProducer producer = session.createProducer(address);
+      ClientMessage message = session.createClientMessage(false);
+      SimpleString key = randomSimpleString();
+      long value = randomLong();
+      message.putLongProperty(key, value);
+      producer.send(message);
+      
+      Thread.sleep(liveServerControl.getMessageCounterSamplePeriod() * 2);
+
+      // check the count is to 1 on both live & backup nodes
+      CompositeData counter = liveQueueControl.listMessageCounter();
+      assertEquals(1, counter.get("count"));
+      counter = backupQueueControl.listMessageCounter();
+      assertEquals(1, counter.get("count"));
+      
+      liveServerControl.resetAllMessageCounters();
+      Thread.sleep(liveServerControl.getMessageCounterSamplePeriod() * 2);
+      
+      // check the count has been reset to 0 on both live & backup nodes
+      counter = liveQueueControl.listMessageCounter();
+      assertEquals(0, counter.get("count"));
+      counter = backupQueueControl.listMessageCounter();
+      assertEquals(0, counter.get("count"));
+   }
+
+   public void testSetMessageCounterSamplePeriod() throws Exception
+   {
+      long newPeriod = randomPositiveLong();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      MessagingServerControlMBean backupServerControl = createMessagingServerControl(backupMBeanServer);
+
+      assertEquals(liveServerControl.getMessageCounterSamplePeriod(),
+                   backupServerControl.getMessageCounterSamplePeriod());
+
+      liveServerControl.setMessageCounterSamplePeriod(newPeriod);
+
+      assertEquals(newPeriod, liveServerControl.getMessageCounterSamplePeriod());
+      assertEquals(newPeriod, backupServerControl.getMessageCounterSamplePeriod());
+   }
+
+   public void testSetMessageCounterMaxDayCount() throws Exception
+   {
+      int newCount = RandomUtil.randomPositiveInt();
+
+      MessagingServerControlMBean liveServerControl = createMessagingServerControl(liveMBeanServer);
+      MessagingServerControlMBean backupServerControl = createMessagingServerControl(backupMBeanServer);
+
+      assertEquals(liveServerControl.getMessageCounterMaxDayCount(), backupServerControl.getMessageCounterMaxDayCount());
+
+      liveServerControl.setMessageCounterMaxDayCount(newCount);
+
+      assertEquals(newCount, liveServerControl.getMessageCounterMaxDayCount());
+      assertEquals(newCount, backupServerControl.getMessageCounterMaxDayCount());
+   }
+
+   // Package protected ---------------------------------------------
+
+   // Protected -----------------------------------------------------
+
+   @Override
+   protected void setUp() throws Exception
+   {
+      super.setUp();
+
+      address = RandomUtil.randomSimpleString();
+
+      ClientSessionFactoryInternal sf = new ClientSessionFactoryImpl(new TransportConfiguration(InVMConnectorFactory.class.getName()),
+                                                                     new TransportConfiguration(InVMConnectorFactory.class.getName(),
+                                                                                                backupParams));
+
+      session = sf.createSession(false, true, true);
+
+      session.createQueue(address, address, null, false, false, true);
+   }
+
+   @Override
+   protected void tearDown() throws Exception
+   {
+      session.close();
+
+      super.tearDown();
+   }
+   
+   // Private -------------------------------------------------------
+
+   // Inner classes -------------------------------------------------
+
+}

Modified: trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareTestBase.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareTestBase.java	2008-12-08 15:54:01 UTC (rev 5478)
+++ trunk/tests/src/org/jboss/messaging/tests/integration/cluster/management/ReplicationAwareTestBase.java	2008-12-08 16:15:01 UTC (rev 5479)
@@ -27,6 +27,7 @@
 
 import javax.management.MBeanServer;
 import javax.management.MBeanServerFactory;
+import javax.management.ObjectName;
 
 import junit.framework.TestCase;
 
@@ -73,6 +74,49 @@
 
    // Static --------------------------------------------------------
 
+   protected static void assertResourceExists(MBeanServer mbeanServer, ObjectName objectName)
+   {
+      boolean registered = mbeanServer.isRegistered(objectName);
+      if (!registered)
+      {
+         fail("Resource does not exist: " + objectName);
+      }
+   }
+
+   protected static void assertResourceNotExists(MBeanServer mbeanServer, ObjectName objectName)
+   {
+      boolean registered = mbeanServer.isRegistered(objectName);
+      if (registered)
+      {
+         fail("Resource exists: " + objectName);
+      }
+   }
+
+   protected static MessagingService createNullStorageMessagingServer(final Configuration config, MBeanServer mbeanServer)
+   {
+      StorageManager storageManager = new NullStorageManager();
+      
+      RemotingService remotingService = new RemotingServiceImpl(config);
+
+      JBMSecurityManager securityManager = new JBMSecurityManagerImpl(true);
+      
+      ManagementService managementService = new ManagementServiceImpl(mbeanServer, config.isJMXManagementEnabled());
+      
+      MessagingServer server = new MessagingServerImpl();
+      
+      server.setConfiguration(config);
+      
+      server.setStorageManager(storageManager);
+      
+      server.setRemotingService(remotingService);
+      
+      server.setSecurityManager(securityManager);
+      
+      server.setManagementService(managementService);
+      
+      return new MessagingServiceImpl(server, storageManager, remotingService);
+   }
+   
    // Constructors --------------------------------------------------
 
    // Public --------------------------------------------------------
@@ -126,31 +170,6 @@
    }
 
    // Private -------------------------------------------------------
-
-   private static MessagingService createNullStorageMessagingServer(final Configuration config, MBeanServer mbeanServer)
-   {
-      StorageManager storageManager = new NullStorageManager();
-      
-      RemotingService remotingService = new RemotingServiceImpl(config);
-
-      JBMSecurityManager securityManager = new JBMSecurityManagerImpl(true);
-      
-      ManagementService managementService = new ManagementServiceImpl(mbeanServer, config.isJMXManagementEnabled());
-      
-      MessagingServer server = new MessagingServerImpl();
-      
-      server.setConfiguration(config);
-      
-      server.setStorageManager(storageManager);
-      
-      server.setRemotingService(remotingService);
-      
-      server.setSecurityManager(securityManager);
-      
-      server.setManagementService(managementService);
-      
-      return new MessagingServiceImpl(server, storageManager, remotingService);
-   }
    
    // Inner classes -------------------------------------------------
 

Modified: trunk/tests/src/org/jboss/messaging/tests/util/RandomUtil.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/util/RandomUtil.java	2008-12-08 15:54:01 UTC (rev 5478)
+++ trunk/tests/src/org/jboss/messaging/tests/util/RandomUtil.java	2008-12-08 16:15:01 UTC (rev 5479)
@@ -67,6 +67,11 @@
       return random.nextLong();
    }
 
+   public static long randomPositiveLong()
+   {
+      return Math.abs(randomLong());
+   }
+
    public static int randomInt()
    {
       return random.nextInt();




More information about the jboss-cvs-commits mailing list