package org.infinispan.tx.synchronisation;
import org.infinispan.config.Configuration;
import org.infinispan.manager.DefaultCacheManager;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.infinispan.transaction.lookup.DummyTransactionManagerLookup;
import org.infinispan.transaction.tm.DummyTransaction;
import org.infinispan.tx.LocalModeTxTest;
import org.testng.annotations.Test;
import javax.transaction.NotSupportedException;
import javax.transaction.SystemException;
import static org.testng.Assert.assertEquals;
/**
* @author Mircea.Markus@jboss.com
* @since 5.0
*/
@Test(groups = "functional", testName = "tx.synchronisation.LocalModeWithSyncTxTest")
public class LocalModeWithSyncTxTest extends LocalModeTxTest {
@Override
protected EmbeddedCacheManager createCacheManager() {
Configuration config = getDefaultStandaloneConfig(true);
config.fluent().transaction().transactionManagerLookupClass(DummyTransactionManagerLookup.class);
config.fluent().transaction().useSynchronization(true);
return TestCacheManagerFactory.createCacheManager(config);
}
public void testSyncRegisteredWithCommit() throws Exception {
DummyTransaction dt = startTx();
tm().commit();
assertEquals(0, dt.getEnlistedResources().size());
assertEquals(0, dt.getEnlistedSynchronization().size());
assertEquals("v", cache.get("k"));
}
public void testSyncRegisteredWithRollback() throws Exception {
DummyTransaction dt = startTx();
tm().rollback();
assertEquals(null, cache.get("k"));
assertEquals(0, dt.getEnlistedResources().size());
assertEquals(0, dt.getEnlistedSynchronization().size());
}
private DummyTransaction startTx() throws NotSupportedException, SystemException {
tm().begin();
cache.put("k","v");
DummyTransaction dt = (DummyTransaction) tm().getTransaction();
assertEquals(0, dt.getEnlistedResources().size());
assertEquals(1, dt.getEnlistedSynchronization().size());
cache.put("k2","v2");
assertEquals(0, dt.getEnlistedResources().size());
assertEquals(1, dt.getEnlistedSynchronization().size());
return dt;
}
}