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
48 for (int i = 0; i < numberOfMessages; i++) {
49 producer.send(destination, session.createTextMessage("hi"));
50 }
51
52
53 countDown.attempt(5000);
54 assertEquals("Undelivered messages for count: " + numberOfMessages, 0, countDown.currentCount());
55 }
56 }