HHH-18571 Added test case demonstrating issue

This commit is contained in:
Noel Trout 2024-09-05 10:55:41 +02:00 committed by Marco Belladelli
parent 626ef42e7d
commit 98d9ac3654
No known key found for this signature in database
GPG Key ID: DBD6E257ABE9EA3D
1 changed files with 139 additions and 0 deletions

View File

@ -0,0 +1,139 @@
package org.hibernate.orm.test.engine.spi;
import java.util.Set;
import org.hibernate.annotations.BatchSize;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.engine.spi.LoadQueryInfluencers;
import org.hibernate.metamodel.model.domain.NavigableRole;
import org.hibernate.persister.collection.CollectionPersister;
import org.hibernate.persister.entity.EntityPersister;
import org.hibernate.testing.orm.junit.DomainModel;
import org.hibernate.testing.orm.junit.ServiceRegistry;
import org.hibernate.testing.orm.junit.SessionFactory;
import org.hibernate.testing.orm.junit.SessionFactoryScope;
import org.hibernate.testing.orm.junit.Setting;
import org.junit.jupiter.api.Test;
import jakarta.persistence.Entity;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.GenerationType;
import jakarta.persistence.Id;
import jakarta.persistence.OneToMany;
import jakarta.persistence.Table;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
@DomainModel(annotatedClasses = {
LoadQueryInfluencersTest.EntityWithBatchSize1.class,
LoadQueryInfluencersTest.ChildEntity.class
})
@SessionFactory
@ServiceRegistry(
settings = {
@Setting(name = AvailableSettings.DEFAULT_BATCH_FETCH_SIZE, value = "100")
}
)
public class LoadQueryInfluencersTest {
@Test
public void usesCustomEntityBatchSizeForEffectivelyBatchLoadable(SessionFactoryScope scope) {
scope.inSession(
(nonTransactedSession) -> {
LoadQueryInfluencers influencers = new LoadQueryInfluencers( nonTransactedSession.getSessionFactory() );
assertTrue( influencers.getBatchSize() > 1, "Expecting default batch size > 1" );
EntityPersister persister = nonTransactedSession.getSessionFactory()
.getRuntimeMetamodels()
.getMappingMetamodel()
.getEntityDescriptor( EntityWithBatchSize1.class );
assertFalse( persister.isBatchLoadable(), "Incorrect persister batch loadable." );
assertEquals( 1, influencers.effectiveBatchSize( persister ), "Incorrect effective batch size." );
assertFalse(
influencers.effectivelyBatchLoadable( persister ),
"Incorrect effective batch loadable."
);
}
);
}
@Test
public void usesCustomCollectionBatchSizeForEffectivelyBatchLoadable(SessionFactoryScope scope) {
scope.inSession(
(nonTransactedSession) -> {
LoadQueryInfluencers influencers = new LoadQueryInfluencers( nonTransactedSession.getSessionFactory() );
assertTrue( influencers.getBatchSize() > 1, "Expecting default batch size > 1" );
EntityPersister entityPersister = nonTransactedSession.getSessionFactory()
.getRuntimeMetamodels()
.getMappingMetamodel()
.getEntityDescriptor( EntityWithBatchSize1.class );
NavigableRole collectionRole = entityPersister.getNavigableRole()
.append( "childrenWithBatchSize1" );
CollectionPersister persister = nonTransactedSession.getSessionFactory()
.getRuntimeMetamodels()
.getMappingMetamodel()
.findCollectionDescriptor( collectionRole.getFullPath() );
assertFalse( persister.isBatchLoadable(), "Incorrect persister batch loadable." );
assertEquals( 1, influencers.effectiveBatchSize( persister ), "Incorrect effective batch size." );
assertFalse(
influencers.effectivelyBatchLoadable( persister ),
"Incorrect effective batch loadable."
);
}
);
}
@Entity
@Table(name = "EntityWithBatchSize1")
@BatchSize(size = 1)
public class EntityWithBatchSize1 {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
@OneToMany
@BatchSize(size = 1)
private Set<ChildEntity> childrenWithBatchSize1;
public Long getId() {
return id;
}
public void setId(Long id) {
this.id = id;
}
public Set<ChildEntity> getChildrenWithBatchSize1() {
return childrenWithBatchSize1;
}
public void setChildrenWithBatchSize1(Set<ChildEntity> childrenWithBatchSize1) {
this.childrenWithBatchSize1 = childrenWithBatchSize1;
}
}
@Entity
@Table(name = "ChildEntity")
public class ChildEntity {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
public Long getId() {
return id;
}
public void setId(Long id) {
this.id = id;
}
}
}