/*
 *
 * Copyright (C) 2007-2015 Licensed to the Comunes Association (CA) under
 * one or more contributor license agreements (see COPYRIGHT for details).
 * The CA licenses this file to you under the GNU Affero General Public
 * License version 3, (the "License"); you may not use this file except in
 * compliance with the License. This file is part of kune.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package cc.kune.core.server;

import javax.persistence.EntityManager;
import javax.persistence.EntityTransaction;

import org.junit.Before;

import cc.kune.core.server.integration.IntegrationTestHelper;
import cc.kune.core.server.persist.DataSourceKune;

import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Provider;
import com.google.inject.persist.Transactional;

// TODO: Auto-generated Javadoc
/**
 * The Class PersistenceTest.
 * 
 * @author vjrj@ourproject.org (Vicente J. Ruiz Jurado)
 */
public abstract class PersistenceTest {

  /** The provider. */
  @Inject
  @DataSourceKune
  Provider<EntityManager> provider;

  /**
   * Instantiates a new persistence test.
   */
  public PersistenceTest() {
  }

  /**
   * Close transaction.
   */
  public void closeTransaction() {
    getManager().getTransaction().commit();
  }

  /**
   * Gets the manager.
   * 
   * @return the manager
   */
  protected EntityManager getManager() {
    return provider.get();
  }

  /**
   * Gets the transaction.
   * 
   * @return the transaction
   */
  public EntityTransaction getTransaction() {
    return getManager().getTransaction();
  }

  /**
   * Open transaction.
   * 
   * @return the entity manager
   */
  public EntityManager openTransaction() {
    final EntityManager manager = getManager();
    final EntityTransaction transaction = manager.getTransaction();
    transaction.begin();
    return manager;
  }

  /**
   * Persist.
   * 
   * @param entities
   *          the entities
   */
  public void persist(final Object... entities) {
    for (final Object entity : entities) {
      getManager().persist(entity);
    }
  }

  /**
   * Prepare.
   */
  @Transactional
  @Before
  public void prepare() {
    final Injector injector = IntegrationTestHelper.createInjector();
    injector.injectMembers(this);
  }

  /**
   * Rollback transaction.
   */
  public void rollbackTransaction() {
    final EntityTransaction transaction = getManager().getTransaction();
    if (transaction.isActive()) {
      transaction.rollback();
    }
  }

}