[#3811] Add test for ReactiveSelectionQuery.getKeyedResultList

Author: dreab8

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

@@ -9,6 +9,7 @@
 import java.util.Objects;
 import java.util.concurrent.CompletionStage;
 
+import org.hibernate.query.Order;
 import org.hibernate.annotations.Filter;
 import org.hibernate.annotations.FilterDef;
 import org.hibernate.annotations.ParamDef;
@@ -163,6 +164,39 @@ public void testMaxResultsWithStage(VertxTestContext context) {
 		);
 	}
 
+	@Test
+	public void testReactiveKeyedPagination(VertxTestContext context) {
+		test( context, openSession()
+				.thenCompose( session -> {
+					var query = session.createSelectionQuery(
+							"from FamousPerson p where p.status = :status order by p.name, p.id",
+							FamousPerson.class
+					);
+					final var firstPage = first( 2 ).keyedBy( List.of(
+							Order.asc( FamousPerson.class, "name" ),
+							Order.asc( FamousPerson.class, "id" )
+					) );
+					return query.setParameter( "status", Status.LIVING )
+							.getReactiveKeyedResultList( firstPage )
+							.thenCompose( firstResults -> {
+								assertThat( firstResults.getResultList() ).containsExactly( margaret, rebeccaActress );
+								assertThat( firstResults.getKeyList() ).hasSize( 2 );
+								assertThat( firstResults.isFirstPage() ).isTrue();
+								assertThat( firstResults.isLastPage() ).isFalse();
+								assertThat( firstResults.getPreviousPage() ).isNull();
+								return query.getReactiveKeyedResultList( firstResults.getNextPage() )
+										.thenAccept( secondResults -> {
+											assertThat( secondResults.getResultList() ).containsExactly( rebeccaSinger );
+											assertThat( secondResults.getKeyList() ).hasSize( 1 );
+											assertThat( secondResults.isFirstPage() ).isFalse();
+											assertThat( secondResults.isLastPage() ).isTrue();
+											assertThat( secondResults.getPreviousPage() ).isNotNull();
+										} );
+							} );
+				} )
+		);
+	}
+
 	@Test
 	public void testMaxResultsWithStageAndPage(VertxTestContext context) {
 		test( context, enableFilter( openSession(), FamousPerson.IS_ALIVE_FILTER )