HHH-9902 - fix gradle plugin

Author: barreiro

tooling/hibernate-gradle-plugin/src/main/groovy/org/hibernate/orm/tooling/gradle/HibernatePlugin.java

@@ -18,9 +18,6 @@
 import java.util.ArrayList;
 import java.util.List;
 
-import javax.persistence.Embedded;
-import javax.persistence.Entity;
-
 import javassist.ClassPool;
 import javassist.CtClass;
 import javassist.CtField;
@@ -127,14 +124,16 @@ public boolean isLazyLoadable(CtField field) {
 
 								final CtClass ctClass = toCtClass( file, classPool );
 
-								if ( !ctClass.hasAnnotation( Entity.class )
-										&& !ctClass.hasAnnotation( Embedded.class ) ) {
-									logger.debug( "Skipping class file [" + file.getAbsolutePath() + "], not an entity nor embedded" );
+								if ( !enhancementContext.isEntityClass( ctClass )
+										&& !enhancementContext.isCompositeClass( ctClass ) ) {
+									logger.info( "Skipping class [" + file.getAbsolutePath() + "], not an entity nor embeddable" );
 									continue;
 								}
 
 								final byte[] enhancedBytecode = doEnhancement( ctClass, enhancer );
 								writeOutEnhancedClass( enhancedBytecode, ctClass, file );
+
+								logger.info( "Successfully enhanced class [" + ctClass.getName() + "]" );
 							}
 						}
 					}