HHH-11242 - Basic test case for the maven plugin

tooling/hibernate-enhance-maven-plugin/src/test/java/org/hibernate/orm/tooling/maven/MavenEnhancePluginTest.java

@@ -0,0 +1,108 @@
+/*
+ * 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.orm.tooling.maven;
+
+import org.apache.maven.project.MavenProject;
+import org.codehaus.plexus.util.ReflectionUtils;
+import org.hibernate.engine.spi.Managed;
+import org.junit.Assert;
+import org.junit.Test;
+import org.sonatype.plexus.build.incremental.DefaultBuildContext;
+
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.MappedSuperclass;
+import java.io.File;
+import java.lang.reflect.Field;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Test case for Maven Enhance Plugin
+ *
+ * @author Luis Barreiro
+ */
+public class MavenEnhancePluginTest {
+
+    @Test
+    public void testEnhancePlugin() throws Exception {
+        File baseDir = new File("target/classes/test");
+        URL[] baseURLs = { baseDir.toURI().toURL() };
+
+        MavenEnhancePlugin plugin = new MavenEnhancePlugin();
+
+        Map<String, Object> pluginContext = new HashMap<>();
+        pluginContext.put( "project", new MavenProject() );
+
+        setVariableValueToObject( plugin, "pluginContext", pluginContext );
+        setVariableValueToObject( plugin, "buildContext", new DefaultBuildContext() );
+
+        setVariableValueToObject( plugin, "base", baseDir.getAbsolutePath() );
+        setVariableValueToObject( plugin, "dir", baseDir.getAbsolutePath() );
+
+        setVariableValueToObject( plugin, "failOnError", true );
+        setVariableValueToObject( plugin, "enableLazyInitialization", true );
+        setVariableValueToObject( plugin, "enableDirtyTracking", true );
+        setVariableValueToObject( plugin, "enableAssociationManagement", true );
+        setVariableValueToObject( plugin, "enableExtendedEnhancement", false );
+
+        plugin.execute();
+
+        try ( URLClassLoader classLoader = new URLClassLoader( baseURLs , getClass().getClassLoader() ) ) {
+
+            Assert.assertTrue( declaresManaged( classLoader.loadClass( ParentEntity.class.getName() ) ) );
+            Assert.assertTrue( declaresManaged( classLoader.loadClass( ChildEntity.class.getName() ) ) );
+            Assert.assertTrue( declaresManaged( classLoader.loadClass( TestEntity.class.getName() ) ) );
+
+        }
+
+    }
+
+    private void setVariableValueToObject( Object object, String variable, Object value ) throws IllegalAccessException {
+        Field field = ReflectionUtils.getFieldByNameIncludingSuperclasses( variable, object.getClass() );
+        field.setAccessible( true );
+        field.set( object, value );
+    }
+
+    private boolean declaresManaged(Class<?> clazz) {
+        for ( Class<?> interfaceClazz : clazz.getInterfaces() ) {
+            if ( Managed.class.isAssignableFrom( interfaceClazz ) ) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    // --- //
+
+    @MappedSuperclass
+    public static class ParentEntity {
+
+        String parentValue;
+
+    }
+
+    @MappedSuperclass
+    public static class ChildEntity extends ParentEntity {
+
+        String childValue;
+
+    }
+
+    @Entity
+    public static class TestEntity extends ChildEntity {
+
+        @Id
+        long id;
+
+        String testValue;
+
+    }
+
+}