Source code: org/activemq/test/SendReceiveTest.java
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.activemq.test;
19
20 import EDU.oswego.cs.dl.util.concurrent.CountDown;
21
22 import javax.jms.JMSException;
23 import javax.jms.Message;
24 import javax.jms.TextMessage;
25
26 /**
27 * @version $Revision: 1.1.1.1 $
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 testSendReceiveReusingTheMessageInstance() throws Exception {
41 sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(1);
42 sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(2);
43 sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(3);
44 sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(4);
45 }
46
47 public void onMessage(Message message) {
48 if (countDown != null) {
49 countDown.release();
50 }
51 else {
52 System.out.println("Warning: no countDown available!");
53 }
54 }
55
56 protected void sendSomeMessagesAndCheckTheyAreAllReceived(int numberOfMessages) throws JMSException, InterruptedException {
57 countDown = new CountDown(numberOfMessages);
58
59 // send messages
60 for (int i = 0; i < numberOfMessages; i++) {
61 producer.send(producerDestination, session.createTextMessage("hi"));
62 }
63
64 // block until the listener has received everything it's expecting, or timeout
65 countDown.attempt(5000);
66 assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
67 }
68
69 protected void sendSomeMessagesReusingTheSameMessageAndCheckTheyAreAllReceived(int numberOfMessages) throws JMSException, InterruptedException {
70 countDown = new CountDown(numberOfMessages);
71
72 // send messages
73 TextMessage message = session.createTextMessage();
74 for (int i = 0; i < numberOfMessages; i++) {
75 message.setText("This is message: " + i);
76 producer.send(producerDestination, message);
77 }
78
79 // block until the listener has received everything it's expecting, or timeout
80 countDown.attempt(5000);
81 assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
82 }
83 }