[hibernate#1979] Add tests for find with multiple ids

DavideD · Sep 16, 2024 · 65f69c3 · 65f69c3
1 parent 5ae62f7
commit 65f69c3
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/hibernate-reactive-core/src/test/java/org/hibernate/reactive/MutinySessionTest.java b/hibernate-reactive-core/src/test/java/org/hibernate/reactive/MutinySessionTest.java
@@ -68,6 +68,20 @@ private static String nameFromResult(List<Object> rowSet) {
 		}
 	}
 
+	@Test
+	public void reactiveFindMultipleIds(VertxTestContext context) {
+		final GuineaPig rump = new GuineaPig( 55, "Rumpelstiltskin" );
+		final GuineaPig emma = new GuineaPig( 77, "Emma" );
+		test( context, populateDB()
+				.chain( () -> getMutinySessionFactory().withTransaction( s -> s.persistAll( emma, rump ) ) )
+				.chain( () -> getMutinySessionFactory().withTransaction( s -> s.find( GuineaPig.class, emma.getId(), rump.getId() ) )
+				)
+				.invoke( pigs -> {
+					org.assertj.core.api.Assertions.assertThat( pigs ).containsExactlyInAnyOrder( emma, rump );
+				} )
+		);
+	}
+
 	@Test
 	public void sessionClear(VertxTestContext context) {
 		final GuineaPig guineaPig = new GuineaPig( 81, "Perry" );

diff --git a/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveSessionTest.java b/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveSessionTest.java
@@ -79,6 +79,21 @@ public void reactiveFind(VertxTestContext context) {
 		);
 	}
 
+	@Test
+	public void reactiveFindMultipleIds(VertxTestContext context) {
+		final GuineaPig rump = new GuineaPig( 55, "Rumpelstiltskin" );
+		final GuineaPig emma = new GuineaPig( 77, "Emma" );
+		test( context, populateDB()
+				.thenCompose( v -> getSessionFactory().withTransaction( s -> s.persist( emma, rump ) ) )
+				.thenCompose( v -> getSessionFactory().withTransaction( s -> s
+						.find( GuineaPig.class, emma.getId(), rump.getId() ) )
+				)
+				.thenAccept( pigs -> {
+					org.assertj.core.api.Assertions.assertThat( pigs ).containsExactlyInAnyOrder( emma, rump );
+				} )
+		);
+	}
+
 	@Test
 	public void sessionClear(VertxTestContext context) {
 		final GuineaPig guineaPig = new GuineaPig( 81, "Perry" );