[#1893] Add test cases for fetch(getReference())

Author: DavideD

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

@@ -30,6 +30,7 @@
 import jakarta.persistence.Version;
 
 import static java.util.concurrent.TimeUnit.MINUTES;
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.hibernate.reactive.util.impl.CompletionStages.loop;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
@@ -231,6 +232,25 @@ public void testRefreshDetachedProxy(VertxTestContext context) {
 		);
 	}
 
+	@Test
+	public void testFetchOfReference(VertxTestContext context) {
+		final Book goodOmens = new Book( "Good Omens: The Nice and Accurate Prophecies of Agnes Nutter, Witch" );
+		final Author neil = new Author( "Neil Gaiman", goodOmens );
+		final Author terry = new Author( "Terry Pratchett", goodOmens );
+		goodOmens.getAuthors().add( neil );
+		goodOmens.getAuthors().add( terry );
+
+		test( context, getMutinySessionFactory()
+				.withTransaction( s -> s.persistAll( goodOmens, terry, neil ) )
+				.call( () -> getMutinySessionFactory().withSession( s -> s
+						// Not the most common case, but should be possible
+						.fetch( s.getReference( Book.class, goodOmens.getId() ) )
+						.chain( reference -> s.fetch( reference.getAuthors() ) )
+						.invoke( authors -> assertThat( authors ).containsExactlyInAnyOrder( terry, neil ) )
+				) )
+		);
+	}
+
 	@Entity(name = "Tome")
 	@Table(name = "TBook")
 	public static class Book {

integration-tests/bytecode-enhancements-it/src/test/java/org/hibernate/reactive/it/ReferenceBETest.java

@@ -22,6 +22,7 @@
 import io.vertx.junit5.VertxTestContext;
 
 import static java.util.concurrent.TimeUnit.MINUTES;
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.hibernate.reactive.util.impl.CompletionStages.completedFuture;
 import static org.hibernate.reactive.util.impl.CompletionStages.loop;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -212,4 +213,23 @@ public void testRefreshDetachedProxy(VertxTestContext context) {
 					  ) )
 		);
 	}
+
+	@Test
+	public void testFetchOfReference(VertxTestContext context) {
+		final Book goodOmens = new Book( "Good Omens: The Nice and Accurate Prophecies of Agnes Nutter, Witch" );
+		final Author neil = new Author( "Neil Gaiman", goodOmens );
+		final Author terry = new Author( "Terry Pratchett", goodOmens );
+		goodOmens.getAuthors().add( neil );
+		goodOmens.getAuthors().add( terry );
+
+		test( context, getMutinySessionFactory()
+				.withTransaction( s -> s.persistAll( goodOmens, terry, neil ) )
+				.call( () -> getMutinySessionFactory().withSession( s -> s
+						// Not the most common case, but should be possible
+						.fetch( s.getReference( Book.class, goodOmens.getId() ) )
+						.chain( reference -> s.fetch( reference.getAuthors() ) )
+						.invoke( authors -> assertThat( authors ).containsExactlyInAnyOrder( terry, neil ) )
+				) )
+		);
+	}
 }