Author: gaohoward Date: 2011-09-22 11:30:57 -0400 (Thu, 22 Sep 2011) New Revision: 11396 Modified: branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11/StompTestV11.java Log: test Modified: branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11/StompTestV11.java =================================================================== --- branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11/StompTestV11.java 2011-09-22 01:37:18 UTC (rev 11395) +++ branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11/StompTestV11.java 2011-09-22 15:30:57 UTC (rev 11396) @@ -19,9 +19,12 @@ import java.io.IOException; import java.nio.channels.ClosedChannelException; +import java.util.concurrent.CountDownLatch; +import java.util.concurrent.TimeUnit; import javax.jms.Message; import javax.jms.MessageConsumer; +import javax.jms.MessageListener; import javax.jms.TextMessage; import junit.framework.Assert; @@ -1158,24 +1161,18 @@ public void testRedeliveryWithClientAck() throws Exception { + connV11.connect(defUser, defPass); - String frame = "CONNECT\n" + "login: brianm\n" + "passcode: wombats\n\n" + Stomp.NULL; - sendFrame(frame); + this.subscribe(connV11, "subId", "client"); - frame = receiveFrame(10000); - Assert.assertTrue(frame.startsWith("CONNECTED")); + sendMessage(getName()); - frame = "SUBSCRIBE\n" + "destination:" + getQueuePrefix() + getQueueName() + "\n" + "ack:client\n\n" + Stomp.NULL; + ClientStompFrame frame = connV11.receiveFrame(); + + assertTrue(frame.getCommand().equals("MESSAGE")); - sendFrame(frame); + connV11.disconnect(); - sendMessage(getName()); - frame = receiveFrame(10000); - Assert.assertTrue(frame.startsWith("MESSAGE")); - - frame = "DISCONNECT\n" + "\n\n" + Stomp.NULL; - sendFrame(frame); - // message should be received since message was not acknowledged MessageConsumer consumer = session.createConsumer(queue); Message message = consumer.receive(1000); @@ -1183,6 +1180,37 @@ Assert.assertTrue(message.getJMSRedelivered()); } + public void testSendManyMessages() throws Exception + { + MessageConsumer consumer = session.createConsumer(queue); + + connV11.connect(defUser, defPass); + + int count = 1000; + final CountDownLatch latch = new CountDownLatch(count); + consumer.setMessageListener(new MessageListener() + { + public void onMessage(Message arg0) + { + latch.countDown(); + } + }); + + ClientStompFrame frame = connV11.createFrame("SEND"); + frame.addHeader("destination", getQueuePrefix() + getQueueName()); + frame.setBody("Hello World"); + + for (int i = 1; i <= count; i++) + { + connV11.sendFrame(frame); + } + + assertTrue(latch.await(60, TimeUnit.SECONDS)); + + connV11.disconnect(); + } + + //-----------------private help methods private void abortTransaction(StompClientConnection conn, String txID) throws IOException, InterruptedException