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  }