package org.infinispan.tx.recovery.admin;
import static org.infinispan.tx.recovery.RecoveryTestUtil.commitTransaction;
import static org.infinispan.tx.recovery.RecoveryTestUtil.prepareTransaction;
import static org.testng.Assert.assertEquals;
import java.util.List;
import javax.transaction.xa.XAException;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.distribution.MagicKey;
import org.infinispan.interceptors.impl.InvocationContextInterceptor;
import org.infinispan.test.TestingUtil;
import org.infinispan.transaction.tm.EmbeddedTransaction;
import org.infinispan.util.concurrent.IsolationLevel;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
/**
* This test makes sure that when a tx fails during commit it can still be completed.
*
* @author Mircea Markus
* @since 5.0
*/
@Test (groups = "functional", testName = "tx.recovery.admin.CommitFailsTest")
public class CommitFailsTest extends AbstractRecoveryTest {
private Object key;
private InDoubtWithCommitFailsTest.ForceFailureInterceptor failureInterceptor0;
private InDoubtWithCommitFailsTest.ForceFailureInterceptor failureInterceptor1;
@Override
protected void createCacheManagers() throws Throwable {
ConfigurationBuilder configuration = defaultRecoveryConfig();
configuration.transaction().autoCommit(false);
configuration.locking().isolationLevel(IsolationLevel.READ_COMMITTED); //skip WSC exceptions
createCluster(configuration, 3);
waitForClusterToForm();
key = getKey();
failureInterceptor0 = new InDoubtWithCommitFailsTest.ForceFailureInterceptor();
failureInterceptor1 = new InDoubtWithCommitFailsTest.ForceFailureInterceptor();
advancedCache(0).getAsyncInterceptorChain().addInterceptorAfter(failureInterceptor0, InvocationContextInterceptor.class);
advancedCache(1).getAsyncInterceptorChain().addInterceptorAfter(failureInterceptor1, InvocationContextInterceptor.class);
}
@BeforeMethod
protected void setUpTx() throws Exception {
failureInterceptor0.fail = true;
failureInterceptor1.fail = true;
tm(2).begin();
cache(2).put(this.key, "newValue");
EmbeddedTransaction tx = (EmbeddedTransaction) tm(2).suspend();
prepareTransaction(tx);
try {
commitTransaction(tx);
assert false;
} catch (XAException e) {
//expected
}
assertEquals(countInDoubtTx(recoveryOps(2).showInDoubtTransactions()), 1);
log.trace("here is the remote get...");
assertEquals(countInDoubtTx(recoveryOps(0).showInDoubtTransactions()), 1);
assertEquals(countInDoubtTx(recoveryOps(1).showInDoubtTransactions()), 1);
failureInterceptor0.fail = false;
failureInterceptor1.fail = false;
}
protected Object getKey() {
return new MagicKey(cache(2));
}
public void testForceCommitOnOriginator() throws Exception {
runTest(2);
}
public void testForceCommitNonTxParticipant() throws Exception {
int where = getTxParticipant(false);
runTest(where);
}
public void testForceCommitTxParticipant() throws Exception {
int where = getTxParticipant(true);
runTest(where);
}
private void assertAllHaveNewValue(Object key) throws Exception {
for (Cache c : caches()) {
Object actual;
TestingUtil.getTransactionManager(c).begin();
actual = c.get(key);
TestingUtil.getTransactionManager(c).commit();
assertEquals(actual, "newValue");
}
}
protected void runTest(int where) throws Exception {
List<Long> internalIds = getInternalIds(recoveryOps(where).showInDoubtTransactions());
log.debugf("About to force commit on node %s", address(where));
recoveryOps(where).forceCommit(internalIds.get(0));
assertCleanup(0);
assertCleanup(1);
assertCleanup(2);
assertAllHaveNewValue(key);
assertCleanup(0, 1, 2);
}
@Override
protected int getTxParticipant(boolean txParticipant) {
int expectedNumber = txParticipant ? 1 : 0;
int index = -1;
for (int i = 0; i < 2; i++) {
if (tt(i).getRemoteTxCount() == expectedNumber) {
index = i;
break;
}
}
return index;
}
}