HHH-16895 @Check constraint not generated when annotated on entity

hibernate-core/src/main/java/org/hibernate/boot/internal/InFlightMetadataCollectorImpl.java

@@ -2354,6 +2354,12 @@ private void processExportableProducers() {
 		final Dialect dialect = getDatabase().getJdbcEnvironment().getDialect();
 
 		for ( PersistentClass entityBinding : entityBindingMap.values() ) {
+			entityBinding.assignCheckConstraintsToTable(
+					dialect,
+					bootstrapContext.getTypeConfiguration(),
+					bootstrapContext.getFunctionRegistry()
+			);
+
 			if ( entityBinding.isInherited() ) {
 				continue;
 			}
@@ -2363,6 +2369,7 @@ private void processExportableProducers() {
 					dialect,
 					(RootClass) entityBinding
 			);
+
 		}
 
 		for ( Collection collection : collectionBindingMap.values() ) {

hibernate-core/src/main/java/org/hibernate/mapping/PersistentClass.java

@@ -1258,22 +1258,21 @@ public void setSuperMappedSuperclass(MappedSuperclass superMappedSuperclass) {
 		this.superMappedSuperclass = superMappedSuperclass;
 	}
 
-	// End of @MappedSuperclass support
+	public void assignCheckConstraintsToTable(Dialect dialect, TypeConfiguration types, SqmFunctionRegistry functions) {
-
-	public void prepareForMappingModel(RuntimeModelCreationContext context) {
-		if ( !joins.isEmpty() ) {
-			// we need to deal with references to secondary tables
-			// in SQL formulas and check constraints
-
-			final Dialect dialect = context.getDialect();
-			final TypeConfiguration types = context.getTypeConfiguration();
-			final SqmFunctionRegistry functions = context.getFunctionRegistry();
-
-			// assign check constraints to the right table
 		for ( CheckConstraint checkConstraint : checkConstraints ) {
 			container( collectColumnNames( checkConstraint.getConstraint(), dialect, types, functions ) )
 					.getTable().addCheck( checkConstraint );
 		}
+	}
+
+	// End of @MappedSuperclass support
+	public void prepareForMappingModel(RuntimeModelCreationContext context) {
+		if ( !joins.isEmpty() ) {
+			// we need to deal with references to secondary tables
+			// in SQL formulas
+			final Dialect dialect = context.getDialect();
+			final TypeConfiguration types = context.getTypeConfiguration();
+			final SqmFunctionRegistry functions = context.getFunctionRegistry();
 
 			// now, move @Formulas to the correct AttributeContainers
 			//TODO: skip this step for hbm.xml

hibernate-core/src/main/java/org/hibernate/sql/Template.java

@@ -15,7 +15,6 @@
 
 import org.hibernate.HibernateException;
 import org.hibernate.dialect.Dialect;
-import org.hibernate.engine.spi.SessionFactoryImplementor;
 import org.hibernate.query.sqm.function.SqmFunctionDescriptor;
 import org.hibernate.query.sqm.function.SqmFunctionRegistry;
 import org.hibernate.type.spi.TypeConfiguration;