work on tests for batch fetching

Author: gavinking

hibernate-core/src/test/java/org/hibernate/orm/test/bytecode/enhancement/lazy/proxy/BatchFetchProxyTest.java

@@ -197,6 +197,40 @@ public void testBatchEntityLoadThenModify(SessionFactoryScope scope) {
 		);
 	}
 
+	@Test
+	@JiraKey("HHH-11147")
+	public void testBatchEntityRemoval(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					final Statistics statistics = scope.getSessionFactory().getStatistics();
+					statistics.clear();
+
+					List<Employer> employers = new ArrayList<>();
+					for ( int i = 0 ; i < 5 ; i++ ) {
+						employers.add( session.getReference( Employer.class, i + 1) );
+					}
+
+					assertEquals( 0, statistics.getPrepareStatementCount() );
+
+					session.find( Employer.class, 0 );
+					session.find( Employer.class, 1 );
+					session.find( Employer.class, 2 );
+					session.find( Employer.class, 5 );
+
+					assertEquals( 1, statistics.getPrepareStatementCount() );
+
+					session.find( Employer.class, 6 );
+					session.find( Employer.class, 7 );
+
+					assertEquals( 3, statistics.getPrepareStatementCount() );
+
+					for ( Employer employer : employers ) {
+						assertTrue( Hibernate.isInitialized( employer ) );
+					}
+				}
+		);
+	}
+
 	@BeforeEach
 	public void setUpData(SessionFactoryScope scope) {
 		scope.inTransaction(

hibernate-core/src/test/java/org/hibernate/orm/test/fetching/BatchFetchingTest.java

@@ -12,6 +12,7 @@
 import jakarta.persistence.ManyToOne;
 import jakarta.persistence.OneToMany;
 
+import org.hibernate.annotations.BatchSize;
 import org.hibernate.annotations.NaturalId;
 import org.hibernate.orm.test.jpa.BaseEntityManagerFunctionalTestCase;
 
@@ -80,7 +81,7 @@ public static class Department {
 		private Long id;
 
 		@OneToMany(mappedBy = "department")
-		//@BatchSize(size = 5)
+		@BatchSize(size = 5)
 		private List<Employee> employees = new ArrayList<>();
 
 		//Getters and setters omitted for brevity