Package org.infinispan.tx

Source Code of org.infinispan.tx.NoTxLeakInCacheStoreInterceptorTest

package org.infinispan.tx;

import static org.testng.Assert.*;

import java.util.List;
import java.util.Map;
import java.util.Set;

import org.infinispan.Cache;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.cache.CacheLoaderConfigurationBuilder;
import org.infinispan.configuration.cache.VersioningScheme;
import org.infinispan.interceptors.CacheStoreInterceptor;
import org.infinispan.interceptors.InterceptorChain;
import org.infinispan.interceptors.base.CommandInterceptor;
import org.infinispan.loaders.dummy.DummyInMemoryCacheStore;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.test.MultipleCacheManagersTest;
import org.infinispan.test.TestingUtil;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.infinispan.transaction.LockingMode;
import org.infinispan.transaction.xa.GlobalTransaction;
import org.testng.annotations.Test;

/**
* Test for ISPN-2321: No GlobalTransaction objects must be left behind in
* CacheStoreInterceptor after one phase commit transactions. This test has 1PC transactions as the transactional
* mode is pessimistic.
*
* @author Carsten Lohmann
*/
@Test(groups = "functional", testName = "tx.NoTxLeakInCacheStoreInterceptorTest")
public class NoTxLeakInCacheStoreInterceptorTest extends MultipleCacheManagersTest {

   private final String testCacheName = "testCache" + getClass().getSimpleName();
  
   @Override
   protected void createCacheManagers() throws Throwable {
      ConfigurationBuilder cb = new ConfigurationBuilder();
      cb.clustering().invocationBatching().enable().versioning().enable()
               .scheme(VersioningScheme.SIMPLE).transaction().lockingMode(LockingMode.PESSIMISTIC);
      cb.loaders().passivation(false).preload(true).shared(true);
      // Make it really shared by adding the test's name as store name
      CacheLoaderConfigurationBuilder lb = cb.loaders().addCacheLoader()
               .cacheLoader(new DummyInMemoryCacheStore());
      lb.addProperty("storeName", getClass().getSimpleName());

      EmbeddedCacheManager cm1 = TestCacheManagerFactory.createClusteredCacheManager();
      EmbeddedCacheManager cm2 = TestCacheManagerFactory.createClusteredCacheManager();
      registerCacheManager(cm1, cm2);
     
      defineConfigurationOnAllManagers(testCacheName, cb);
      waitForClusterToForm(testCacheName);
   }

   public void test() throws Exception {
      Cache<Object, Object> cache = cache(0, testCacheName);
      cache.put("k1", "v1");
     
      // ensure that the implicit transaction created for the "put" is cleaned up from the
      // CacheStoreInterceptor "preparingTxs" and "txStores" maps
      for (int i = 0; i < getCacheManagers().size(); i++) {
         CacheStoreInterceptor cacheStoreInterceptor = getCacheStoreInterceptor(i);
         assertTxFieldsEmpty(cacheStoreInterceptor, i);
      }
   }
  
   @SuppressWarnings("unchecked")
   private void assertTxFieldsEmpty(CacheStoreInterceptor cacheStoreInterceptor,
            int cacheManagerIndex) throws Exception {
      Map<GlobalTransaction, Set<Object>> preparingTxs = cacheStoreInterceptor.getPreparingTxs();
      log.debug("cm" + cacheManagerIndex + ": preparingTxs: " + preparingTxs);

      Map<GlobalTransaction, Integer> txStores = cacheStoreInterceptor.getTxStores();
      log.debug("cm" + cacheManagerIndex + ": txStores: " + txStores);

      assertTrue(preparingTxs.isEmpty());
      assertTrue(txStores.isEmpty());
   }

   private CacheStoreInterceptor getCacheStoreInterceptor(int cacheManagerIndex) {
      InterceptorChain interceptorChain = TestingUtil.extractComponentRegistry(
               cache(cacheManagerIndex, testCacheName)).getComponent(InterceptorChain.class);

      List<CommandInterceptor> interceptors = interceptorChain
               .getInterceptorsWhichExtend(CacheStoreInterceptor.class);
      if (interceptors.isEmpty()) {
         throw new IllegalStateException("no CacheStoreInterceptor found");
      }
      return (CacheStoreInterceptor) interceptors.get(0);
   }
}
TOP

Related Classes of org.infinispan.tx.NoTxLeakInCacheStoreInterceptorTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.