Fix

Author: DavideD

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

@@ -5,6 +5,7 @@
  */
 package org.hibernate.reactive;
 
+import java.util.concurrent.CompletableFuture;
 import javax.persistence.Entity;
 import javax.persistence.Id;
 
@@ -14,6 +15,7 @@
 
 import org.junit.Test;
 
+import io.smallrye.mutiny.Uni;
 import io.vertx.core.Context;
 import io.vertx.core.Vertx;
 import io.vertx.ext.unit.Async;
@@ -53,7 +55,7 @@ public void testPersistWithStage(TestContext testContext) throws Exception {
 		newContext.runOnContext( event -> test( async, testContext, session
 				.persist( new Competition( "Cheese Rolling" ) )
 				.thenCompose( v -> session.flush() )
-				.handle( (v, e) -> assertExceptionThrown( e ) ) )
+				.handle( (v, e) -> assertExceptionThrown( e ).join() ) )
 		);
 	}
 
@@ -70,7 +72,7 @@ public void testFindWithStage(TestContext testContext) throws Exception {
 		// Run test in the new context
 		newContext.runOnContext( event -> test( async, testContext, session
 				.find( Competition.class, "Chess boxing" )
-				.handle( (v, e) -> assertExceptionThrown( e ) ) )
+				.handle( (v, e) -> assertExceptionThrown( e ).join() ) )
 		);
 	}
 
@@ -88,8 +90,8 @@ public void testOnPersistWithMutiny(TestContext testContext) throws Exception {
 		newContext.runOnContext( event -> test( async, testContext, session
 				.persist( new Competition( "Cheese Rolling" ) )
 				.call( session::flush )
-				.onItem().failWith( () -> new AssertionError( "No exception thrown" ) )
-				.onFailure().recoverWithItem( e -> assertExceptionThrown( e ) ) )
+				.onItemOrFailure()
+				.transformToUni( (unused, e) -> Uni.createFrom().completionStage( assertExceptionThrown( e ) ) ) )
 		);
 	}
 
@@ -106,22 +108,25 @@ public void testFindWithMutiny(TestContext testContext) throws Exception {
 		// Run test in the new context
 		newContext.runOnContext( event -> test( async, testContext, session
 				.find( Competition.class, "Chess boxing" )
-				.onItem().failWith( () -> new AssertionError( "No exception thrown" ) )
-				.onFailure().recoverWithItem( e -> assertExceptionThrown( e ) ) )
+				.onItemOrFailure()
+				.transformToUni( (unused, e) -> Uni.createFrom().completionStage( assertExceptionThrown( e ) ) ) )
 		);
 	}
 
 	// Check that at least one exception has the expected message
-	private static <T> T assertExceptionThrown(Throwable e) {
+	private static CompletableFuture<Void> assertExceptionThrown(Throwable e) {
+		CompletableFuture<Void> result = new CompletableFuture<>();
 		Throwable t = e;
 		while ( t != null ) {
 			if ( t.getClass().equals( IllegalStateException.class )
 					&& t.getMessage().contains( ERROR_MESSAGE ) ) {
-				return null;
+				result.complete( null );
+				return result;
 			}
 			t = t.getCause();
 		}
-		throw new AssertionError( "Expected exception not thrown. Exception thrown: " + e );
+		result.completeExceptionally( new AssertionError( "Expected exception not thrown. Exception thrown: " + e ) );
+		return result;
 	}
 
 	@Entity