/*
* 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 in the distribution for a
* full listing of individual contributors.
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU Lesser General Public License, v. 2.1.
* This program is distributed in the hope that it will be useful, but WITHOUT A
* 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,
* v.2.1 along with this distribution; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*
* (C) 2005-2006,
* @author JBoss Inc.
*/
package com.hp.mwtests.ts.jta.subordinate;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import javax.resource.spi.XATerminator;
import javax.transaction.RollbackException;
import javax.transaction.Transaction;
import javax.transaction.xa.XAException;
import javax.transaction.xa.XAResource;
import javax.transaction.xa.Xid;
import org.junit.Test;
import com.arjuna.ats.arjuna.common.Uid;
import com.arjuna.ats.arjuna.coordinator.ActionManager;
import com.arjuna.ats.arjuna.coordinator.TwoPhaseOutcome;
import com.arjuna.ats.internal.jta.resources.spi.XATerminatorExtensions;
import com.arjuna.ats.internal.jta.transaction.arjunacore.jca.SubordinateTransaction;
import com.arjuna.ats.internal.jta.transaction.arjunacore.jca.SubordinationManager;
import com.arjuna.ats.internal.jta.transaction.arjunacore.subordinate.jca.TransactionImple;
import com.arjuna.ats.jta.xa.XidImple;
public class SubordinateTestCase
{
// This test class is subclassed by the JTAX version of the tests, so we isolate
// the module specific tx creation code to this function, which then gets overridden.
public SubordinateTransaction createTransaction() {
return new TransactionImple(0); // implicit begin
}
@Test
public void testCleanupCommit () throws Exception
{
for (int i = 0; i < 1000; i++)
{
final SubordinateTransaction tm = createTransaction();
assertEquals(TwoPhaseOutcome.PREPARE_READONLY, tm.doPrepare());
// don't call commit for read only case.
}
assertEquals(ActionManager.manager().getNumberOfInflightTransactions(), 0);
}
@Test
public void testCleanupRollback () throws Exception
{
for (int i = 0; i < 1000; i++)
{
final SubordinateTransaction tm = createTransaction();
tm.doRollback();
}
assertEquals(ActionManager.manager().getNumberOfInflightTransactions(), 0);
}
@Test
public void testCleanupSecondPhaseRollback () throws Exception
{
for (int i = 0; i < 1000; i++)
{
final SubordinateTransaction tm = createTransaction();
assertEquals(TwoPhaseOutcome.PREPARE_READONLY, tm.doPrepare());
// don't call rollback for read only case
}
assertEquals(ActionManager.manager().getNumberOfInflightTransactions(), 0);
}
@Test
public void testCleanupOnePhaseCommit () throws Exception
{
for (int i = 0; i < 1000; i++)
{
final SubordinateTransaction tm = createTransaction();
tm.doOnePhaseCommit();
}
assertEquals(ActionManager.manager().getNumberOfInflightTransactions(), 0);
}
/////////////
@Test
public void testOnePhaseCommitSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.doOnePhaseCommit();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testOnePhaseCommitSyncViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
xaTerminator.commit(xid, true);
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testOnePhaseCommitSyncWithRollbackOnly() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.setRollbackOnly();
try {
tm.doOnePhaseCommit();
fail("did not get expected rollback exception");
} catch(RollbackException e) {
// expected - we tried to commit a rollbackonly tx.
}
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testOnePhaseCommitSyncWithRollbackOnlyViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
t.setRollbackOnly();
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
try {
xaTerminator.commit(xid, true);
((TransactionImple)t).doOnePhaseCommit();
fail("did not get expected rollback exception");
} catch(XAException e) {
assertEquals("javax.transaction.RollbackException", e.getCause().getClass().getName());
assertEquals(XAException.XA_RBROLLBACK, e.errorCode);
// expected - we tried to commit a rollbackonly tx.
}
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
@Test
public void testRollbackSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.doRollback();
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testRollbackSyncViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
xaTerminator.rollback(xid);
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
@Test
public void testTwoPhaseCommitSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
assertEquals(TwoPhaseOutcome.PREPARE_READONLY, tm.doPrepare());
// don't call commit for read only case
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
assertEquals(XAResource.XA_RDONLY, xaTerminator.prepare(xid));
// don't call commit for read only case
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithXAOK() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
final TestXAResource xaResource = new TestXAResource();
xaResource.setPrepareReturnValue(XAResource.XA_OK);
tm.enlistResource(xaResource);
assertEquals(TwoPhaseOutcome.PREPARE_OK, tm.doPrepare());
tm.doCommit();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithXAOKViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final TestXAResource xaResource = new TestXAResource();
xaResource.setPrepareReturnValue(XAResource.XA_OK);
t.enlistResource(xaResource);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
assertEquals(XAResource.XA_OK, xaTerminator.prepare(xid));
xaTerminator.commit(xid, false);
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithRollbackOnly() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.setRollbackOnly();
assertEquals(TwoPhaseOutcome.PREPARE_NOTOK, tm.doPrepare());
tm.doRollback();
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithRollbackOnlyViaXATerminator() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
t.setRollbackOnly();
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
try {
xaTerminator.prepare(xid);
} catch(XAException e) {
assertEquals(XAException.XA_RBROLLBACK, e.errorCode);
// expected - we tried to prepare a rollbackonly tx.
}
// no need to call rollback - the XA_RBROLLBACK code indicates its been done.
assertFalse(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
/////////////
@Test
public void testOnePhaseCommitSyncWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.doBeforeCompletion();
tm.doOnePhaseCommit();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testOnePhaseCommitSyncViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
xaTerminator.commit(xid, true);
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testOnePhaseCommitSyncWithRollbackOnlyWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.setRollbackOnly();
tm.doBeforeCompletion();
try {
tm.doOnePhaseCommit();
fail("did not get expected rollback exception");
} catch(RollbackException e) {
// expected - we tried to commit a rollbackonly tx.
}
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testOnePhaseCommitSyncWithRollbackOnlyViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
t.setRollbackOnly();
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
try {
xaTerminator.commit(xid, true);
((TransactionImple)t).doOnePhaseCommit();
fail("did not get expected rollback exception");
} catch(XAException e) {
assertEquals("javax.transaction.RollbackException", e.getCause().getClass().getName());
assertEquals(XAException.XA_RBROLLBACK, e.errorCode);
// expected - we tried to commit a rollbackonly tx.
}
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
@Test
public void testRollbackSyncWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.doBeforeCompletion();
tm.doRollback();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testRollbackSyncViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
xaTerminator.rollback(xid);
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.doBeforeCompletion();
assertEquals(TwoPhaseOutcome.PREPARE_READONLY, tm.doPrepare());
// don't call commit for read only case
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
assertEquals(XAResource.XA_RDONLY, xaTerminator.prepare(xid));
// don't call commit for read only case
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithXAOKWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
final TestXAResource xaResource = new TestXAResource();
xaResource.setPrepareReturnValue(XAResource.XA_OK);
tm.enlistResource(xaResource);
tm.doBeforeCompletion();
assertEquals(TwoPhaseOutcome.PREPARE_OK, tm.doPrepare());
tm.doCommit();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithXAOKViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
final TestXAResource xaResource = new TestXAResource();
xaResource.setPrepareReturnValue(XAResource.XA_OK);
t.enlistResource(xaResource);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
assertEquals(XAResource.XA_OK, xaTerminator.prepare(xid));
xaTerminator.commit(xid, false);
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_COMMITTED, t.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithRollbackOnlyWithSeparateSync() throws Exception
{
final SubordinateTransaction tm = createTransaction();
final TestSynchronization sync = new TestSynchronization();
tm.registerSynchronization(sync);
tm.setRollbackOnly();
tm.doBeforeCompletion();
assertEquals(TwoPhaseOutcome.PREPARE_NOTOK, tm.doPrepare());
tm.doRollback();
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, tm.getStatus());
}
@Test
public void testTwoPhaseCommitSyncWithRollbackOnlyViaXATerminatorWithSeparateSync() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestSynchronization sync = new TestSynchronization();
t.registerSynchronization(sync);
t.setRollbackOnly();
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
final XATerminatorExtensions xaTerminatorExtensions = (XATerminatorExtensions)xaTerminator;
xaTerminatorExtensions.beforeCompletion(xid);
try {
xaTerminator.prepare(xid);
} catch(XAException e) {
assertEquals(XAException.XA_RBROLLBACK, e.errorCode);
// expected - we tried to prepare a rollbackonly tx.
}
// no need to call rollback - the XA_RBROLLBACK code indicates its been done.
assertTrue(sync.isBeforeCompletionDone());
assertTrue(sync.isAfterCompletionDone());
assertEquals(javax.transaction.Status.STATUS_ROLLEDBACK, t.getStatus());
}
@Test
public void testFailOnCommitOnePhase () throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestXAResource xaResource = new TestXAResource();
// provoke commit into failing with TwoPhaseOutcome.FINISH_ERROR
// warning: this is sensitive to the impl exception handling in
// XAResourceRecord.topLevelCommit
xaResource.setCommitException(new XAException(XAException.XA_HEURRB)); // should cause an exception!
t.enlistResource(xaResource);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
try
{
xaTerminator.commit(xid, true);
}
catch (final XAException ex)
{
// success!
return;
}
assertTrue("commit should throw an exception and not get to here", false);
}
@Test
public void testFailOnCommitRetry () throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestXAResource xaResource = new TestXAResource();
xaResource.setCommitException(new XAException(XAException.XA_RETRY));
t.enlistResource(xaResource);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
Xid[] recover1 = xaTerminator.recover(XAResource.TMSTARTRSCAN);
xaTerminator.recover(XAResource.TMENDRSCAN);
xaTerminator.prepare(xid);
/*
* This should not cause problems. The transaction really has committed, or will once
* recovery kicks off. So nothing for the parent to do. The subordinate log will
* maintain enough information to drive recovery locally if we get to the point of
* issuing a commit call from parent to child.
*/
try {
xaTerminator.commit(xid, false);
fail("Expected an error");
} catch (XAException e) {
assertTrue(e.errorCode == XAException.XAER_RMFAIL);
}
Xid[] recover2 = xaTerminator.recover(XAResource.TMSTARTRSCAN);
xaTerminator.recover(XAResource.TMENDRSCAN);
if (recover1 == null) {
recover1 = new Xid[0];
}
int difference = recover2.length - recover1.length;
assertTrue("" + difference, difference == 1);
}
@Test
public void testFailOnCommit() throws Exception
{
final Xid xid = new XidImple(new Uid());
final Transaction t = SubordinationManager.getTransactionImporter().importTransaction(xid);
final TestXAResource xaResource = new TestXAResource();
// provoke commit into failing with TwoPhaseOutcome.FINISH_ERROR
// warning: this is sensitive to the impl exception handling in
// XAResourceRecord.topLevelCommit
xaResource.setCommitException(new XAException(XAException.XA_HEURHAZ)); // throw a little spice into things!
t.enlistResource(xaResource);
final XATerminator xaTerminator = SubordinationManager.getXATerminator();
try
{
xaTerminator.prepare(xid);
xaTerminator.commit(xid, false);
}
catch (final XAException ex)
{
// success!!
return;
}
assertTrue("commit should throw an exception and not get to here", false);
}
@Test
public void testPrepareRollback() throws Exception
{
final SubordinateTransaction tm = createTransaction();
assertEquals(TwoPhaseOutcome.PREPARE_READONLY, tm.doPrepare());
tm.doRollback(); // Due to the readonly we allow the massage - this matches doPhase2Abort in ServerTransaction
}
}