[#1280] tests for getCurrentSession()

Author: gavinking

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

@@ -27,12 +27,7 @@
 import jakarta.persistence.metamodel.EntityType;
 
 import static java.util.concurrent.TimeUnit.MINUTES;
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertFalse;
-import static org.junit.jupiter.api.Assertions.assertNotEquals;
-import static org.junit.jupiter.api.Assertions.assertNotNull;
-import static org.junit.jupiter.api.Assertions.assertTrue;
-import static org.junit.jupiter.api.Assertions.fail;
+import static org.junit.jupiter.api.Assertions.*;
 
 @Timeout(value = 10, timeUnit = MINUTES)
 
@@ -668,6 +663,38 @@ public void testForceFlushWithDelete(VertxTestContext context) {
 		);
 	}
 
+	@Test
+	public void testCurrentSession(VertxTestContext context) {
+		test( context,
+				getMutinySessionFactory().withSession(session ->
+						getMutinySessionFactory().withSession(s -> {
+							assertEquals(session, s);
+							Mutiny.Session currentSession = getMutinySessionFactory().getCurrentSession();
+							assertNotNull(currentSession);
+							assertTrue(currentSession.isOpen());
+							assertEquals(session, currentSession);
+							return Uni.createFrom().voidItem();
+						}).invoke(() -> assertNotNull(getMutinySessionFactory().getCurrentSession()))
+				).invoke(() -> assertNull(getMutinySessionFactory().getCurrentSession()))
+		);
+	}
+
+	@Test
+	public void testCurrentStatelessSession(VertxTestContext context) {
+		test( context,
+				getMutinySessionFactory().withStatelessSession(session ->
+						getMutinySessionFactory().withStatelessSession(s -> {
+							assertEquals(session, s);
+							Mutiny.StatelessSession currentSession = getMutinySessionFactory().getCurrentStatelessSession();
+							assertNotNull(currentSession);
+							assertTrue(currentSession.isOpen());
+							assertEquals(session, currentSession);
+							return Uni.createFrom().voidItem();
+						}).invoke(() -> assertNotNull(getMutinySessionFactory().getCurrentStatelessSession()))
+				).invoke(() -> assertNull(getMutinySessionFactory().getCurrentStatelessSession()))
+		);
+	}
+
 	private void assertThatPigsAreEqual(GuineaPig expected, GuineaPig actual) {
 		assertNotNull( actual );
 		assertEquals( expected.getId(), actual.getId() );

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

@@ -32,12 +32,7 @@
 
 import static java.util.concurrent.TimeUnit.MINUTES;
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertFalse;
-import static org.junit.jupiter.api.Assertions.assertInstanceOf;
-import static org.junit.jupiter.api.Assertions.assertNotEquals;
-import static org.junit.jupiter.api.Assertions.assertNotNull;
-import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.*;
 
 @Timeout(value = 10, timeUnit = MINUTES)
 
@@ -969,6 +964,42 @@ context, openSession()
 		);
 	}
 
+	@Test
+	public void testCurrentSession(VertxTestContext context) {
+		test( context,
+				getSessionFactory().withSession(session ->
+						getSessionFactory().withSession(s -> {
+							assertEquals(session, s);
+							Stage.Session currentSession = getSessionFactory().getCurrentSession();
+							assertNotNull(currentSession);
+							assertTrue(currentSession.isOpen());
+							assertEquals(session, currentSession);
+							return CompletionStages.voidFuture();
+						})
+						.thenAccept(v -> assertNotNull(getSessionFactory().getCurrentSession()))
+				)
+				.thenAccept(v -> assertNull(getSessionFactory().getCurrentSession()))
+		);
+	}
+
+	@Test
+	public void testCurrentStatelessSession(VertxTestContext context) {
+		test( context,
+				getSessionFactory().withStatelessSession(session ->
+						getSessionFactory().withStatelessSession(s -> {
+							assertEquals(session, s);
+							Stage.StatelessSession currentSession = getSessionFactory().getCurrentStatelessSession();
+							assertNotNull(currentSession);
+							assertTrue(currentSession.isOpen());
+							assertEquals(session, currentSession);
+							return CompletionStages.voidFuture();
+						})
+						.thenAccept(v -> assertNotNull(getSessionFactory().getCurrentStatelessSession()))
+				)
+				.thenAccept(v -> assertNull(getSessionFactory().getCurrentStatelessSession()))
+		);
+	}
+
 	private void assertThatPigsAreEqual(GuineaPig expected, GuineaPig actual) {
 		assertNotNull( actual );
 		assertEquals( expected.getId(), actual.getId() );