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

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

Author: clebert.suconic at jboss.com
Date: 2007-10-04 13:33:36 -0400 (Thu, 04 Oct 2007)
New Revision: 65854

Modified:
   branches/JBPAPP_4_2_0_GA_CP/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java
Log:
Flushing queue before testcase

Modified: branches/JBPAPP_4_2_0_GA_CP/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java
===================================================================
--- branches/JBPAPP_4_2_0_GA_CP/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java	2007-10-04 17:29:02 UTC (rev 65853)
+++ branches/JBPAPP_4_2_0_GA_CP/testsuite/src/main/org/jboss/test/timer/test/BasicTimerUnitTestCase.java	2007-10-04 17:33:36 UTC (rev 65854)
@@ -27,13 +27,17 @@
 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;
@@ -310,13 +314,17 @@
    public void testMDBTimer() throws Exception
    {
       InitialContext ctx = new InitialContext();
+
+      Queue queueA = (Queue) ctx.lookup("queue/A");
+      Queue queueB = (Queue) ctx.lookup("queue/B");
+      
+      flush(queueA);
+      flush(queueB);
+      
       QueueConnectionFactory factory = (QueueConnectionFactory) ctx.lookup("ConnectionFactory");
       QueueConnection queConn = factory.createQueueConnection();
       queConn.start();
 
-      Queue queueA = (Queue) ctx.lookup("queue/A");
-      Queue queueB = (Queue) ctx.lookup("queue/B");
-
       QueueSession session = queConn.createQueueSession(false, QueueSession.AUTO_ACKNOWLEDGE);
       QueueSender sender = session.createSender(queueA);
       TextMessage message = session.createTextMessage();
@@ -341,7 +349,29 @@
       session.close();
       queConn.close();
    }
+   
+   protected void flush(Queue queue) throws Exception
+   {
+	   InitialContext ctx = new InitialContext();
+	   ConnectionFactory factory = (ConnectionFactory) ctx.lookup("ConnectionFactory");
+	   Connection conn = factory.createConnection();
+	   
+	   conn.start();
+	   
+	   Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
+	   
+	   try
+	   {
+		   MessageConsumer cons = sess.createConsumer(queue);
+		   while (cons.receive(1000) != null);
+	   }
+	   finally
+	   {
+		   conn.close();
+	   }
 
+   }
+
    /** Test an mdb that creates a timer in its ejbCreate method
     * @throws Exception
     */