/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
 */
package org.hibernate.jpa.test.factory.puUtil;

import java.io.Serializable;
import javax.persistence.EntityManager;

import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/**
 * @author Steve Ebersole
 */
public class GetIdentifierTest extends BaseEntityManagerFunctionalTestCase {
	@Override
	protected Class<?>[] getAnnotatedClasses() {
		return new Class[] {
				LegacyEntity.class,
				ModernEntity.class,
				NestedLegacyEntity.class
		};
	}

	@Before
	public void createData() {

	}

	@After
	public void dropData() {

	}

	@Test
	public void getIdentifierTest() throws Exception {
		EntityManager entityManager = getOrCreateEntityManager();
		entityManager.getTransaction().begin();

		// This gives a NullPointerException right now. Look at HHH-10623 when this issue is fixed
		Serializable nestedLegacyEntityId = (Serializable) entityManager.getEntityManagerFactory()
				.getPersistenceUnitUtil().getIdentifier(createExisitingNestedLegacyEntity());

		entityManager.getTransaction().commit();
		entityManager.close();
	}

	private NestedLegacyEntity createExisitingNestedLegacyEntity() {

		ModernEntity modernEntity = new ModernEntity();
		modernEntity.setFoo(2);

		LegacyEntity legacyEntity = new LegacyEntity();
		legacyEntity.setPrimitivePk1(1);
		legacyEntity.setPrimitivePk2(2);
		legacyEntity.setFoo("Foo");

		NestedLegacyEntity nestedLegacyEntity = new NestedLegacyEntity();
		nestedLegacyEntity.setModernEntity(modernEntity);
		nestedLegacyEntity.setLegacyEntity(legacyEntity);

		return nestedLegacyEntity;
	}
}