package org.infinispan.persistence.jpa.configuration;

import static org.testng.AssertJUnit.assertEquals;
import static org.testng.AssertJUnit.assertTrue;

import java.io.IOException;

import org.infinispan.Cache;
import org.infinispan.configuration.cache.Configuration;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.cache.StoreConfiguration;
import org.infinispan.configuration.global.GlobalConfiguration;
import org.infinispan.configuration.global.GlobalConfigurationBuilder;
import org.infinispan.manager.DefaultCacheManager;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.persistence.jpa.JpaSCI;
import org.infinispan.persistence.jpa.entity.User;
import org.infinispan.persistence.jpa.entity.Vehicle;
import org.infinispan.persistence.jpa.entity.VehicleId;
import org.testng.annotations.Test;

/**
 * @author <a href="mailto:rtsang@redhat.com">Ray Tsang</a>
 */
@Test(groups = "functional", testName = "persistence.JpaConfigurationTest")
public class JpaConfigurationTest {

   private static final String PERSISTENCE_UNIT_NAME = "org.infinispan.persistence.jpa.configurationTest";

   public void testConfigBuilder() {
      GlobalConfiguration globalConfig = new GlobalConfigurationBuilder()
            .transport().defaultTransport()
            .serialization().addContextInitializer(JpaSCI.INSTANCE).build();

      Configuration cacheConfig = new ConfigurationBuilder().persistence()
            .addStore(JpaStoreConfigurationBuilder.class)
            .segmented(false)
            .persistenceUnitName(PERSISTENCE_UNIT_NAME)
            .entityClass(User.class).build();


      StoreConfiguration storeConfiguration = cacheConfig.persistence().stores().get(0);
      assertTrue(storeConfiguration instanceof JpaStoreConfiguration);
      JpaStoreConfiguration jpaCacheLoaderConfig = (JpaStoreConfiguration) storeConfiguration;
      assertEquals(PERSISTENCE_UNIT_NAME, jpaCacheLoaderConfig.persistenceUnitName());
      assertEquals(User.class, jpaCacheLoaderConfig.entityClass());

      EmbeddedCacheManager cacheManager = new DefaultCacheManager(globalConfig);

      cacheManager.defineConfiguration("userCache", cacheConfig);

      cacheManager.start();
      Cache<String, User> userCache = cacheManager.getCache("userCache");
      User user = new User();
      user.setUsername("rtsang");
      user.setFirstName("Ray");
      user.setLastName("Tsang");
      userCache.put(user.getUsername(), user);
      userCache.stop();
      cacheManager.stop();
   }

   protected void validateConfig(Cache<VehicleId, Vehicle> vehicleCache) {
      StoreConfiguration config = vehicleCache.getCacheConfiguration().persistence().stores().get(0);

      assertTrue(config instanceof JpaStoreConfiguration);
      JpaStoreConfiguration jpaConfig = (JpaStoreConfiguration) config;
      assertEquals(1, jpaConfig.batchSize());
      assertEquals(Vehicle.class, jpaConfig.entityClass());
      assertEquals(PERSISTENCE_UNIT_NAME, jpaConfig.persistenceUnitName());
   }

   public void testXmlConfig() throws IOException {
      EmbeddedCacheManager cacheManager = new DefaultCacheManager("configs/all/jpa-config.xml");

      Cache<VehicleId, Vehicle> vehicleCache = cacheManager.getCache("vehicleCache");
      validateConfig(vehicleCache);

      Vehicle v = new Vehicle();
      v.setId(new VehicleId("NC", "123456"));
      v.setColor("BLUE");
      vehicleCache.put(v.getId(), v);

      vehicleCache.stop();
      cacheManager.stop();
   }

}
