package org.infinispan.tx;

import org.infinispan.configuration.cache.Configuration;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.manager.DefaultCacheManager;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.test.SingleCacheManagerTest;
import org.infinispan.test.fwk.AtomikosTransactionManagerLookup;
import org.infinispan.transaction.lookup.TransactionManagerLookup;
import org.testng.annotations.Test;

import javax.transaction.TransactionManager;

/**
 * @author Mircea.Markus@jboss.org
 */
@Test(groups = "functional", testName = "tx.BatchingWithAtomikosTest")
public class BatchingWithAtomikosTest extends SingleCacheManagerTest {

   @Override
   protected EmbeddedCacheManager createCacheManager() throws Exception {
      ConfigurationBuilder dc = getDefaultStandaloneCacheConfig(true);
      Configuration build = dc.invocationBatching().enable()
            .build();
      return new DefaultCacheManager(build);
   }

   public void testBatching() throws Exception {
      cache.getAdvancedCache().getComponentRegistry().registerComponent(AtomikosTransactionManagerLookup.INSTANCE.getTransactionManager(), TransactionManager.class);
      cache.getAdvancedCache().getComponentRegistry().rewire();
      System.out.println("BatchingWithAtomikosTest.testBatching");
      System.out.println("tm() = " + tm());
      cache.startBatch();
      cache.put("k", "v");
      cache.endBatch(true);
   }
}