[#2768] Test batching with ProxyConnection

Author: DavideD

hibernate-reactive-core/src/test/java/org/hibernate/reactive/BatchingConnectionTest.java

@@ -8,11 +8,13 @@
 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
 import org.hibernate.cfg.AvailableSettings;
 import org.hibernate.cfg.Configuration;
+import org.hibernate.reactive.mutiny.Mutiny;
 import org.hibernate.reactive.mutiny.impl.MutinySessionImpl;
 import org.hibernate.reactive.mutiny.impl.MutinyStatelessSessionImpl;
 import org.hibernate.reactive.pool.BatchingConnection;
 import org.hibernate.reactive.pool.ReactiveConnection;
 import org.hibernate.reactive.pool.impl.SqlClientConnection;
+import org.hibernate.reactive.stage.Stage;
 import org.hibernate.reactive.stage.impl.StageSessionImpl;
 import org.hibernate.reactive.stage.impl.StageStatelessSessionImpl;
 import org.hibernate.reactive.testing.SqlStatementTracker;
@@ -85,6 +87,56 @@ protected void assertConnectionIsLazy(ReactiveConnection connection, boolean sta
 				.isEqualTo( org.hibernate.reactive.pool.impl.SqlClientPool.class.getName() + "$ProxyConnection" );
 	}
 
+	@Test
+	public void testBatchingWithPersistAllAndProxyConnection(VertxTestContext context) {
+		Stage.Session session = getSessionFactory().createSession();
+		test( context, session
+				.persist(
+						new GuineaPig( 11, "One" ),
+						new GuineaPig( 22, "Two" ),
+						new GuineaPig( 33, "Three" )
+				)
+				// Auto-flush
+				.thenCompose( v -> session
+						.createSelectionQuery( "select name from GuineaPig", String.class )
+						.getResultList()
+						.thenAccept( names -> {
+							assertThat( names ).containsExactlyInAnyOrder( "One", "Two", "Three" );
+							assertThat( sqlTracker.getLoggedQueries() ).hasSize( 1 );
+							// Parameters are different for different dbs, so we cannot do an exact match
+							assertThat( sqlTracker.getLoggedQueries().get( 0 ) )
+									.startsWith( "insert into pig (name,version,id) values " );
+							sqlTracker.clear();
+						} )
+				)
+		);
+	}
+
+	@Test
+	public void testBatchingWithPersistAllAndProxyConnectionAndMutiny(VertxTestContext context) {
+		Mutiny.Session session = getMutinySessionFactory().createSession();
+		test( context, session
+				.persistAll(
+						new GuineaPig( 11, "One" ),
+						new GuineaPig( 22, "Two" ),
+						new GuineaPig( 33, "Three" )
+				)
+				// Auto-flush
+				.chain( v -> session
+						.createSelectionQuery( "select name from GuineaPig", String.class )
+						.getResultList()
+						.invoke( names -> {
+							assertThat( names ).containsExactlyInAnyOrder( "One", "Two", "Three" );
+							assertThat( sqlTracker.getLoggedQueries() ).hasSize( 1 );
+							// Parameters are different for different dbs, so we cannot do an exact match
+							assertThat( sqlTracker.getLoggedQueries().get( 0 ) )
+									.startsWith( "insert into pig (name,version,id) values " );
+							sqlTracker.clear();
+						} )
+				)
+		);
+	}
+
 	@Test
 	public void testBatchingWithPersistAll(VertxTestContext context) {
 		test( context, openSession().thenCompose( s -> s