import javax.persistence.Entity;
import javax.persistence.EnumType;
import javax.persistence.Enumerated;
import javax.persistence.Id;

import org.hibernate.boot.MetadataSources;
import org.hibernate.boot.registry.StandardServiceRegistry;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.boot.spi.MetadataImplementor;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.mapping.Property;
import org.hibernate.type.CustomType;

import org.hibernate.testing.junit4.BaseUnitTestCase;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;

/**
 * @author Steve Ebersole
 */
public class EnumeratedSmokeTest extends BaseUnitTestCase {
	private StandardServiceRegistry ssr;

	@Before
	public void prepare() {
		ssr = new StandardServiceRegistryBuilder().build();
	}

	@After
	public void release() {
		if ( ssr != null ) {
			StandardServiceRegistryBuilder.destroy( ssr );
		}
	}

	@Test
	public void testEnumeratedTypeResolutions() {
		final MetadataImplementor mappings = (MetadataImplementor) new MetadataSources( ssr )
				.addAnnotatedClass( EntityWithEnumeratedAttributes.class )
				.buildMetadata();
		mappings.validate();

		final PersistentClass entityBinding = mappings.getEntityBinding( EntityWithEnumeratedAttributes.class.getName() );

		validateEnumMapping( entityBinding.getProperty( "notAnnotated" ), EnumType.ORDINAL );
		validateEnumMapping( entityBinding.getProperty( "noEnumType" ), EnumType.ORDINAL );
		validateEnumMapping( entityBinding.getProperty( "ordinalEnumType" ), EnumType.ORDINAL );
		validateEnumMapping( entityBinding.getProperty( "stringEnumType" ), EnumType.STRING );
	}

	private void validateEnumMapping(Property property, EnumType expectedJpaEnumType) {
		assertThat( property.getType(), instanceOf( CustomType.class ) );
		final CustomType customType = (CustomType) property.getType();
		assertThat( customType.getUserType(), instanceOf( org.hibernate.type.EnumType.class ) );
		final org.hibernate.type.EnumType hibernateMappingEnumType = (org.hibernate.type.EnumType) customType.getUserType();
		assertThat( hibernateMappingEnumType.isOrdinal(), is(expectedJpaEnumType==EnumType.ORDINAL) );
	}

	@Entity
	public static class EntityWithEnumeratedAttributes {
		@Id
		public Integer id;
		public Gender notAnnotated;
		@Enumerated
		public Gender noEnumType;
		@Enumerated(EnumType.ORDINAL)
		public Gender ordinalEnumType;
		@Enumerated(EnumType.STRING)
		public Gender stringEnumType;
	}

	public static enum Gender {
		MALE, FEMALE, UNKNOWN;
	}
}