[jboss-cvs] JBossAS SVN: r68458 - in branches/Branch_4_2/testsuite: src/main/org/jboss/test/jca/test and 2 other directories.

jboss-cvs-commits at lists.jboss.org jboss-cvs-commits at lists.jboss.org
Thu Dec 20 11:53:00 EST 2007


Author: adrian at jboss.org
Date: 2007-12-20 11:53:00 -0500 (Thu, 20 Dec 2007)
New Revision: 68458

Added:
   branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteJMSDuringRollbackStressTestCase.java
   branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteSQLDuringRollbackStressTestCase.java
   branches/Branch_4_2/testsuite/src/resources/jca/executejmsrollback/
   branches/Branch_4_2/testsuite/src/resources/jca/executejmsrollback/test-jms-local-ds.xml
Modified:
   branches/Branch_4_2/testsuite/imports/sections/jca.xml
Log:
[JBAS-5080] - Add stress tests for JDBC and JMS activity competing with asynchronous rollback

Modified: branches/Branch_4_2/testsuite/imports/sections/jca.xml
===================================================================
--- branches/Branch_4_2/testsuite/imports/sections/jca.xml	2007-12-20 16:21:31 UTC (rev 68457)
+++ branches/Branch_4_2/testsuite/imports/sections/jca.xml	2007-12-20 16:53:00 UTC (rev 68458)
@@ -222,6 +222,23 @@
          </fileset>
       </jar>
       
+      <jar destfile="${build.lib}/jcaexecutejmsrollback.jar">
+         <fileset dir="${build.classes}">
+            <patternset refid="common.test.client.classes"/>
+            <include name="org/jboss/test/jca/test/ExecuteJMS*"/>
+            <include name="org/jboss/test/util/ejb/*"/>
+         </fileset>
+         <fileset dir="${build.resources}/jca/test">
+            <include name="**/*.*"/>
+         </fileset>
+         <fileset dir="${build.resources}/jca/executejmsrollback">
+            <include name="**/*.*"/>
+         </fileset>
+         <zipfileset src="${junit.junit.lib}/junit.jar">
+            <patternset refid="ejbrunner.set"/>
+         </zipfileset>
+      </jar>
+      
       <jar destfile="${build.lib}/jca-tests.jar">
          <fileset dir="${build.classes}">
             <patternset refid="common.test.client.classes"/>

