* JBoss, Home of Professional Open Source

  * Copyright 2005, JBoss Inc., and individual contributors as indicated

  * 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.

  */

*

* A XATestBase

*

* @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a>

* @version <tt>$Revision: 1.1 $</tt>

*

* $Id$

*

*/

   // Constants -----------------------------------------------------

   // Static --------------------------------------------------------

  

   // Attributes ----------------------------------------------------

   protected InitialContext initialContext;

  

   protected JBossConnectionFactory cf;

  

   protected Destination queue;

  

   protected TransactionManager tm;

  

   protected Transaction suspendedTx;

  

   protected ServiceContainer sc;

  

   protected boolean useArjuna;

   // Constructors --------------------------------------------------

   public XATestBase(String name)

   {

      super(name);

   }

  

  

   // TestCase overrides -------------------------------------------

   public void setUp() throws Exception

   {

      super.setUp();

     

      if (useArjuna)

      {

         ServerManagement.start("all,-transaction,jbossjta");

      }

      else

      {     

         ServerManagement.start("all");

      }

     

      //Also need a local tx mgr if test is running remote

      if (ServerManagement.isRemote())

      {

         if (useArjuna)

         {

            sc = new ServiceContainer("all,-transaction,jbossjta");

         }

         else

         {

            sc = new ServiceContainer("transaction");

         }

        

         //Don't drop the tables again!

         sc.start(false);

      }

     

      initialContext = new InitialContext(ServerManagement.getJNDIEnvironment());

     

      InitialContext localIc = new InitialContext(InVMInitialContextFactory.getJNDIEnvironment());

     

      cf = (JBossConnectionFactory)initialContext.lookup("/ConnectionFactory");

           

      tm = (TransactionManager)localIc.lookup(ServiceContainer.TRANSACTION_MANAGER_JNDI_NAME);

     

      if (useArjuna)

      {

         log.info("tm is " + tm.getClass().getName());

         assertTrue(tm instanceof TransactionManagerImple);

      }

      else

      {

         assertTrue(tm instanceof org.jboss.tm.TxManager);

      }

   

      ServerManagement.undeployQueue("Queue");

     

      ServerManagement.deployQueue("Queue");

     

      queue = (Destination)initialContext.lookup("/queue/Queue");

     

      drainDestination(cf, queue);

      if (!ServerManagement.isRemote())

      {

         suspendedTx = tm.suspend();

      }

   }

   public void tearDown() throws Exception

   {

      ServerManagement.undeployQueue("Queue");

     

      if (TxUtils.isUncommitted(tm))

      {

         //roll it back

         try

         {

            tm.rollback();

         }

         catch (Throwable ignore)

         {

            //The connection will probably be closed so this may well throw an exception

         }

      }

      if (tm.getTransaction() != null)

      {

         Transaction tx = tm.suspend();

         if (tx != null)

            log.warn("Transaction still associated with thread " + tx + " at status " + TxUtils.getStatusAsString(tx.getStatus()));

      }   

     

      if (suspendedTx != null)

      {

         tm.resume(suspendedTx);

      }

     

      if (ServerManagement.isRemote())

      {

         sc.stop();

      }

     

      ServerManagement.stop();

      super.tearDown();

   }

  

  

   // Public --------------------------------------------------------

  

  

  

   //See http://jira.jboss.com/jira/browse/JBMESSAGING-638

   public void testResourceManagerMemoryLeakOnCommit() throws Exception

   {

      XAConnection xaConn = null;

     

      try

      {

         xaConn = cf.createXAConnection();

        

         JBossConnection jbConn = (JBossConnection)xaConn;

        

         ClientConnectionDelegate del = (ClientConnectionDelegate)jbConn.getDelegate();

        

         ConnectionState state = (ConnectionState)del.getState();

        

         ResourceManager rm = state.getResourceManager();

        

         XASession xaSession = xaConn.createXASession();

        

         xaConn.start();

        

         XAResource res = xaSession.getXAResource();

        

         XAResource dummy = new DummyXAResource();

        

         for (int i = 0; i < 100; i++)

         {

           

            tm.begin();

                    

            Transaction tx = tm.getTransaction();

           

            tx.enlistResource(res);

           

            tx.enlistResource(dummy);

           

            assertEquals(1, rm.size());

           

            tx.delistResource(res, XAResource.TMSUCCESS);

           

            tx.delistResource(dummy, XAResource.TMSUCCESS);

           

            tm.commit();

         }                 

        

         assertEquals(1, rm.size());

        

         xaConn.close();

        

         xaConn = null;

        

         assertEquals(0, rm.size());

      }

      finally

      {

         if (xaConn != null)

         {

            xaConn.close();

         }

      }

   }

  

   //See http://jira.jboss.com/jira/browse/JBMESSAGING-638

   public void testResourceManagerMemoryLeakOnRollback() throws Exception

   {

      XAConnection xaConn = null;

     

      try

      {

         xaConn = cf.createXAConnection();

        

         JBossConnection jbConn = (JBossConnection)xaConn;

        

         ClientConnectionDelegate del = (ClientConnectionDelegate)jbConn.getDelegate();

        

         ConnectionState state = (ConnectionState)del.getState();

        

         ResourceManager rm = state.getResourceManager();

        

         XASession xaSession = xaConn.createXASession();

        

         xaConn.start();

        

         XAResource res = xaSession.getXAResource();

        

         XAResource dummy = new DummyXAResource();

        

         for (int i = 0; i < 100; i++)

         {           

            tm.begin();

                    

            Transaction tx = tm.getTransaction();

           

            tx.enlistResource(res);

           

            tx.enlistResource(dummy);

           

            assertEquals(1, rm.size());

           

            tx.delistResource(res, XAResource.TMSUCCESS);

           

            tx.delistResource(dummy, XAResource.TMSUCCESS);

           

            tm.rollback();

         }                 

        

         assertEquals(1, rm.size());

        

         xaConn.close();

        

         xaConn = null;

        

         assertEquals(0, rm.size());

      }

      finally

      {

         if (xaConn != null)

         {

            xaConn.close();

         }

      }

   }

  

   //http://jira.jboss.com/jira/browse/JBMESSAGING-721

   public void testConvertFromLocalTx() throws Exception

   {

      Connection conn = null;

     

      XAConnection xaConn = null;

     

      try

      {

     

         //First send some messages to a queue

        

         conn = cf.createConnection();

        

         Session sessSend = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);

        

         MessageProducer prod = sessSend.createProducer(queue);

        

         TextMessage tm1 = sessSend.createTextMessage("message1");

        

         TextMessage tm2 = sessSend.createTextMessage("message2");

        

         prod.send(tm1);

        

         prod.send(tm2);

        

        

         xaConn = cf.createXAConnection();

        

         XASession xaSession = xaConn.createXASession();

        

         xaConn.start();

        

         MessageConsumer cons = xaSession.createConsumer(queue);

        

         //Receive the two messages outside of a transaction

        

         TextMessage rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         TextMessage rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         Message rm3 = cons.receive(1000);

        

         assertNull(rm3);

        

         //Now we enlist the session in an xa transaction

        

         XAResource res = xaSession.getXAResource();

        

         tm.begin();

        

         Transaction tx = tm.getTransaction();

         tx.enlistResource(res);

        

         //This should cause the work done previously to be converted into work done in the xa transaction

         //this is what an MDB does

         //There is a difficulty in transactional delivery with an MDB.

         //The message is received from the destination and then sent to the mdb container so

         //it can call onMessage.

         //For transactional delivery the receipt of the message should be in a transaction but by the time

         //the mdb container is invoked the message has already been received it is too late - the message

         //has already been received and passed on (see page 199 (chapter 5 JMS and Transactions, section "Application Server Integration"

         //of Mark Little's book Java Transaction processing

         //for a discussion of how different app serves deal with this)

         //The way jboss messaging (and jboss mq) deals with this is to convert any work done

         //prior to when the xasession is enlisted in the tx, into work done in the xa tx

        

         tx.delistResource(res, XAResource.TMSUCCESS);

        

         //Now rollback the tx - this should cause redelivery of the two messages

         tx.rollback();

        

         rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         rm3 = cons.receive(1000);

        

         assertNull(rm3);

      }

      finally

      {        

         if (conn != null)

         {

            conn.close();

         }

        

         if (xaConn != null)

         {

            xaConn.close();

         }

      }

   }

  

   //http://jira.jboss.com/jira/browse/JBMESSAGING-721

   public void testTransactionIdSetAfterCommit() throws Exception

   {

      Connection conn = null;

     

      XAConnection xaConn = null;

     

      try

      {

     

         //First send some messages to a queue

        

         conn = cf.createConnection();

        

         Session sessSend = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);

        

         MessageProducer prod = sessSend.createProducer(queue);

        

         TextMessage tm1 = sessSend.createTextMessage("message1");

        

         TextMessage tm2 = sessSend.createTextMessage("message2");

        

         prod.send(tm1);

        

         prod.send(tm2);

        

        

         xaConn = cf.createXAConnection();

        

         XASession xaSession = xaConn.createXASession();

        

         xaConn.start();

        

         MessageConsumer cons = xaSession.createConsumer(queue);

        

         //Now we enlist the session in an xa transaction

        

         XAResource res = xaSession.getXAResource();

        

         tm.begin();

        

         Transaction tx = tm.getTransaction();

         tx.enlistResource(res);

        

         tx.delistResource(res, XAResource.TMSUCCESS);

        

         //Then we do a commit

         tm.commit();

                             

         //Then we receive the messages outside the tx

        

         TextMessage rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         TextMessage rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         Message rm3 = cons.receive(1000);

        

         assertNull(rm3);

        

         //And enlist again - this should convert the work done in the local tx

         //into the global branch

        

         tx = tm.getTransaction();

        

         tm.begin();

        

         tx = tm.getTransaction();

         tx.enlistResource(res);

        

         tx.delistResource(res, XAResource.TMSUCCESS);        

              

         //Now rollback the tx - this should cause redelivery of the two messages

         tx.rollback();

        

         rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         rm3 = cons.receive(1000);

        

         assertNull(rm3);

      }

      finally

      {        

         if (conn != null)

         {

            conn.close();

         }

        

         if (xaConn != null)

         {

            xaConn.close();

         }

      }

   }

  

   //http://jira.jboss.com/jira/browse/JBMESSAGING-721

   public void testTransactionIdSetAfterRollback() throws Exception

   {

      Connection conn = null;

     

      XAConnection xaConn = null;

     

      try

      {

     

         //First send some messages to a queue

        

         conn = cf.createConnection();

        

         Session sessSend = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);

        

         MessageProducer prod = sessSend.createProducer(queue);

        

         TextMessage tm1 = sessSend.createTextMessage("message1");

        

         TextMessage tm2 = sessSend.createTextMessage("message2");

        

         prod.send(tm1);

        

         prod.send(tm2);

        

        

         xaConn = cf.createXAConnection();

        

         XASession xaSession = xaConn.createXASession();

        

         xaConn.start();

        

         MessageConsumer cons = xaSession.createConsumer(queue);

        

         //Now we enlist the session in an xa transaction

        

         XAResource res = xaSession.getXAResource();

        

         tm.begin();

        

         Transaction tx = tm.getTransaction();

         tx.enlistResource(res);

         tx.delistResource(res, XAResource.TMSUCCESS);

        

         //Then we do a rollback

         tm.rollback();                

        

         //Then we receive the messages outside the global tx

        

         TextMessage rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         TextMessage rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         Message rm3 = cons.receive(1000);

        

         assertNull(rm3);

        

         tm.begin();

        

         //And enlist again - the work should then be converted into the global tx branch

        

         tx = tm.getTransaction();

        

         tx.enlistResource(res);

        

         tx.delistResource(res, XAResource.TMSUCCESS);

              

         //Now rollback the tx - this should cause redelivery of the two messages

         tx.rollback();

        

         rm1 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm1);

        

         assertEquals("message1", rm1.getText());

        

         rm2 = (TextMessage)cons.receive(1000);

        

         assertNotNull(rm2);

        

         assertEquals("message2", rm2.getText());

        

         rm3 = cons.receive(1000);

        

         assertNull(rm3);

      }

      finally

      {        

         if (conn != null)

         {

            conn.close();

         }

        

         if (xaConn != null)

         {

            xaConn.close();

         }

      }

   }

  

   // See http://jira.jboss.org/jira/browse/JBMESSAGING-825

   // Need to test that ids with trailing zeros are dealt with properly - sybase has the habit

   // of truncating trailing zeros in varbinary columns

   public void testXidsWithTrailingZeros() throws Exception

   {

      if (!ServerManagement.isRemote())

      {

         return;

      }

     

      XAConnection conn1 = null;

     

      try

      {     

         conn1 = cf.createXAConnection();

  

         XASession sess1 = conn1.createXASession();

  

         XAResource res1 = sess1.getXAResource();

        

         byte[] branchQualifier = new byte[] { 1, 2, 3, 4, 5, 6, 0, 0, 0, 0 };

        

         byte[] globalTxId = new byte[] { 6, 5, 4, 3, 2, 1, 0, 0, 0, 0 };

                 

         Xid trailing = new MessagingXid(branchQualifier, 12435, globalTxId);

        

         res1.start(trailing, XAResource.TMNOFLAGS);

  

         MessageProducer prod1 = sess1.createProducer(queue);

  

         TextMessage tm1 = sess1.createTextMessage("testing1");

  

         prod1.send(tm1);

  

         res1.end(trailing, XAResource.TMSUCCESS);

  

  

         res1.prepare(trailing);

  

         //Now "crash" the server

  

         ServerManagement.stopServerPeer();

  

         ServerManagement.startServerPeer();

  

         ServerManagement.deployQueue("Queue");

  

  

         XAResource res = cf.createXAConnection().createXASession().getXAResource();

  

         Xid[] xids = res.recover(XAResource.TMSTARTRSCAN);