Skip to content

Commit

Permalink
HHH-17606 Add test for issue
Browse files Browse the repository at this point in the history
  • Loading branch information
mbladel authored and beikov committed Jan 5, 2024
1 parent 9c65a22 commit d41981e
Showing 1 changed file with 175 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,175 @@
/*
* 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.annotations.generics;

import java.util.List;

import org.hibernate.metamodel.model.domain.ManagedDomainType;
import org.hibernate.query.sqm.tree.domain.SqmPath;

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.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.ManyToOne;
import jakarta.persistence.MappedSuperclass;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.CriteriaQuery;
import jakarta.persistence.criteria.Join;
import jakarta.persistence.criteria.Root;

import static org.assertj.core.api.Assertions.assertThat;

/**
* @author Marco Belladelli
*/
@DomainModel( annotatedClasses = {
GenericMappedSuperclassNestedJoinTest.Selection.class,
GenericMappedSuperclassNestedJoinTest.SelectionProductRule.class,
GenericMappedSuperclassNestedJoinTest.SelectionProductRuleProductLink.class,

} )
@SessionFactory
@Jira( "https://hibernate.atlassian.net/browse/HHH-17606" )
public class GenericMappedSuperclassNestedJoinTest {
@BeforeAll
public void setUp(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final Selection s1 = new Selection();
s1.id = 1L;
s1.ident = "s1";
session.persist( s1 );
final SelectionProductRule s2 = new SelectionProductRule();
s2.id = 2L;
s2.ident = 2;
s2.parent = s1;
session.persist( s2 );
final SelectionProductRuleProductLink s3 = new SelectionProductRuleProductLink();
s3.id = 3L;
s3.parent = s2;
session.persist( s3 );
} );
}

@AfterAll
public void tearDown(SessionFactoryScope scope) {
scope.inTransaction( session -> {
session.createMutationQuery( "delete from SelectionProductRuleProductLink" ).executeUpdate();
session.createMutationQuery( "delete from SelectionProductRule" ).executeUpdate();
session.createMutationQuery( "delete from Selection" ).executeUpdate();
} );
}

@Test
public void testSimpleGenericJoin(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final List<Integer> resultList = session.createQuery(
"select p1.ident from SelectionProductRuleProductLink s join s.parent p1",
Integer.class
).getResultList();
assertThat( resultList ).containsOnly( 2 );
} );
}

@Test
public void testSimpleGenericJoinCriteria(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final CriteriaBuilder cb = session.getCriteriaBuilder();
final CriteriaQuery<Integer> cq = cb.createQuery( Integer.class );
final Root<SelectionProductRuleProductLink> root = cq.from( SelectionProductRuleProductLink.class );
final Join<SelectionProductRuleProductLink, Object> parent = root.join( "parent" );
assertThat( parent.getJavaType() ).isEqualTo( SeqOrderLinkObjectWithUserContext.class );
assertThat( parent.getModel() ).isSameAs( root.getModel().getAttribute( "parent" ) );
assertThat( ( (SqmPath<?>) parent ).getResolvedModel().getBindableJavaType() )
.isEqualTo( SelectionProductRule.class );
final List<Integer> resultList = session.createQuery( cq.select( parent.get( "ident" ) ) ).getResultList();
assertThat( resultList ).containsOnly( 2 );
} );
}

@Test
public void testNestedGenericJoin(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final List<String> resultList = session.createQuery(
"select p2.ident from SelectionProductRuleProductLink s join s.parent p1 join p1.parent p2",
String.class
).getResultList();
assertThat( resultList ).containsOnly( "s1" );
} );
}

@Test
public void testNestedGenericJoinCriteria(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final CriteriaBuilder cb = session.getCriteriaBuilder();
final CriteriaQuery<String> cq = cb.createQuery( String.class );
final Root<SelectionProductRuleProductLink> root = cq.from( SelectionProductRuleProductLink.class );
final Join<SelectionProductRuleProductLink, ?> p1 = root.join( "parent" );
assertThat( p1.getJavaType() ).isEqualTo( SeqOrderLinkObjectWithUserContext.class );
assertThat( p1.getModel() ).isSameAs( root.getModel().getAttribute( "parent" ) );
assertThat( ( (SqmPath<?>) p1 ).getResolvedModel().getBindableJavaType() )
.isEqualTo( SelectionProductRule.class );
final Join<Object, Object> p2 = p1.join( "parent" );
assertThat( p2.getJavaType() ).isEqualTo( SimpleObject.class );
final ManagedDomainType<?> joinType = (ManagedDomainType<?>) ( (SqmPath<?>) p1 ).getReferencedPathSource()
.getSqmPathType();
assertThat( p2.getModel() ).isSameAs( joinType.getAttribute( "parent" ) );
assertThat( ( (SqmPath<?>) p2 ).getResolvedModel().getBindableJavaType() )
.isEqualTo( Selection.class );
final List<String> resultList = session.createQuery( cq.select( p2.get( "ident" ) ) ).getResultList();
assertThat( resultList ).containsOnly( "s1" );
} );
}

@MappedSuperclass
public static abstract class SimpleObject {
@Id
protected Long id;
}

@MappedSuperclass
public abstract static class CommonLinkObject<T extends SimpleObject> extends SimpleObject {
@ManyToOne
protected T parent;
}

@MappedSuperclass
public abstract static class SeqOrderLinkObject<T extends SimpleObject> extends CommonLinkObject<T> {
}

@MappedSuperclass
public abstract static class SeqOrderLinkObjectWithUserContext<T extends SimpleObject>
extends SeqOrderLinkObject<T> {
}

@Entity( name = "Selection" )
public static class Selection extends SimpleObject {
private String ident;
}

@Entity( name = "SelectionProductRule" )
public static class SelectionProductRule extends SeqOrderLinkObjectWithUserContext<Selection> {
private Integer ident;
}

@MappedSuperclass
public abstract static class AbsProductRuleProductLink<T extends SeqOrderLinkObjectWithUserContext<?>>
extends SimpleObject {
@ManyToOne
protected T parent;
}

@Entity( name = "SelectionProductRuleProductLink" )
public static class SelectionProductRuleProductLink extends AbsProductRuleProductLink<SelectionProductRule> {
}
}

0 comments on commit d41981e

Please sign in to comment.