Hbm2JavaEjb3Test.java

/*
 * Created on 2004-12-01
 *
 */
package org.hibernate.tool.hbm2x;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import javax.persistence.Persistence;

import org.hibernate.Version;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.mapping.Property;
import org.hibernate.tool.NonReflectiveTestCase;
import org.hibernate.tool.hbm2x.pojo.AnnotationBuilder;
import org.hibernate.tool.hbm2x.pojo.EntityPOJOClass;
import org.hibernate.tool.hbm2x.pojo.IteratorTransformer;
import org.hibernate.tool.hbm2x.pojo.POJOClass;
import org.hibernate.tool.test.TestHelper;

/**
 * @author max
 *
 */
public class Hbm2JavaEjb3Test extends NonReflectiveTestCase {

	public Hbm2JavaEjb3Test(String name) {
		super( name, "hbm2javaejb3output" );
	}

	protected void setUp() throws Exception {
		super.setUp();

		generateComparator();
		
		POJOExporter exporter = new POJOExporter(getCfg(), getOutputDir() );
		exporter.setTemplatePath(new String[0]);
		exporter.getProperties().setProperty("ejb3", "true");
		exporter.getProperties().setProperty("jdk5", "true");

		exporter.start();
	}

	public void testFileExistence() {
		assertFileAndExists( new File(getOutputDir().getAbsolutePath() + "/org/hibernate/tool/hbm2x/Author.java") );
		assertFileAndExists( new File(getOutputDir().getAbsolutePath() + "/org/hibernate/tool/hbm2x/Article.java") );
		assertFileAndExists( new File(getOutputDir().getAbsolutePath() + "/org/hibernate/tool/hbm2x/Train.java") );
		assertFileAndExists( new File(getOutputDir().getAbsolutePath() + "/org/hibernate/tool/hbm2x/Passenger.java") );
		assertFileAndExists( new File(getOutputDir().getAbsolutePath() + "/org/hibernate/tool/hbm2x/TransportationPk.java") );
	}

	public void testBasicComponent() {
		assertEquals( "@Embeddable", findFirstString( "@Embeddable", new File( getOutputDir(),
		"org/hibernate/tool/hbm2x/TransportationPk.java" ) ) );
	}

	public void testCompile() {

		File file = new File("ejb3compilable");
		file.mkdir();

		ArrayList<String> list = new ArrayList<String>();
		List<File> jars = new ArrayList<File>();
		jars.add(TestHelper.findJarFileFor(Persistence.class)); // for jpa api
		jars.add(TestHelper.findJarFileFor(Version.class)); // for hibernate core
		TestHelper.compile(getOutputDir(), file, TestHelper.visitAllFiles(getOutputDir(), list), "1.5", TestHelper.buildClasspathFromFileList(jars));

		TestHelper.deleteDir(file);
	}

	public void testEqualsHashCode() {
		PersistentClass classMapping = getMetadata().getEntityBinding("org.hibernate.tool.hbm2x.Passenger");
		POJOClass clazz = new Cfg2JavaTool().getPOJOClass(classMapping);
		
		assertFalse(clazz.needsEqualsHashCode());
		
		classMapping = getMetadata().getEntityBinding("org.hibernate.tool.hbm2x.Article");
		clazz = new Cfg2JavaTool().getPOJOClass(classMapping);
		
		assertTrue(clazz.needsEqualsHashCode());
		
	}
	

