HHH-17594 HHH-17665 Add test for issue

This commit is contained in:
Marco Belladelli 2024-01-05 11:58:51 +01:00 committed by Christian Beikov
parent 225740bce5
commit 8ed1e1cb40
1 changed files with 199 additions and 0 deletions

View File

@ -0,0 +1,199 @@
/*
* 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.proxy.narrow;
import java.util.HashSet;
import java.util.Set;
import org.hibernate.Hibernate;
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.FetchType;
import jakarta.persistence.Id;
import jakarta.persistence.JoinColumn;
import jakarta.persistence.ManyToOne;
import jakarta.persistence.OneToMany;
import static jakarta.persistence.FetchType.LAZY;
import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Marco Belladelli
*/
@DomainModel( annotatedClasses = {
BidirectionalManyToOneNarrowingTest.Address.class,
BidirectionalManyToOneNarrowingTest.MyAddress.class,
BidirectionalManyToOneNarrowingTest.Message.class,
BidirectionalManyToOneNarrowingTest.AddressContainer.class
} )
@SessionFactory
@Jira( "https://hibernate.atlassian.net/browse/HHH-17594" )
@Jira( "https://hibernate.atlassian.net/browse/HHH-17665" )
public class BidirectionalManyToOneNarrowingTest {
@BeforeAll
public void setUp(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final MyAddress address = new MyAddress( 1L );
session.persist( address );
session.persist( new Message( 2L, address ) );
final AddressContainer relation = new AddressContainer();
relation.setId( 3L );
relation.setMyAddress( address );
session.persist( relation );
} );
}
@AfterAll
public void tearDown(SessionFactoryScope scope) {
scope.inTransaction( session -> {
session.createMutationQuery( "delete from Message" ).executeUpdate();
session.createMutationQuery( "delete from AddressContainer" ).executeUpdate();
session.createMutationQuery( "delete from Address" ).executeUpdate();
} );
}
@Test
public void testQuery(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final Message result = session.createQuery( "from Message", Message.class ).getSingleResult();
assertResult( result );
} );
}
@Test
public void testFind(SessionFactoryScope scope) {
scope.inTransaction( session -> {
final Message result = session.find( Message.class, 2L );
assertResult( result );
} );
}
private void assertResult(Message result) {
final Address address = result.getReceiverAddress();
assertThat( Hibernate.isInitialized( address ) ).isFalse();
assertThat( Hibernate.getClass( address ) ).isEqualTo( MyAddress.class );
final MyAddress myAddress = (MyAddress) Hibernate.unproxy( address );
final Set<AddressContainer> relations = myAddress.getAddressUserRelations();
assertThat( relations ).hasSize( 1 );
final MyAddress relatedAddress = relations.iterator().next().getMyAddress();
assertThat( relatedAddress.getId() ).isEqualTo( 1L );
assertThat( relatedAddress ).isSameAs( myAddress );
}
@Test
public void testProxyReuse(SessionFactoryScope scope) {
// uninitialized proxy
scope.inTransaction( session -> {
final Address address = session.getReference( Address.class, 1L );
assertThat( Hibernate.isInitialized( address ) ).isFalse();
final AddressContainer addressContainer = session.find( AddressContainer.class, 3L );
final MyAddress myAddress = addressContainer.getMyAddress();
assertThat( Hibernate.isInitialized( myAddress ) ).isFalse();
assertThat( myAddress.getId() ).isEqualTo( address.getId() );
} );
// initialized proxy
scope.inTransaction( session -> {
final Address address = session.getReference( Address.class, 1L );
assertThat( Hibernate.getClass( address ) ).isEqualTo( MyAddress.class );
assertThat( Hibernate.isInitialized( address ) ).isTrue();
final AddressContainer addressContainer = session.find( AddressContainer.class, 3L );
final MyAddress myAddress = addressContainer.getMyAddress();
assertThat( Hibernate.isInitialized( myAddress ) ).isTrue();
assertThat( myAddress.getId() ).isEqualTo( address.getId() );
} );
}
@Entity( name = "Address" )
public static abstract class Address {
@Id
private Long id;
public Address() {
}
public Address(Long id) {
this.id = id;
}
public Long getId() {
return id;
}
}
@Entity( name = "Message" )
public static class Message {
@Id
private Long id;
@ManyToOne( fetch = FetchType.LAZY )
@JoinColumn( name = "receiver_address_id" )
private Address receiverAddress;
public Message() {
}
public Message(Long id, Address receiverAddress) {
this.id = id;
this.receiverAddress = receiverAddress;
}
public Address getReceiverAddress() {
return receiverAddress;
}
}
@Entity( name = "MyAddress" )
public static class MyAddress extends Address {
@OneToMany( mappedBy = "myAddress" )
private Set<AddressContainer> addressContainers = new HashSet<>();
public MyAddress() {
}
public MyAddress(Long id) {
super( id );
}
public Set<AddressContainer> getAddressUserRelations() {
return addressContainers;
}
}
@Entity( name = "AddressContainer" )
public static class AddressContainer {
@Id
private Long id;
@ManyToOne( fetch = LAZY )
@JoinColumn( name = "address_id" )
private MyAddress myAddress;
public Long getId() {
return id;
}
public void setId(Long id) {
this.id = id;
}
public MyAddress getMyAddress() {
return myAddress;
}
public void setMyAddress(MyAddress myAddress) {
this.myAddress = myAddress;
}
}
}