HHH-19477 Test aggressive release mode for HQL queries

Author: beikov

hibernate-core/src/test/java/org/hibernate/orm/test/jdbc/internal/AggressiveReleaseTest.java

@@ -20,6 +20,7 @@
 
 import org.hibernate.testing.boot.BasicTestingJdbcServiceImpl;
 import org.hibernate.testing.orm.junit.BaseSessionFactoryFunctionalTest;
+import org.hibernate.testing.orm.junit.Jira;
 import org.hibernate.testing.orm.junit.RequiresDialect;
 import org.hibernate.testing.orm.jdbc.PreparedStatementSpyConnectionProvider;
 import org.junit.jupiter.api.AfterEach;
@@ -118,9 +119,9 @@ protected void cleanupTest() throws Exception {
 
 	@Test
 	public void testBasicRelease() {
-		connectionProvider.clear();
 		ResourceRegistry registry = sessionFactoryScope().fromSession(
 				session -> {
+					connectionProvider.clear();
 					JdbcCoordinatorImpl jdbcCoord = (JdbcCoordinatorImpl) session.getJdbcCoordinator();
 					ResourceRegistry resourceRegistry = jdbcCoord.getLogicalConnection().getResourceRegistry();
 					try {
@@ -154,9 +155,9 @@ public void testBasicRelease() {
 
 	@Test
 	public void testReleaseCircumventedByHeldResources() {
-		connectionProvider.clear();
 		ResourceRegistry registry = sessionFactoryScope().fromSession(
 				session -> {
+					connectionProvider.clear();
 					JdbcCoordinatorImpl jdbcCoord = (JdbcCoordinatorImpl) session.getJdbcCoordinator();
 					ResourceRegistry resourceRegistry = jdbcCoord.getLogicalConnection().getResourceRegistry();
 
@@ -217,10 +218,9 @@ public void testReleaseCircumventedByHeldResources() {
 
 	@Test
 	public void testReleaseCircumventedManually() {
-		connectionProvider.clear();
-		connectionProvider.clear();
 		ResourceRegistry registry = sessionFactoryScope().fromSession(
 				session -> {
+					connectionProvider.clear();
 					JdbcCoordinatorImpl jdbcCoord = (JdbcCoordinatorImpl) session.getJdbcCoordinator();
 					ResourceRegistry resourceRegistry = jdbcCoord.getLogicalConnection().getResourceRegistry();
 
@@ -271,4 +271,20 @@ public void testReleaseCircumventedManually() {
 		assertEquals( 0, connectionProvider.getAcquiredConnections().size() );
 		assertEquals( 2, connectionProvider.getReleasedConnections().size() );
 	}
+
+	@Test
+	@Jira("https://hibernate.atlassian.net/browse/HHH-19477")
+	public void testHql() {
+		sessionFactoryScope().inTransaction( session -> {
+			connectionProvider.clear();
+			JdbcCoordinatorImpl jdbcCoord = (JdbcCoordinatorImpl) session.getJdbcCoordinator();
+			ResourceRegistry resourceRegistry = jdbcCoord.getLogicalConnection().getResourceRegistry();
+
+			session.createSelectionQuery( "select 1" ).uniqueResult();
+
+			assertFalse( resourceRegistry.hasRegisteredResources() );
+			assertEquals( 0, connectionProvider.getAcquiredConnections().size() );
+			assertEquals( 1, connectionProvider.getReleasedConnections().size() );
+		} );
+	}
 }