[jboss-cvs] JBossAS SVN: r67341 - branches/JBPAPP_4_2/testsuite/src/main/org/jboss/test/timer/test.

[jboss-cvs-commits at lists.jboss.org]

Author: clebert.suconic at jboss.com
Date: 2007-11-21 11:24:55 -0500 (Wed, 21 Nov 2007)
New Revision: 67341

Modified:
   branches/JBPAPP_4_2/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java
Log:
JBPAPP-417 - drainQueue before usage

Modified: branches/JBPAPP_4_2/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java
===================================================================
--- branches/JBPAPP_4_2/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java	2007-11-21 16:01:43 UTC (rev 67340)
+++ branches/JBPAPP_4_2/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java	2007-11-21 16:24:55 UTC (rev 67341)
@@ -27,18 +27,23 @@
 import java.util.HashMap;
 import javax.ejb.EJBHome;
 import javax.ejb.NoSuchObjectLocalException;
+import javax.jms.Connection;
+import javax.jms.ConnectionFactory;
 import javax.jms.Message;
+import javax.jms.MessageConsumer;
 import javax.jms.Queue;
 import javax.jms.QueueConnection;
 import javax.jms.QueueConnectionFactory;
 import javax.jms.QueueReceiver;
 import javax.jms.QueueSender;
 import javax.jms.QueueSession;
+import javax.jms.Session;
 import javax.jms.TextMessage;
 import javax.naming.InitialContext;
 import javax.naming.NamingException;
 
 import junit.framework.Test;
+
 import org.jboss.test.JBossTestCase;
 import org.jboss.test.timer.interfaces.TimerEntity;
 import org.jboss.test.timer.interfaces.TimerEntityHome;
@@ -309,6 +314,8 @@
     */
    public void testMDBTimer() throws Exception
    {
+      drainQueue("queue/A");
+      drainQueue("queue/B");
       InitialContext ctx = new InitialContext();
       QueueConnectionFactory factory = (QueueConnectionFactory) ctx.lookup("ConnectionFactory");
       QueueConnection queConn = factory.createQueueConnection();
@@ -437,6 +444,36 @@
    }
 
 
+   // Emptys out all the messages in a queue
+   protected void drainQueue(String queueName) throws Exception
+   {
+       
+      Connection conn = null;
+      
+      try
+      {
+          InitialContext ctx = new InitialContext();
+          ConnectionFactory cf = (ConnectionFactory)ctx.lookup("ConnectionFactory");
+          conn = cf.createConnection();
+          Session session = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
+          Queue queue = (Queue) ctx.lookup(queueName);
+    
+          MessageConsumer receiver = session.createConsumer(queue);
+          Message message = receiver.receive(50);
+          do
+          {
+              message = receiver.receive(200);
+          } while (message != null);
+      }
+      finally
+      {
+          try {conn.close();} catch (Exception ignored){}
+      }
+      
+      
+   }
+
+   
    private EJBHome getEJBHome(String pJNDIName)
       throws NamingException
    {