[hornetq-commits] JBoss hornetq SVN: r11396 - branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11.

[do-not-reply at jboss.org]

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