Hibernate
/
hibernate-orm
mirror of https://github.com/hibernate/hibernate-orm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

HHH-17292 Add test for issue

This commit is contained in:
Marco Belladelli 2023-10-09 11:03:11 +02:00
parent a5a96007a8
commit b5d26d19c1
1 changed files with 157 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

157
hibernate-core/src/test/java/org/hibernate/orm/test/inheritance/MultiLevelInheritanceQueryTest.java Normal file
View File

@ -0,0 +1,157 @@
/*
* 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.test.inheritance;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.testing.orm.junit.DomainModel;
import org.hibernate.testing.orm.junit.Jira;
import org.hibernate.testing.orm.junit.SessionFactory;
import org.hibernate.testing.orm.junit.SessionFactoryScope;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import jakarta.persistence.DiscriminatorColumn;
import jakarta.persistence.DiscriminatorType;
import jakarta.persistence.DiscriminatorValue;
import jakarta.persistence.Entity;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.Id;
import jakarta.persistence.MappedSuperclass;
import jakarta.persistence.Table;
import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Marco Belladelli
*/
@DomainModel( annotatedClasses = {
MultiLevelInheritanceQueryTest.AbstractRootEntity.class,
MultiLevelInheritanceQueryTest.AbstractSuperclass.class,
MultiLevelInheritanceQueryTest.ChildOneEntity.class,
MultiLevelInheritanceQueryTest.ChildTwoEntity.class,
} )
@SessionFactory
@Jira( "https://hibernate.atlassian.net/browse/HHH-17292" )
public class MultiLevelInheritanceQueryTest {
@BeforeAll
public void setUp(SessionFactoryScope scope) {
scope.inTransaction( session -> {
session.persist( new ChildTwoEntity( 1, "superclass_1", "child_one_1", "child_two_1" ) );
session.persist( new ChildTwoEntity( 2, "superclass_2", "child_one_2", "child_two_2" ) );
} );
}
@AfterAll
public void tearDown(SessionFactoryScope scope) {
scope.inTransaction( session -> session.createMutationQuery( "delete from AbstractRootEntity" ).executeUpdate() );
}
@Test
public void testSelectionQuery(SessionFactoryScope scope) {
scope.inTransaction( session -> {
for ( PersistentClass pc : scope.getMetadataImplementor().getEntityBindings() ) {
// Run test for each class in the inheritance tree
final Class<?> entityClass = pc.getMappedClass();
executeSelectionQuery( session, entityClass, "baseProp = 1" );
executeSelectionQuery( session, entityClass, "mappedSuperclassProp = 'superclass_1'" );
executeSelectionQuery( session, entityClass, "childOneProp = 'child_one_1'" );
executeSelectionQuery( session, entityClass, "childTwoProp = 'child_two_1'" );
}
} );
}
private void executeSelectionQuery(SessionImplementor session, Class<?> entityClass, String whereClause) {
assertThat( session.createSelectionQuery(
String.format( "from %s where %s", entityClass.getName(), whereClause ),
entityClass
).getResultList() ).hasSize( 1 );
}
@Test
public void testMutationQuery(SessionFactoryScope scope) {
scope.inTransaction( session -> {
for ( PersistentClass pc : scope.getMetadataImplementor().getEntityBindings() ) {
// Run test for each class in the inheritance tree
final Class<?> entityClass = pc.getMappedClass();
executeMutationQuery( session, entityClass, "baseProp = 2" );
executeMutationQuery( session, entityClass, "mappedSuperclassProp = 'superclass_2'" );
executeMutationQuery( session, entityClass, "childOneProp = 'child_one_2'" );
executeMutationQuery( session, entityClass, "childTwoProp = 'child_two_2'" );
}
} );
}
private void executeMutationQuery(SessionImplementor session, Class<?> entityClass, String whereClause) {
assertThat( session.createMutationQuery( String.format(
"update %s set baseProp = 2 where %s",
entityClass.getSimpleName(),
whereClause
) ).executeUpdate() ).isEqualTo( 1 );
}
@Entity( name = "AbstractRootEntity" )
@Table( name = "entity" )
@DiscriminatorColumn( name = "disc_col", discriminatorType = DiscriminatorType.INTEGER )
public abstract static class AbstractRootEntity {
@Id
@GeneratedValue
private Long id;
private Integer baseProp;
public AbstractRootEntity() {
}
public AbstractRootEntity(Integer baseProp) {
this.baseProp = baseProp;
}
}
@MappedSuperclass
public abstract static class AbstractSuperclass extends AbstractRootEntity {
private String mappedSuperclassProp;
public AbstractSuperclass() {
}
public AbstractSuperclass(Integer baseProp, String mappedSuperclassProp) {
super( baseProp );
this.mappedSuperclassProp = mappedSuperclassProp;
}
}
@Entity( name = "ChildOneEntity" )
@DiscriminatorValue( "1" )
public static class ChildOneEntity extends AbstractSuperclass {
private String childOneProp;
public ChildOneEntity() {
}
public ChildOneEntity(Integer baseProp, String mappedSuperclassProp, String childOneProp) {
super( baseProp, mappedSuperclassProp );
this.childOneProp = childOneProp;
}
}
@Entity( name = "ChildTwoEntity" )
@DiscriminatorValue( "2" )
public static class ChildTwoEntity extends ChildOneEntity {
private String childTwoProp;
public ChildTwoEntity() {
}
public ChildTwoEntity(Integer baseProp, String mappedSuperclassProp, String childOneProp, String childTwoProp) {
super( baseProp, mappedSuperclassProp, childOneProp );
this.childTwoProp = childTwoProp;
}
}
}