Package org.hornetq.jms.tests

Source Code of org.hornetq.jms.tests.TopicTest

/*
* Copyright 2009 Red Hat, Inc.
* Red Hat licenses this file to you under the Apache License, version
* 2.0 (the "License"); you may not use this file except in compliance
* with the License.  You may obtain a copy of the License at
*    http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied.  See the License for the specific language governing
* permissions and limitations under the License.
*/

package org.hornetq.jms.tests;

import java.io.Serializable;

import javax.jms.Connection;
import javax.jms.DeliveryMode;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageListener;
import javax.jms.MessageProducer;
import javax.jms.ObjectMessage;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.Topic;

import org.hornetq.jms.tests.util.ProxyAssertSupport;

/**
* @author <a href="mailto:ovidiu@feodorov.com">Ovidiu Feodorov</a>
* @version <tt>$Revision: 8611 $</tt>
*
* $Id: TopicTest.java 8611 2009-12-08 01:06:31Z timfox $
*/
public class TopicTest extends JMSTestCase
{
   // Constants -----------------------------------------------------

   // Static --------------------------------------------------------

   // Attributes ----------------------------------------------------

   // Constructors --------------------------------------------------

   // Public --------------------------------------------------------

   /**
    * The simplest possible topic test.
    */
   public void testTopic() throws Exception
   {
      Connection conn = null;

      try
      {
         conn = JMSTestCase.cf.createConnection();

         Session s = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
         MessageProducer p = s.createProducer(HornetQServerTestCase.topic1);
         MessageConsumer c = s.createConsumer(HornetQServerTestCase.topic1);
         conn.start();

         p.send(s.createTextMessage("payload"));
         TextMessage m = (TextMessage)c.receive();

         ProxyAssertSupport.assertEquals("payload", m.getText());
      }
      finally
      {
         if (conn != null)
         {
            conn.close();
         }
      }
   }

   public void testTopic2() throws Exception
   {
      Connection conn = null;

      try
      {
         conn = JMSTestCase.cf.createConnection();

         Session s = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
         MessageProducer p = s.createProducer(HornetQServerTestCase.topic1);
         MessageConsumer c = s.createConsumer(HornetQServerTestCase.topic1);
         conn.start();

         p.send(s.createTextMessage("payload"));
         TextMessage m = (TextMessage)c.receive();

         ProxyAssertSupport.assertEquals("payload", m.getText());
      }
      finally
      {
         if (conn != null)
         {
            conn.close();
         }
      }
   }

   public void testTopicName() throws Exception
   {
      Topic topic = (Topic)JMSTestCase.ic.lookup("/topic/Topic1");
      ProxyAssertSupport.assertEquals("Topic1", topic.getTopicName());
   }

   /*
   * See http://jira.jboss.com/jira/browse/JBMESSAGING-399
   */
   public void testRace() throws Exception
   {
      Connection conn = null;

      try
      {
         conn = JMSTestCase.cf.createConnection();

         Session sSend = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);

         MessageProducer prod = sSend.createProducer(HornetQServerTestCase.topic1);
         prod.setDeliveryMode(DeliveryMode.NON_PERSISTENT);

         Session s1 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
         Session s2 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
         Session s3 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);

         MessageConsumer c1 = s1.createConsumer(HornetQServerTestCase.topic1);
         MessageConsumer c2 = s2.createConsumer(HornetQServerTestCase.topic1);
         MessageConsumer c3 = s3.createConsumer(HornetQServerTestCase.topic1);

         final int numMessages = 500;

         TestListener l1 = new TestListener(numMessages);
         TestListener l2 = new TestListener(numMessages);
         TestListener l3 = new TestListener(numMessages);

         c1.setMessageListener(l1);
         c2.setMessageListener(l2);
         c3.setMessageListener(l3);

         conn.start();

         for (int i = 0; i < numMessages; i++)
         {
            byte[] blah = new byte[10000];
            String str = new String(blah);

            Wibble2 w = new Wibble2();
            w.s = str;
            ObjectMessage om = sSend.createObjectMessage(w);

            prod.send(om);
         }

         l1.waitForMessages();
         l2.waitForMessages();
         l3.waitForMessages();

         ProxyAssertSupport.assertFalse(l1.failed);
         ProxyAssertSupport.assertFalse(l2.failed);
         ProxyAssertSupport.assertFalse(l3.failed);
      }
      finally
      {
         if (conn != null)
         {
            conn.close();
         }
      }
   }

   // Package protected ---------------------------------------------

   // Protected -----------------------------------------------------

   // Private -------------------------------------------------------

   // Inner classes -------------------------------------------------

   static class Wibble2 implements Serializable
   {
      private static final long serialVersionUID = -5146179676719808756L;

      String s;
   }

   static class TestListener implements MessageListener
   {
      boolean failed;

      int count;

      int num;

      TestListener(final int num)
      {
         this.num = num;
      }

      public synchronized void onMessage(final Message m)
      {
         ObjectMessage om = (ObjectMessage)m;

         try
         {
            Wibble2 w = (Wibble2)om.getObject();
         }
         catch (Exception e)
         {
            failed = true;
         }

         count++;

         if (count == num)
         {
            notify();
         }
      }

      synchronized void waitForMessages() throws Exception
      {
         while (count < num)
         {
            this.wait();
         }
      }
   }

}
TOP

Related Classes of org.hornetq.jms.tests.TopicTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.