/*
* JBoss, Home of Professional Open Source
* Copyright 2011 Red Hat Inc. and/or its affiliates and other
* contributors as indicated by the @author tags. All rights reserved.
* 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.
*/
package org.infinispan.tx.recovery.admin;
import org.infinispan.config.Configuration;
import org.infinispan.transaction.RemoteTransaction;
import org.infinispan.transaction.tm.DummyTransaction;
import org.infinispan.transaction.tm.DummyXid;
import org.infinispan.transaction.xa.XaTransactionTable;
import org.infinispan.transaction.xa.recovery.RecoverableTransactionIdentifier;
import org.infinispan.tx.recovery.PostCommitRecoveryStateTest;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import javax.transaction.xa.XAException;
import javax.transaction.xa.Xid;
import static junit.framework.Assert.assertEquals;
import static org.infinispan.tx.recovery.RecoveryTestUtil.*;
/**
* @author Mircea Markus
* @since 5.0
*/
@Test (groups = "functional", testName = "tx.recovery.admin.ForgetTest")
public class ForgetTest extends AbstractRecoveryTest {
private PostCommitRecoveryStateTest.RecoveryManagerDelegate recoveryManager;
private DummyTransaction tx;
@Override
protected void createCacheManagers() throws Throwable {
Configuration configuration = defaultRecoveryConfig();
createCluster(configuration, 2);
waitForClusterToForm();
XaTransactionTable txTable = tt(0);
recoveryManager = new PostCommitRecoveryStateTest.RecoveryManagerDelegate(txTable.getRecoveryManager());
txTable.setRecoveryManager(recoveryManager);
}
@BeforeMethod
public void runTx() throws XAException {
tx = beginAndSuspendTx(cache(0));
prepareTransaction(tx);
assertEquals(recoveryManager(0).getPreparedTransactionsFromCluster().all().length, 1);
assertEquals(tt(0).getLocalPreparedXids().size(), 1);
assertEquals(tt(1).getRemoteTxCount(), 1);
commitTransaction(tx);
assertEquals(tt(1).getRemoteTxCount(), 1);
}
public void testInternalIdOnSameNode() throws Exception {
Xid xid = tx.getXid();
String result = recoveryOps(0).forget(xid.getFormatId(), xid.getGlobalTransactionId(), xid.getBranchQualifier());
System.out.println("result = " + result);
assertEquals(tt(1).getRemoteTxCount(), 0);//make sure tx has been removed
}
public void testForgetXidOnSameNode() throws Exception {
forgetWithXid(0);
}
public void testForgetXidOnOtherNode() throws Exception {
forgetWithXid(1);
}
public void testForgetInternalIdOnSameNode() throws Exception {
forgetWithInternalId(0);
}
public void testForgetInternalIdOnOtherNode() throws Exception {
forgetWithInternalId(1);
}
protected void forgetWithInternalId(int cacheIndex) {
long internalId = -1;
for (RemoteTransaction rt : tt(1).getRemoteTransactions()) {
RecoverableTransactionIdentifier a = (RecoverableTransactionIdentifier) rt.getGlobalTransaction();
if (a.getXid().equals(tx.getXid())) {
internalId = a.getInternalId();
}
}
if (internalId == -1) throw new IllegalStateException();
log.tracef("About to forget... %s", internalId);
recoveryOps(cacheIndex).forget(internalId);
assertEquals(tt(0).getRemoteTxCount(), 0);
assertEquals(tt(1).getRemoteTxCount(), 0);
}
private void forgetWithXid(int nodeIndex) {
Xid xid = tx.getXid();
recoveryOps(nodeIndex).forget(xid.getFormatId(), xid.getGlobalTransactionId(), xid.getBranchQualifier());
assertEquals(tt(1).getRemoteTxCount(), 0);//make sure tx has been removed
}
}