[hornetq-commits] JBoss hornetq SVN: r11355 - 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-16 10:39:57 -0400 (Fri, 16 Sep 2011)
New Revision: 11355

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-16 10:28:52 UTC (rev 11354)
+++ branches/STOMP11/tests/integration-tests/src/test/java/org/hornetq/tests/integration/stomp/v11/StompTestV11.java	2011-09-16 14:39:57 UTC (rev 11355)
@@ -608,7 +608,98 @@
       Message message = consumer.receive(1000);
       Assert.assertNotNull(message);
    }
+   
+   
+   public void testAck() throws Exception
+   {
+      connV11.connect(defUser, defPass);
 
+      subscribe(connV11, "sub1", "client");
+
+      sendMessage(getName());
+
+      ClientStompFrame frame = connV11.receiveFrame();
+
+      String messageID = frame.getHeader("message-id");
+      
+      System.out.println("Received message with id " + messageID);
+      
+      ack(connV11, "sub1", messageID);
+      
+      connV11.disconnect();
+
+      //Nack makes the message be dropped.
+      MessageConsumer consumer = session.createConsumer(queue);
+      Message message = consumer.receive(1000);
+      Assert.assertNull(message);
+   }
+
+   public void testAckWithWrongSubId() throws Exception
+   {
+      connV11.connect(defUser, defPass);
+
+      subscribe(connV11, "sub1", "client");
+
+      sendMessage(getName());
+
+      ClientStompFrame frame = connV11.receiveFrame();
+
+      String messageID = frame.getHeader("message-id");
+      
+      System.out.println("Received message with id " + messageID);
+      
+      ack(connV11, "sub2", messageID);
+      
+      ClientStompFrame error = connV11.receiveFrame();
+      
+      System.out.println("Receiver error: " + error);
+      
+      connV11.disconnect();
+
+      //message should be still there
+      MessageConsumer consumer = session.createConsumer(queue);
+      Message message = consumer.receive(1000);
+      Assert.assertNotNull(message);
+   }
+
+   public void testAckWithWrongMessageId() throws Exception
+   {
+      connV11.connect(defUser, defPass);
+
+      subscribe(connV11, "sub1", "client");
+
+      sendMessage(getName());
+
+      ClientStompFrame frame = connV11.receiveFrame();
+
+      String messageID = frame.getHeader("message-id");
+      
+      System.out.println("Received message with id " + messageID);
+      
+      ack(connV11, "sub2", "someother");
+      
+      ClientStompFrame error = connV11.receiveFrame();
+      
+      System.out.println("Receiver error: " + error);
+      
+      connV11.disconnect();
+
+      //message should still there
+      MessageConsumer consumer = session.createConsumer(queue);
+      Message message = consumer.receive(1000);
+      Assert.assertNotNull(message);
+   }
+   
+
+   private void ack(StompClientConnection conn, String subId, String mid) throws IOException, InterruptedException
+   {
+      ClientStompFrame ackFrame = conn.createFrame("ACK");
+      ackFrame.addHeader("subscription", subId);
+      ackFrame.addHeader("message-id", mid);
+      
+      conn.sendFrame(ackFrame);
+   }
+
    private void nack(StompClientConnection conn, String subId, String mid) throws IOException, InterruptedException
    {
       ClientStompFrame ackFrame = conn.createFrame("NACK");