[jbosscache-commits] JBoss Cache SVN: r4310 - in core/branches/Branch_JBossCache_1_4_0: src/org/jboss/cache/transaction and 1 other directories.

[jbosscache-commits at lists.jboss.org]

Author: manik.surtani at jboss.com
Date: 2007-08-23 12:46:05 -0400 (Thu, 23 Aug 2007)
New Revision: 4310

Added:
   core/branches/Branch_JBossCache_1_4_0/tests/functional/org/jboss/cache/transaction/StatusUnknownTest.java
Modified:
   core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/interceptors/OrderedSynchronizationHandler.java
   core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/transaction/DummyTransaction.java
Log:
Unit test for JBCACHE-1167

Modified: core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/interceptors/OrderedSynchronizationHandler.java
===================================================================
--- core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/interceptors/OrderedSynchronizationHandler.java	2007-08-22 14:35:57 UTC (rev 4309)
+++ core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/interceptors/OrderedSynchronizationHandler.java	2007-08-23 16:46:05 UTC (rev 4310)
@@ -77,6 +77,7 @@
    }
 
    public void afterCompletion(int status) {
+      RuntimeException exceptionInAfterCompletion = null;
       for(Iterator it=handlers.iterator(); it.hasNext();) {
          Synchronization sync=(Synchronization)it.next();
          try {
@@ -84,11 +85,15 @@
          }
          catch(Throwable t) {
             log.error("failed calling afterCompletion() on " + sync, t);
+            exceptionInAfterCompletion = (RuntimeException) t;
          }
       }
 
       // finally unregister us from the hashmap
       instances.remove(tx);
+
+      // throw the exception so the TM can deal with it.
+      if (exceptionInAfterCompletion != null) throw exceptionInAfterCompletion;
    }
 
    public String toString() {

Modified: core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/transaction/DummyTransaction.java
===================================================================
--- core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/transaction/DummyTransaction.java	2007-08-22 14:35:57 UTC (rev 4309)
+++ core/branches/Branch_JBossCache_1_4_0/src/org/jboss/cache/transaction/DummyTransaction.java	2007-08-23 16:46:05 UTC (rev 4310)
@@ -230,7 +230,7 @@
    }
 
 
-   boolean notifyBeforeCompletion() {
+   protected boolean notifyBeforeCompletion() {
       boolean retval=true;
       List tmp;
 
@@ -254,7 +254,7 @@
       return retval;
    }
 
-   void notifyAfterCompletion(int status) {
+   protected void notifyAfterCompletion(int status) {
       List tmp;
 
       synchronized(participants) {

Added: core/branches/Branch_JBossCache_1_4_0/tests/functional/org/jboss/cache/transaction/StatusUnknownTest.java
===================================================================
--- core/branches/Branch_JBossCache_1_4_0/tests/functional/org/jboss/cache/transaction/StatusUnknownTest.java	                        (rev 0)
+++ core/branches/Branch_JBossCache_1_4_0/tests/functional/org/jboss/cache/transaction/StatusUnknownTest.java	2007-08-23 16:46:05 UTC (rev 4310)
@@ -0,0 +1,146 @@
+package org.jboss.cache.transaction;
+
+import junit.framework.TestCase;
+import org.jboss.cache.TreeCache;
+import org.jboss.cache.DummyTransactionManagerLookup;
+import org.jboss.cache.TransactionManagerLookup;
+
+import javax.transaction.TransactionManager;
+import javax.transaction.Transaction;
+import javax.transaction.SystemException;
+import javax.transaction.NotSupportedException;
+import javax.transaction.Status;
+import javax.transaction.RollbackException;
+import javax.transaction.Synchronization;
+import javax.naming.Context;
+import javax.naming.InitialContext;
+import javax.naming.NamingException;
+import java.util.Properties;
+import java.util.List;
+import java.util.LinkedList;
+import java.util.Iterator;
+
+/**
+ * This test checks how the cache behaves when a JTA STATUS_UNKNOWN is passed in to the cache during afterCompletion().
+ *
+ * @author <a href="mailto:manik at jboss.org">Manik Surtani</a>
+ */
+public class StatusUnknownTest extends TestCase
+{
+   private TreeCache cache;
+   private TransactionManager tm;
+
+   protected void setUp() throws Exception
+   {
+      cache = new TreeCache();
+      cache.setTransactionManagerLookupClass(HeuristicFailingDummyTransactionManagerLookup.class.getName());
+      cache.startService();
+      tm = cache.getTransactionManager();
+   }
+
+   protected void tearDown()
+   {
+      cache.stopService();
+   }
+
+   public void testStatusUnknown() throws Exception
+   {
+      tm.begin();
+      cache.put("/a/b/c", "k", "v");
+      tm.commit();
+   }
+
+   public static class HeuristicFailingDummyTransactionManager extends DummyTransactionManager
+   {
+      public void begin() throws SystemException, NotSupportedException
+      {
+         super.begin();
+
+         Transaction tx = new HeuristicFailingDummyTransaction(this);
+         setTransaction(tx);
+      }
+
+      public static DummyTransactionManager getInstance() 
+      {
+         if(instance == null)
+         {
+            instance=new HeuristicFailingDummyTransactionManager();
+            try
+            {
+               Properties p=new Properties();
+               p.put(Context.INITIAL_CONTEXT_FACTORY, "org.jboss.cache.transaction.DummyContextFactory");
+               Context ctx=new InitialContext(p);
+               ctx.bind("java:/TransactionManager", instance);
+               ctx.bind("UserTransaction", new DummyUserTransaction(instance));
+            }
+            catch(NamingException e)
+            {
+               log.error("binding of DummyTransactionManager failed", e);
+            }
+         }
+         return instance;
+      }
+   }
+
+   public static class HeuristicFailingDummyTransaction extends DummyTransaction
+   {
+      public HeuristicFailingDummyTransaction(DummyBaseTransactionManager mgr)
+      {
+         super(mgr);
+      }
+
+      public void commit() throws RollbackException
+      {
+         status= Status.STATUS_PREPARING;
+         try
+         {
+            notifyBeforeCompletion();
+            notifyAfterCompletion(Status.STATUS_UNKNOWN);
+         }
+         finally
+         {
+            // Disassociate tx from thread.
+            tm_.setTransaction(null);
+         }
+      }
+
+      protected void notifyAfterCompletion(int status)
+      {
+         List tmp;
+
+         synchronized(participants)
+         {
+            tmp=new LinkedList(participants);
+         }
+
+         for(Iterator it=tmp.iterator(); it.hasNext();)
+         {
+            Synchronization s=(Synchronization)it.next();
+            try
+            {
+               s.afterCompletion(status);
+            }
+            catch(Throwable t)
+            {
+               throw (RuntimeException) t;
+            }
+         }
+
+         synchronized(participants)
+         {
+            participants.clear();
+         }
+      }
+   }
+
+   public static class HeuristicFailingDummyTransactionManagerLookup implements TransactionManagerLookup
+   {
+
+      public TransactionManager getTransactionManager() throws Exception
+      {
+         return HeuristicFailingDummyTransactionManager.getInstance();
+      }
+   }
+}
+
+