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  
26  /***
27   * @version $Revision: 1.3 $
28   */
29  public class SendReceiveTest extends JmsTopicSendReceiveTest {
30  
31      private CountDown countDown;
32  
33      public void testSendReceive() throws Exception {
34          sendSomeMessagesAndCheckTheyAreAllReceived(1);
35          sendSomeMessagesAndCheckTheyAreAllReceived(2);
36          sendSomeMessagesAndCheckTheyAreAllReceived(3);
37          sendSomeMessagesAndCheckTheyAreAllReceived(4);
38      }
39  
40      public void onMessage(Message message) {
41          countDown.release();
42      }
43  
44      protected void sendSomeMessagesAndCheckTheyAreAllReceived(int numberOfMessages) throws JMSException, InterruptedException {
45          countDown = new CountDown(numberOfMessages);
46  
47          // send messages
48          for (int i = 0; i < numberOfMessages; i++) {
49              producer.send(destination, session.createTextMessage("hi"));
50          }
51  
52          // block until the listener has received everything it's expecting, or timeout
53          countDown.attempt(5000);
54          assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
55      }
56  }