SendReceiveTest xref

    1   /***
    2    *
    3    * Copyright 2004 Protique Ltd
    4    *
    5    * Licensed under the Apache License, Version 2.0 (the "License");
    6    * you may not use this file except in compliance with the License.
    7    * You may obtain a copy of the License at
    8    *
    9    * http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   *
    17   **/
    18  package org.codehaus.activemq.test;
    19
    20  import EDU.oswego.cs.dl.util.concurrent.CountDown;
    21  import org.codehaus.activemq.JmsTopicSendReceiveTest;
    22
    23  import javax.jms.JMSException;
    24  import javax.jms.Message;
    25  import javax.jms.TextMessage;
    26
    27  /***
    28   * @version $Revision: 1.6 $
    29   */
    30  public class SendReceiveTest extends JmsTopicSendReceiveTest {
    31
    32      private CountDown countDown;
    33
    34      public void testSendReceive() throws Exception {
    35          sendSomeMessagesAndCheckTheyAreAllReceived(1);
    36          sendSomeMessagesAndCheckTheyAreAllReceived(2);
    37          sendSomeMessagesAndCheckTheyAreAllReceived(3);
    38          sendSomeMessagesAndCheckTheyAreAllReceived(4);
    39      }
    40
    41      public void testSendReceiveReusingTheMessageInstance() throws Exception {
    42          sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(1);
    43          sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(2);
    44          sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(3);
    45          sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(4);
    46      }
    47
    48      public void onMessage(Message message) {
    49          if (countDown != null) {
    50              countDown.release();
    51          }
    52          else {
    53              System.out.println("Warning: no countDown available!");
    54          }
    55      }
    56
    57      protected void sendSomeMessagesAndCheckTheyAreAllReceived(int numberOfMessages) throws JMSException, InterruptedException {
    58          countDown = new CountDown(numberOfMessages);
    59
    60          // send messages
    61          for (int i = 0; i < numberOfMessages; i++) {
    62              producer.send(producerDestination, session.createTextMessage("hi"));
    63          }
    64
    65          // block until the listener has received everything it's expecting, or timeout
    66          countDown.attempt(5000);
    67          assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
    68      }
    69
    70      protected void sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(int numberOfMessages) throws JMSException, InterruptedException {
    71          countDown = new CountDown(numberOfMessages);
    72
    73          // send messages
    74          TextMessage message = session.createTextMessage();
    75          for (int i = 0; i < numberOfMessages; i++) {
    76              message.setText("This is message: " + i);
    77              producer.send(producerDestination, message);
    78          }
    79
    80          // block until the listener has received everything it's expecting, or timeout
    81          countDown.attempt(5000);
    82          assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
    83      }
    84  }