	public void testAnnotationColumnDefaults() {
		PersistentClass classMapping = getMetadata().getEntityBinding("org.hibernate.tool.hbm2x.Article");
		Cfg2JavaTool cfg2java = new Cfg2JavaTool();
		POJOClass clazz = cfg2java.getPOJOClass(classMapping);
		
		Property p = classMapping.getProperty("content");
		
		String string = clazz.generateAnnColumnAnnotation( p );
		
		assertNotNull(string);
		assertEquals(-1, string.indexOf("unique="));
		assertTrue(string.indexOf("nullable=")>=0);
		assertEquals(-1, string.indexOf("insertable="));
		assertEquals(-1, string.indexOf("updatable="));
		assertTrue(string.indexOf("length=10000")>0);
		
		p = classMapping.getProperty("name");
		string = clazz.generateAnnColumnAnnotation( p );
		
		assertNotNull(string);
		assertEquals(-1, string.indexOf("unique="));
		assertTrue(string.indexOf("nullable=")>=0);
		assertEquals(-1, string.indexOf("insertable="));
		assertTrue(string.indexOf("updatable=false")>0);
		assertTrue(string.indexOf("length=100")>0);
		
		
		classMapping = getMetadata().getEntityBinding( "org.hibernate.tool.hbm2x.Train" );
		clazz = cfg2java.getPOJOClass(classMapping);
		
		p = classMapping.getProperty( "name" );
		string = clazz.generateAnnColumnAnnotation( p );
		assertNotNull(string);
		assertTrue(string.indexOf("unique=true")>0);
		assertTrue(string.indexOf("nullable=")>=0);
		assertEquals(-1, string.indexOf("insertable="));
		assertEquals(-1,string.indexOf("updatable="));
		assertEquals(-1, string.indexOf("length="));
		
	}
	
	public void testEmptyCascade() {
		PersistentClass classMapping = getMetadata().getEntityBinding("org.hibernate.tool.hbm2x.Article");
		
		Cfg2JavaTool cfg2java = new Cfg2JavaTool();
		EntityPOJOClass clazz = (EntityPOJOClass) cfg2java.getPOJOClass(classMapping);
		Property property = classMapping.getProperty( "author" );
		
		assertEquals(0, clazz.getCascadeTypes( property ).length);
		
		assertEquals(null,findFirstString( "cascade={}", new File(getOutputDir(), "org/hibernate/tool/hbm2x/Article.java") ));
	}
		
	public void testAnnotationBuilder() {

		AnnotationBuilder builder =  AnnotationBuilder.createAnnotation("SingleCleared").resetAnnotation( "Single" );
		
		assertEquals("@Single", builder.getResult());
		
		builder = AnnotationBuilder.createAnnotation("javax.persistence.OneToMany")
				    .addAttribute("willbecleared", (String)null)
				    .resetAnnotation("javax.persistence.OneToMany")
					.addAttribute("cascade", new String[] { "val1", "val2"})
					.addAttribute("fetch", "singleValue");
		
		assertEquals("@javax.persistence.OneToMany(cascade={val1, val2}, fetch=singleValue)", builder.getResult());
		
		builder = AnnotationBuilder.createAnnotation("javax.persistence.OneToMany");
		builder.addAttribute("cascade", (String[])null);
		builder.addAttribute("fetch", (String)null);
		
		assertEquals("@javax.persistence.OneToMany", builder.getResult());

		builder = AnnotationBuilder.createAnnotation("abc");
		ArrayList<Object> list = new ArrayList<Object>();
		list.add(new Integer(42));
		list.add( new String("xxx") );
		builder.addQuotedAttributes( "it", list.iterator() );
		
		assertEquals("@abc(it={\"42\", \"xxx\"})", builder.getResult());		
	
		List<String> columns = new ArrayList<String>();
		columns.add("first");
		columns.add("second");
		
		AnnotationBuilder constraint = AnnotationBuilder.createAnnotation( "UniqueConstraint" );
		constraint.addQuotedAttributes( "columnNames", new IteratorTransformer<String>(columns.iterator()) {
			public String transform(String object) {					
				return object.toString();
			}
		});
		constraint.addAttribute( "single", "value" );
		
		String attribute = constraint.getAttributeAsString("columnNames");
		assertEquals("{\"first\", \"second\"}", attribute);
		
		assertEquals("value", constraint.getAttributeAsString( "single" ));
		
	}
	
	
	
	
	
	protected String getBaseForMappings() {
		return "org/hibernate/tool/hbm2x/";
	}

	protected String[] getMappings() {
		return new String[] {
				"Author.hbm.xml",
				"Article.hbm.xml",
				"Train.hbm.xml",
				"Passenger.hbm.xml"
		};
	}

}