/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jboss.test.messaging.jms;

import static java.util.concurrent.TimeUnit.MILLISECONDS;

import java.util.concurrent.CountDownLatch;

import javax.jms.Connection;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageListener;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.jms.TextMessage;

/**
 * 
 * A DeliveryOrderTest
 *
 * @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a>
 * @version <tt>$Revision: 1.1 $</tt>
 *
 * $Id$
 *
 */
public class DeliveryOrderTest extends JMSTestCase
{

   public DeliveryOrderTest(String name)
   {
      super(name);
   }
     
   public void testOutOfOrder() throws Exception
   {
      Connection conn = null;
      try
      {
         conn = cf.createConnection();
         
         Session sess = conn.createSession(true, Session.SESSION_TRANSACTED);
         
         Session sess2 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
         
         MessageProducer prod = sess.createProducer(queue1);
         
         MessageConsumer cons = sess2.createConsumer(queue1);
         
         CountDownLatch latch = new CountDownLatch(1);
         
         final int NUM_MESSAGES = 1000;
                  
         MyListener listener = new MyListener(latch, NUM_MESSAGES);
         
         cons.setMessageListener(listener);
         
         conn.start();
         
         for (int i = 0; i < NUM_MESSAGES; i++)
         {
            TextMessage tm = sess.createTextMessage("message" + i);
            
            prod.send(tm);
            
            if (i % 10 == 0)
            {
               sess.commit();
            }
         }
         
         // need extra commit for cases in which the last message index is not a multiple of 10
         sess.commit();

         latch.await(20000, MILLISECONDS);
         
         if (listener.failed)
         {
            fail("listener failed: " + listener.getError());
         }
                  
      }
      finally
      {
         if (conn != null)
         {
            conn.close();
         }
      }
   }
   
   class MyListener implements MessageListener
   {
      private int c;
      private int num;
      private CountDownLatch latch;
      private volatile boolean failed;
      private String error;

      MyListener(CountDownLatch latch, int num)
      {
         this.latch = latch;
         this.num = num;
      }

      public void onMessage(Message msg)
      {
         // preserve the first error
         if (failed)
         {
            return;
         }

         try
         {
            TextMessage tm = (TextMessage)msg;
            
            if (!("message" + c).equals(tm.getText()))
            {
               // Failed
               failed = true;
               setError("Listener was supposed to get " + ("message" + c) +
                        " but got " + tm.getText());
               latch.countDown();
            }
            
            c++;
            
            if (c == num)
            {
               latch.countDown();
            }
         }
         catch (JMSException e)
         {
            e.printStackTrace();
            
            // Failed
            failed = true;
            setError("Listener got exception " + e.toString());
            latch.countDown();
         }
      }

      public synchronized String getError()
      {
         return error;
      }

      private synchronized void setError(String s)
      {
         error = s;
      }

      
   }

}