Added: branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteJMSDuringRollbackStressTestCase.java
===================================================================
--- branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteJMSDuringRollbackStressTestCase.java	                        (rev 0)
+++ branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteJMSDuringRollbackStressTestCase.java	2007-12-20 16:53:00 UTC (rev 68458)
@@ -0,0 +1,367 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file 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.jca.test;
+
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.atomic.AtomicReference;
+
+import javax.jms.Connection;
+import javax.jms.ConnectionFactory;
+import javax.jms.JMSException;
+import javax.jms.Message;
+import javax.jms.MessageConsumer;
+import javax.jms.MessageProducer;
+import javax.jms.Queue;
+import javax.jms.Session;
+import javax.jms.TextMessage;
+import javax.naming.InitialContext;
+import javax.transaction.Transaction;
+import javax.transaction.TransactionManager;
+
+import junit.framework.Test;
+
+import org.jboss.logging.Logger;
+import org.jboss.test.JBossTestCase;
+import org.jboss.test.util.ejb.EJBTestCase;
+import org.jboss.tm.TransactionManagerLocator;
+import org.jboss.tm.TxUtils;
+
+/**
+ * ExecuteJMSDuringRollbackStressTestCase.
+ * 
+ * @author <a href="adrian at jboss.com">Adrian Brock</a>
+ * @version $Revision: 1.1 $
+ */
+public class ExecuteJMSDuringRollbackStressTestCase extends EJBTestCase
+{
+   protected final Logger log = Logger.getLogger(getClass());
+   
+   private ConnectionFactory cf;
+   
+   private Queue queue;
+   
+   private CountDownLatch latch;
+
+   private TransactionManager tm;
+   
+   private AtomicReference<Transaction> transaction = new AtomicReference<Transaction>();
+
+   public void testExecuteJMSDuringRollback() throws Throwable
+   {
+      queue = (Queue) new InitialContext().lookup("queue/testQueue");
+      cf = (ConnectionFactory) new InitialContext().lookup("java:TestJmsLocal");
+      setupQueue();
+      // This test takes too long to iterate 100 times because of the sleeps
+      for (int i = 0; i < getIterationCount()/10; ++i)
+      {
+         log.info("Running " + getName() + " iteration=" + i);
+         latch = new CountDownLatch(2);
+
+         ExecuteJMS executeJMS = new ExecuteJMS();
+         Thread thread1 = new Thread(executeJMS);
+         thread1.start();
+
+         Rollback rollback = new Rollback();
+         Thread thread2 = new Thread(rollback);
+         thread2.start();
+         
+         thread1.join();
+         thread2.join();
+         if (executeJMS.error != null)
+         {
+            log.error(executeJMS.error);
+            throw executeJMS.error;
+         }
+         if (rollback.error != null)
+         {
+            log.error(rollback.error);
+            throw rollback.error;
+         }
+         checkQueue();
+      }
+   }
+   
+   public class ExecuteJMS extends TestRunnable
+   {
+      private Connection c;
+      private MessageProducer p;
+      private Message m;
+      
+      public void setup() throws Throwable
+      {
+         tm.begin();
+         transaction.set(tm.getTransaction());
+         
+         try
+         {
+            c = cf.createConnection();
+            c.start();
+            Session s = c.createSession(true, Session.SESSION_TRANSACTED);
+            MessageConsumer r = s.createConsumer(queue);
+            r.receive(1000);
+            r.close();
+            p = s.createProducer(queue);
+            m = s.createTextMessage("100");
+         }
+         catch (Throwable t)
+         {
+            try
+            {
+               if (c != null)
+                  c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+            try
+            {
+               tm.rollback();
+            }
+            catch (Exception ignored)
+            {
+               log.warn("Ignored", ignored);
+            }
+            throw t;
+         }
+      }
+
+      public void test() throws Throwable
+      {
+         try
+         {
+            p.send(m);
+         }
+         catch (JMSException expected)
+         {
+         }
+         finally
+         {
+            try
+            {
+               if (c != null)
+                  c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+            try
+            {
+               synchronized (transaction)
+               {
+                  if (TxUtils.isActive(tm))
+                     tm.rollback();
+                  else
+                     tm.suspend();
+               }
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+         
+         tm.begin();
+         try
+         {
+            Connection c = cf.createConnection();
+            try
+            {
+               c.start();
+               c.createSession(true, Session.SESSION_TRANSACTED);
+            }
+            finally
+            {
+               try
+               {
+                  c.close();
+               }
+               catch (Exception ignored)
+               {
+               }
+            }
+         }
+         finally
+         {
+            tm.commit();
+         }
+      }
+   }
+   
+   public class Rollback extends TestRunnable
+   {
+      public void test() throws Throwable
+      {
+         Transaction tx = transaction.get();
+         if (tx != null)
+         {
+            try
+            {
+               synchronized (transaction)
+               {
+                  if (TxUtils.isActive(tx))
+                     tx.rollback();
+               }
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+      }
+   }
+   
+   protected void setupQueue() throws Throwable
+   {
+      log.info("setupQueue");
+      tm.begin();
+      try
+      {
+         Connection c = cf.createConnection();
+         try
+         {
+            c.start();
+            Session s = c.createSession(true, Session.SESSION_TRANSACTED);
+            MessageConsumer mc = s.createConsumer(queue);
+            while (mc.receive(1000) != null);
+            mc.close();
+            
+            MessageProducer p = s.createProducer(queue);
+            Message m = s.createTextMessage("101");
+            p.send(m);
+         }
+         finally
+         {
+            try
+            {
+               c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+      }
+      finally
+      {
+         tm.commit();
+      }
+   }
+   
+   protected void checkQueue() throws Throwable
+   {
+      log.info("checking queue");
+      tm.begin();
+      try
+      {
+         Connection c = cf.createConnection();
+         try
+         {
+            c.start();
+            Session s = c.createSession(true, Session.SESSION_TRANSACTED);
+            MessageConsumer mc = s.createConsumer(queue);
+            Message m = mc.receive(1000);
+            if (m == null || m instanceof TextMessage == false)
+               throw new RuntimeException("Expected one text message: " + m);
+            String value = ((TextMessage) m).getText();
+            if ("101".equals(value) == false)
+               throw new RuntimeException("Message should have text 101 got: " + value);
+            if (mc.receive(1000) != null)
+               throw new RuntimeException("Did not expect two messages");
+         }
+         catch (Throwable t)
+         {
+            log.error("Error checking queue", t);
+            throw t;
+         }
+         finally
+         {
+            try
+            {
+               c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+      }
+      finally
+      {
+         tm.rollback();
+      }
+   }
+   
+   public class TestRunnable implements Runnable
+   {
+      public Throwable error;
+      
+      public void setup() throws Throwable
+      {
+      }
+      
+      public void test() throws Throwable
+      {
+      }
+      
+      public void run()
+      {
+         try
+         {
+            setup();
+         }
+         catch (Throwable t)
+         {
+            error = t;
+            latch.countDown();
+            return;
+         }
+         latch.countDown();
+         try
+         {
+            latch.await();
+         }
+         catch (InterruptedException e)
+         {
+            log.warn("Ignored", e);
+         }
+         try
+         {
+            test();
+         }
+         catch (Throwable t)
+         {
+            error = t;
+         }
+      }
+   }
+   
+   protected void setUp() throws Exception
+   {
+      tm = TransactionManagerLocator.getInstance().locate();
+   }
+   
+   public ExecuteJMSDuringRollbackStressTestCase(String name)
+   {
+      super(name);
+   }
+
+   public static Test suite() throws Exception
+   {
+      return JBossTestCase.getDeploySetup(ExecuteJMSDuringRollbackStressTestCase.class, "jcaexecutejmsrollback.jar");
+   }
+}

Added: branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteSQLDuringRollbackStressTestCase.java
===================================================================
--- branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteSQLDuringRollbackStressTestCase.java	                        (rev 0)
+++ branches/Branch_4_2/testsuite/src/main/org/jboss/test/jca/test/ExecuteSQLDuringRollbackStressTestCase.java	2007-12-20 16:53:00 UTC (rev 68458)
@@ -0,0 +1,327 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file 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.jca.test;
+
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.atomic.AtomicReference;
+
+import javax.naming.InitialContext;
+import javax.sql.DataSource;
+import javax.transaction.Transaction;
+import javax.transaction.TransactionManager;
+
+import junit.framework.Test;
+
+import org.jboss.logging.Logger;
+import org.jboss.test.JBossTestCase;
+import org.jboss.test.util.ejb.EJBTestCase;
+import org.jboss.tm.TransactionManagerLocator;
+import org.jboss.tm.TxUtils;
+
+/**
+ * Abstract concurrent stress test.
+ * 
+ * @author <a href="adrian at jboss.com">Adrian Brock</a>
+ * @version $Revision: 65495 $
+ */
+public class ExecuteSQLDuringRollbackStressTestCase extends EJBTestCase
+{
+   protected final Logger log = Logger.getLogger(getClass());
+   
+   private DataSource dataSource;
+   
+   private CountDownLatch latch;
+
+   private TransactionManager tm;
+   
+   private AtomicReference<Transaction> transaction = new AtomicReference<Transaction>();
+
+   public void testExecuteSQLDuringRollback() throws Throwable
+   {
+      dataSource = (DataSource) new InitialContext().lookup("java:DefaultDS");
+      createDatabase();
+      for (int i = 0; i < getIterationCount(); ++i)
+      {
+         latch = new CountDownLatch(2);
+
+         ExecuteSQL executeSQL = new ExecuteSQL();
+         Thread thread1 = new Thread(executeSQL);
+         thread1.start();
+
+         Rollback rollback = new Rollback();
+         Thread thread2 = new Thread(rollback);
+         thread2.start();
+         
+         thread1.join();
+         thread2.join();
+         if (executeSQL.error != null)
+            throw executeSQL.error;
+         if (rollback.error != null)
+            throw rollback.error;
+         checkDatabase();
+      }
+   }
+   
+   public class ExecuteSQL extends TestRunnable
+   {
+      private Connection c;
+      private Statement stmt;
+      
+      public void setup() throws Throwable
+      {
+         tm.begin();
+         transaction.set(tm.getTransaction());
+         
+         try
+         {
+            c = dataSource.getConnection();
+            stmt = c.createStatement();
+            stmt.executeUpdate("insert into JCA_EXECUTE_ROLLBACK values ('101')");
+         }
+         catch (Throwable t)
+         {
+            try
+            {
+               if (c != null)
+                  c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+            try
+            {
+               tm.rollback();
+            }
+            catch (Exception ignored)
+            {
+               log.warn("Ignored", ignored);
+            }
+            throw t;
+         }
+      }
+
+      public void test() throws Throwable
+      {
+         try
+         {
+            stmt.executeUpdate("delete from JCA_EXECUTE_ROLLBACK where name='100'");
+         }
+         catch (SQLException expected)
+         {
+         }
+         finally
+         {
+            try
+            {
+               if (c != null)
+                  c.close();
+            }
+            catch (Exception ignored)
+            {
+            }
+            try
+            {
+               synchronized (transaction)
+               {
+                  if (TxUtils.isActive(tm))
+                     tm.rollback();
+                  else
+                     tm.suspend();
+               }
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+         
+         tm.begin();
+         try
+         {
+            Connection c = dataSource.getConnection();
+            try
+            {
+               Statement s = c.createStatement();
+               s.executeQuery("select * from JCA_EXECUTE_ROLLBACK");
+            }
+            finally
+            {
+               try
+               {
+                  c.close();
+               }
+               catch (Exception ignored)
+               {
+               }
+            }
+         }
+         finally
+         {
+            tm.commit();
+         }
+      }
+   }
+   
+   public class Rollback extends TestRunnable
+   {
+      public void test() throws Throwable
+      {
+         Transaction tx = transaction.get();
+         if (tx != null)
+         {
+            try
+            {
+               synchronized (transaction)
+               {
+                  if (TxUtils.isActive(tx))
+                     tx.rollback();
+               }
+            }
+            catch (Exception ignored)
+            {
+            }
+         }
+      }
+   }
+   
+   protected void createDatabase() throws Throwable
+   {
+      Connection c = dataSource.getConnection();
+      try
+      {
+         Statement stmt = c.createStatement();
+         try
+         {
+            stmt.executeUpdate("create table JCA_EXECUTE_ROLLBACK (name varchar(100))");
+         }
+         catch (SQLException ignored)
+         {
+         }
+         stmt.executeUpdate("delete from JCA_EXECUTE_ROLLBACK");
+         stmt.executeUpdate("insert into JCA_EXECUTE_ROLLBACK values ('100')");
+      }
+      finally
+      {
+         try
+         {
+            c.close();
+         }
+         catch (Exception ignored)
+         {
+         }
+      }
+   }
+   
+   protected void checkDatabase() throws Throwable
+   {
+      Connection c = dataSource.getConnection();
+      try
+      {
+         Statement stmt = c.createStatement();
+         ResultSet rs = stmt.executeQuery("select name from JCA_EXECUTE_ROLLBACK");
+         if (rs.next() == false)
+            throw new RuntimeException("Expected a first row");
+         String value = rs.getString(1);
+         if ("100".equals(value) == false)
+            throw new RuntimeException("Expected first row to be 100 got " + value);
+         if (rs.next())
+            throw new RuntimeException("Expected only one row");
+      }
+      catch (Throwable t)
+      {
+         log.error("Error checking database", t);
+         throw t;
+      }
+      finally
+      {
+         try
+         {
+            c.close();
+         }
+         catch (Exception ignored)
+         {
+         }
+      }
+   }
+   
+   public class TestRunnable implements Runnable
+   {
+      public Throwable error;
+      
+      public void setup() throws Throwable
+      {
+      }
+      
+      public void test() throws Throwable
+      {
+      }
+      
+      public void run()
+      {
+         try
+         {
+            setup();
+         }
+         catch (Throwable t)
+         {
+            error = t;
+            latch.countDown();
+            return;
+         }
+         latch.countDown();
+         try
+         {
+            latch.await();
+         }
+         catch (InterruptedException e)
+         {
+            log.warn("Ignored", e);
+         }
+         try
+         {
+            test();
+         }
+         catch (Throwable t)
+         {
+            error = t;
+         }
+      }
+   }
+   
+   protected void setUp() throws Exception
+   {
+      tm = TransactionManagerLocator.getInstance().locate();
+   }
+   
+   public ExecuteSQLDuringRollbackStressTestCase(String name)
+   {
+      super(name);
+   }
+
+   public static Test suite() throws Exception
+   {
+      return JBossTestCase.getDeploySetup(ExecuteSQLDuringRollbackStressTestCase.class, "jca-tests.jar");
+   }
+}

Added: branches/Branch_4_2/testsuite/src/resources/jca/executejmsrollback/test-jms-local-ds.xml
===================================================================
--- branches/Branch_4_2/testsuite/src/resources/jca/executejmsrollback/test-jms-local-ds.xml	                        (rev 0)
+++ branches/Branch_4_2/testsuite/src/resources/jca/executejmsrollback/test-jms-local-ds.xml	2007-12-20 16:53:00 UTC (rev 68458)
@@ -0,0 +1,29 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<connection-factories>
+
+  <mbean code="org.jboss.jms.jndi.JMSProviderLoader"
+	 name="jboss.mq:service=JMSProviderLoader,name=TestJMSLocalProvider">
+    <attribute name="ProviderName">TestJMSLocalProvider</attribute>
+    <attribute name="ProviderAdapterClass">
+      org.jboss.jms.jndi.JNDIProviderAdapter
+    </attribute>
+    <!-- The combined connection factory -->
+    <attribute name="FactoryRef">java:/ConnectionFactory</attribute>
+    <!-- The queue connection factory -->
+    <attribute name="QueueFactoryRef">java:/ConnectionFactory</attribute>
+    <!-- The topic factory -->
+    <attribute name="TopicFactoryRef">java:/ConnectionFactory</attribute>
+  </mbean>
+
+  <tx-connection-factory>
+    <jndi-name>TestJmsLocal</jndi-name>
+    <rar-name>jms-ra.rar</rar-name>
+    <connection-definition>org.jboss.resource.adapter.jms.JmsConnectionFactory</connection-definition>
+    <config-property name="SessionDefaultType" type="java.lang.String">javax.jms.Topic</config-property>
+    <config-property name="JmsProviderAdapterJNDI" type="java.lang.String">java:/TestJMSLocalProvider</config-property>
+    <max-pool-size>20</max-pool-size>
+    <application-managed-security/>
+  </tx-connection-factory>
+
+</connection-factories>




More information about the jboss-cvs-commits mailing list