Skip to content

Commit

Permalink
[hibernate#1885] Add test for createMutationQuery methods that accept…
Browse files Browse the repository at this point in the history 
… insert and insertSelect criterias
  • Loading branch information
@dreab8 @DavideD
dreab8 authored and DavideD committed Feb 3, 2025
1 parent 229a3a3 commit d9cd9bf
Showing 1 changed file with 124 additions and 1 deletion.
125 changes: 124 additions & 1 deletion hibernate-reactive-core/src/test/java/org/hibernate/reactive/CriteriaMutationQueryTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,10 @@
import java.util.Objects;

import org.hibernate.query.criteria.HibernateCriteriaBuilder;
import org.hibernate.query.criteria.JpaCriteriaInsertSelect;
import org.hibernate.query.criteria.JpaCriteriaInsertValues;
import org.hibernate.query.criteria.JpaCriteriaQuery;
import org.hibernate.query.criteria.JpaRoot;
import org.hibernate.reactive.mutiny.Mutiny;
import org.hibernate.reactive.stage.Stage;

Expand All @@ -21,6 +25,7 @@
import jakarta.persistence.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.Table;
import jakarta.persistence.Tuple;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.CriteriaDelete;
import jakarta.persistence.criteria.CriteriaUpdate;
Expand All @@ -31,7 +36,10 @@

@Timeout(value = 10, timeUnit = MINUTES)
public class CriteriaMutationQueryTest extends BaseReactiveTest {
Flour spelt = new Flour( 1, "Spelt", "An ancient grain, is a hexaploid species of wheat.", "Wheat flour" );
private static final Integer SPELT_ID = 1;
private static final String SPELT_NAME = "Spelt";
private static final String SPELT_TYPE = "Wheat flour";
Flour spelt = new Flour( SPELT_ID, SPELT_NAME, "An ancient grain, is a hexaploid species of wheat.", SPELT_TYPE );
Flour rye = new Flour( 2, "Rye", "Used to bake the traditional sourdough breads of Germany.", "Wheat flour" );
Flour almond = new Flour( 3, "Almond", "made from ground almonds.", "Gluten free" );

Expand Down Expand Up @@ -102,6 +110,92 @@ public void testMutinyDeleteCriteriaQuery(VertxTestContext context) {
);
}

@Test
public void testStageInsertCriteriaQuery(VertxTestContext context) {
final int id = 4;
final String flourName = "Rye";
final String flourDescription = "Used to bake the traditional sourdough breads of Germany.";
final String flourType = "Wheat flour";
test( context, getSessionFactory()
.withTransaction( s -> s
.createMutationQuery( insertCriteria( getCriteriaBuilder( s ), id, flourName, flourDescription, flourType ) )
.executeUpdate()
)
.thenAccept( resultCount -> assertThat( resultCount ).isEqualTo( 1 ) )
.thenCompose( v -> getSessionFactory()
.withTransaction( s -> s.find( Flour.class, id ) ) )
.thenAccept( result -> {
assertThat( result ).isNotNull();
assertThat( result.name ).isEqualTo( flourName );
assertThat( result.description ).isEqualTo( flourDescription );
assertThat( result.type ).isEqualTo( flourType );
} )
);
}

@Test
public void testMutinyInsertCriteriaQuery(VertxTestContext context) {
final int id = 4;
final String flourName = "Almond";
final String flourDescription = "made from ground almonds.";
final String flourType = "Gluten free";
test( context, getMutinySessionFactory()
.withTransaction( s -> s
.createMutationQuery( insertCriteria( getCriteriaBuilder( s ), id, flourName, flourDescription, flourType ) )
.executeUpdate()
)
.invoke( resultCount -> assertThat( resultCount ).isEqualTo( 1 ) )
.chain( v -> getMutinySessionFactory()
.withTransaction( s -> s.find( Flour.class, id ) ) )
.invoke( result -> {
assertThat( result ).isNotNull();
assertThat( result.name ).isEqualTo( flourName );
assertThat( result.description ).isEqualTo( flourDescription );
assertThat( result.type ).isEqualTo( flourType );
} )
);
}

@Test
public void testStageInsertSelectCriteriaQuery(VertxTestContext context) {
final int idOfTheNewFlour = 4;
test( context, getSessionFactory()
.withTransaction( s -> s
.createMutationQuery( insertSelectCriteria( getCriteriaBuilder( s ), idOfTheNewFlour ) )
.executeUpdate()
)
.thenAccept( resultCount -> assertThat( resultCount ).isEqualTo( 1 ) )
.thenCompose( v -> getSessionFactory()
.withTransaction( s -> s.find( Flour.class, idOfTheNewFlour ) ) )
.thenAccept( result -> {
assertThat( result ).isNotNull();
assertThat( result.name ).isEqualTo( SPELT_NAME );
assertThat( result.description ).isNull();
assertThat( result.type ).isEqualTo( SPELT_TYPE );
} )
);
}

@Test
public void testMutinyInsertSelectCriteriaQuery(VertxTestContext context) {
final int idOfTheNewFlour = 4;
test( context, getMutinySessionFactory()
.withTransaction( s -> s
.createMutationQuery( insertSelectCriteria( getCriteriaBuilder( s ), idOfTheNewFlour ) )
.executeUpdate()
)
.invoke( resultCount -> assertThat( resultCount ).isEqualTo( 1 ) )
.chain( v -> getMutinySessionFactory()
.withTransaction( s -> s.find( Flour.class, idOfTheNewFlour ) ) )
.invoke( result -> {
assertThat( result ).isNotNull();
assertThat( result.name ).isEqualTo( SPELT_NAME );
assertThat( result.description ).isNull();
assertThat( result.type ).isEqualTo( SPELT_TYPE );
} )
);
}

private CriteriaUpdate<Flour> criteriaUpdate(CriteriaBuilder cb, String updatedDescription, Flour rye) {
CriteriaUpdate<Flour> criteriaUpdate = cb.createCriteriaUpdate( Flour.class );
Root<Flour> from = criteriaUpdate.from( Flour.class );
Expand All @@ -117,6 +211,35 @@ private CriteriaDelete<Flour> criteriaUpdate(CriteriaBuilder criteriaBuilder) {
return criteriaDelete;
}

private static JpaCriteriaInsertValues<Flour> insertCriteria(HibernateCriteriaBuilder cb, int id, String name, String description, String type) {
JpaCriteriaInsertValues<Flour> insert = cb.createCriteriaInsertValues( Flour.class );
JpaRoot<Flour> flour = insert.getTarget();
insert.setInsertionTargetPaths( flour.get( "id" ), flour.get( "name" ), flour.get( "description" ), flour.get( "type" ) );
insert.values( cb.values( cb.value( id ), cb.value( name ), cb.value( description ), cb.value( type ) ) );
return insert;
}

private static JpaCriteriaInsertSelect<Flour> insertSelectCriteria(
HibernateCriteriaBuilder cb,
int idOfTheNewFlour) {
/*
The query executes and insert of Flour with id equals to 2 a name and type
selected from the existing spelt flour saved in the db
*/
JpaCriteriaInsertSelect<Flour> insertSelect = cb.createCriteriaInsertSelect( Flour.class );
// columns to insert
JpaRoot<Flour> flour = insertSelect.getTarget();
insertSelect.setInsertionTargetPaths( flour.get( "id" ), flour.get( "name" ), flour.get( "type" ) );
// select query
JpaCriteriaQuery<Tuple> select = cb.createQuery( Tuple.class );
JpaRoot<Flour> root = select.from( Flour.class );
select.multiselect( cb.literal( idOfTheNewFlour ), root.get( "name" ), root.get( "type" ) );
select.where( cb.equal( root.get( "id" ), SPELT_ID ) );

insertSelect.select( select );
return insertSelect;
}

private static HibernateCriteriaBuilder getCriteriaBuilder(Stage.Session session) {
return session.getFactory().getCriteriaBuilder();
}
Expand Down

0 comments on commit d9cd9bf

Please sign in to comment.