package org.infinispan.tx;
import java.util.List;
import javax.transaction.NotSupportedException;
import javax.transaction.SystemException;
import javax.transaction.TransactionManager;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.remoting.RemoteException;
import org.infinispan.test.MultipleCacheManagersTest;
import org.infinispan.test.TestingUtil;
import org.infinispan.transaction.LockingMode;
import org.infinispan.util.concurrent.TimeoutException;
import org.testng.annotations.Test;
/**
* Tests what happens when a member acquires locks and then dies.
*
* @author Manik Surtani
* @since 4.0
*/
@Test(groups = "functional", testName = "tx.StaleLockRecoveryTest")
public class StaleLockRecoveryTest extends MultipleCacheManagersTest {
private Cache<String, String> c1, c2;
@Override
protected void createCacheManagers() throws Throwable {
ConfigurationBuilder c = getDefaultClusteredCacheConfig(CacheMode.REPL_SYNC, true);
c.transaction().lockingMode(LockingMode.PESSIMISTIC)
.locking().lockAcquisitionTimeout(TestingUtil.shortTimeoutMillis());
List<Cache<String, String>> caches = createClusteredCaches(2, "tx", c);
c1 = caches.get(0);
c2 = caches.get(1);
}
public void testStaleLock() throws SystemException, NotSupportedException {
c1.put("k", "v");
assert c1.get("k").equals("v");
assert c2.get("k").equals("v");
TransactionManager tm = TestingUtil.getTransactionManager(c1);
tm.begin();
c1.getAdvancedCache().lock("k");
tm.suspend();
// test that both c1 and c2 have locked k
assertLocked(c1, "k");
assertLocked(c2, "k");
cacheManagers.get(0).stop();
TestingUtil.blockUntilViewReceived(c2, 1);
EmbeddedCacheManager cacheManager = c2.getCacheManager();
assert cacheManager.getMembers().size() == 1;
// may take a while from when the view change is seen through to when the lock is cleared
TestingUtil.sleepThread(1000);
assertNotLocked(c2, "k");
}
private void assertLocked(Cache<String, String> c, String key) throws SystemException, NotSupportedException {
TransactionManager tm = TestingUtil.getTransactionManager(c);
tm.begin();
try {
c.put(key, "dummy"); // should time out
assert false : "Should have been locked!";
} catch (TimeoutException e) {
// ignoring timeout exception
} catch (RemoteException e) {
assert e.getCause() instanceof TimeoutException;
// ignoring timeout exception
} finally {
tm.rollback();
}
}
private void assertNotLocked(Cache<String, String> c, String key) throws SystemException, NotSupportedException {
TransactionManager tm = TestingUtil.getTransactionManager(c);
tm.begin();
try {
c.put(key, "dummy"); // should time out
} catch (TimeoutException e) {
assert false : "Should not have been locked!";
} finally {
tm.rollback();
}
}
}