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
61 for (int i = 0; i < numberOfMessages; i++) {
62 producer.send(producerDestination, session.createTextMessage("hi"));
63 }
64
65
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
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
81 countDown.attempt(5000);
82 assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
83 }
84 }