[#1702] add test for upsert()

Author: blafond

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

@@ -26,6 +26,7 @@
 import jakarta.persistence.criteria.Root;
 
 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.assertNotNull;
@@ -37,7 +38,7 @@ public class ReactiveStatelessSessionTest extends BaseReactiveTest {
 
 	@Override
 	protected Collection<Class<?>> annotatedEntities() {
-		return List.of( GuineaPig.class );
+		return List.of( Record.class, GuineaPig.class );
 	}
 
 	@Test
@@ -206,6 +207,141 @@ private void assertThatPigsAreEqual( GuineaPig expected, GuineaPig actual) {
 		assertEquals( expected.getName(), actual.getName() );
 	}
 
+	/**
+	 * Mutiny version of Hibernate ORM org.hibernate.orm.test.stateless.UpsertTest
+	 */
+	@Test
+	public void testMutinyUpsertWithEntityName(VertxTestContext context) {
+		test( context, getMutinySessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( Record.class.getName(), new Record( 123L, "hello earth" ) )
+							  .call( () -> ss.upsert( Record.class.getName(), new Record( 456L, "hello mars" ) ) )
+					  )
+					  .call( v -> getMutinySessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .invoke( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "hello earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+					  .call( () -> getMutinySessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( Record.class.getName(), new Record( 123L, "goodbye earth" ) )
+					  ) )
+					  .call( v -> getMutinySessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .invoke( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "goodbye earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+		);
+	}
+
+	/**
+	 * CompletionStage version of Hibernate ORM org.hibernate.orm.test.stateless.UpsertTest
+	 */
+	@Test
+	public void testStageUpsert(VertxTestContext context) {
+		test( context, getSessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( new Record( 123L, "hello earth" ) )
+							  .thenCompose( v -> ss.upsert( new Record( 456L, "hello mars" ) ) )
+					  )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .thenAccept( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "hello earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( new Record( 123L, "goodbye earth" ) )
+					  ) )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .thenAccept( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "goodbye earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+		);
+	}
+
+
+	/**
+	 * CompletionStage version of Hibernate ORM org.hibernate.orm.test.stateless.UpsertTest
+	 */
+	@Test
+	public void testStageUpsertWithEntityName(VertxTestContext context) {
+		test( context, getSessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( Record.class.getName(), new Record( 123L, "hello earth" ) )
+							  .thenCompose( v -> ss.upsert( Record.class.getName(), new Record( 456L, "hello mars" ) ) )
+					  )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .thenAccept( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "hello earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+							  .upsert( Record.class.getName(), new Record( 123L, "goodbye earth" ) )
+					  ) )
+					  .thenCompose( v -> getSessionFactory().withStatelessTransaction( ss -> ss
+									  .createQuery( "from Record order by id", Record.class ).getResultList() )
+							  .thenAccept( results -> assertThat( results ).containsExactly(
+									  new Record( 123L, "goodbye earth" ),
+									  new Record( 456L, "hello mars" )
+							  ) )
+					  )
+		);
+	}
+
+	@Entity(name = "Record")
+	@Table(name = "Record")
+	public static class Record {
+		@Id
+		public Long id;
+		public String message;
+
+		Record(Long id, String message) {
+			this.id = id;
+			this.message = message;
+		}
+
+		Record() {
+		}
+
+		public Long getId() {
+			return id;
+		}
+
+		public String getMessage() {
+			return message;
+		}
+
+		public void setMessage(String msg) {
+			message = msg;
+		}
+
+		// Equals and HashCode for simplifying the test assertions,
+		// not to be taken as an example or for production.
+		@Override
+		public boolean equals(Object o) {
+			if ( this == o ) {
+				return true;
+			}
+			if ( o == null || getClass() != o.getClass() ) {
+				return false;
+			}
+			Record record = (Record) o;
+			return Objects.equals( id, record.id ) && Objects.equals( message, record.message );
+		}
+
+		@Override
+		public int hashCode() {
+			return Objects.hash( id, message );
+		}
+	}
+
 	@NamedQuery(name = "findbyname", query = "from GuineaPig where name=:n")
 	@NamedQuery(name = "updatebyname", query = "update GuineaPig set name='Y'")
 	@NamedQuery(name = "findall", query = "from GuineaPig")