HHH-16062 Add test for issue
This commit is contained in:
Marco Belladelli
Christian Beikov
parent
528d897c4d
commit
136169d43a
|
|
@ -0,0 +1,192 @@
|
|||
/*
|
||||
* Hibernate, Relational Persistence for Idiomatic Java
|
||||
*
|
||||
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
|
||||
* See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
|
||||
*/
|
||||
|
||||
/*
|
||||
* Hibernate, Relational Persistence for Idiomatic Java
|
||||
*
|
||||
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
|
||||
* See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
|
||||
*/
|
||||
package org.hibernate.orm.test.query.criteria;
|
||||
|
||||
import java.sql.SQLException;
|
||||
import java.util.Map;
|
||||
|
||||
import org.hibernate.Session;
|
||||
import org.hibernate.cfg.AvailableSettings;
|
||||
import org.hibernate.jpa.SpecHints;
|
||||
import org.hibernate.query.sqm.internal.SqmCriteriaNodeBuilder;
|
||||
import org.hibernate.query.sqm.tree.insert.SqmInsertSelectStatement;
|
||||
import org.hibernate.query.sqm.tree.select.SqmSelectStatement;
|
||||
|
||||
import org.hibernate.testing.orm.jdbc.PreparedStatementSpyConnectionProvider;
|
||||
import org.hibernate.testing.orm.junit.DialectFeatureChecks;
|
||||
import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
|
||||
import org.hibernate.testing.orm.junit.JiraKey;
|
||||
import org.hibernate.testing.orm.junit.Jpa;
|
||||
import org.hibernate.testing.orm.junit.RequiresDialectFeature;
|
||||
import org.hibernate.testing.orm.junit.Setting;
|
||||
import org.hibernate.testing.orm.junit.SettingProvider;
|
||||
import org.junit.jupiter.api.BeforeAll;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import jakarta.persistence.Entity;
|
||||
import jakarta.persistence.GeneratedValue;
|
||||
import jakarta.persistence.Id;
|
||||
import jakarta.persistence.Table;
|
||||
import jakarta.persistence.Tuple;
|
||||
import jakarta.persistence.criteria.CriteriaDelete;
|
||||
import jakarta.persistence.criteria.CriteriaQuery;
|
||||
import jakarta.persistence.criteria.CriteriaUpdate;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.fail;
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
|
||||
/**
|
||||
* @author Marco Belladelli
|
||||
*/
|
||||
@Jpa(
|
||||
annotatedClasses = CriteriaTimeoutTest.AnEntity.class,
|
||||
integrationSettings = @Setting(name = SpecHints.HINT_SPEC_QUERY_TIMEOUT, value = "123000"),
|
||||
settingProviders = {
|
||||
@SettingProvider(
|
||||
settingName = AvailableSettings.CONNECTION_PROVIDER,
|
||||
provider = CriteriaTimeoutTest.SpyConnectionProviderSettingProvider.class)
|
||||
}
|
||||
)
|
||||
@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsJdbcDriverProxying.class)
|
||||
@JiraKey("HHH-16062")
|
||||
public class CriteriaTimeoutTest {
|
||||
private PreparedStatementSpyConnectionProvider connectionProvider;
|
||||
|
||||
@BeforeAll
|
||||
public void init(EntityManagerFactoryScope scope) {
|
||||
final Map<String, Object> props = scope.getEntityManagerFactory().getProperties();
|
||||
connectionProvider = (PreparedStatementSpyConnectionProvider) props.get( AvailableSettings.CONNECTION_PROVIDER );
|
||||
}
|
||||
|
||||
@BeforeEach
|
||||
public void clear() {
|
||||
connectionProvider.clear();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateQueryCriteriaQuery(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final CriteriaQuery<AnEntity> criteriaSelect = entityManager.getCriteriaBuilder()
|
||||
.createQuery( AnEntity.class );
|
||||
criteriaSelect.select( criteriaSelect.from( AnEntity.class ) );
|
||||
entityManager.createQuery( criteriaSelect ).getResultList();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateQueryCriteriaUpdate(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final CriteriaUpdate<AnEntity> criteriaUpdate = entityManager.getCriteriaBuilder()
|
||||
.createCriteriaUpdate( AnEntity.class );
|
||||
criteriaUpdate.set( "name", "abc" );
|
||||
entityManager.createQuery( criteriaUpdate ).executeUpdate();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateQueryCriteriaDelete(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final CriteriaDelete<AnEntity> criteriaDelete = entityManager.getCriteriaBuilder()
|
||||
.createCriteriaDelete( AnEntity.class );
|
||||
entityManager.createQuery( criteriaDelete ).executeUpdate();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateMutationQueryCriteriaUpdate(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final CriteriaUpdate<AnEntity> criteriaUpdate = entityManager.getCriteriaBuilder()
|
||||
.createCriteriaUpdate( AnEntity.class );
|
||||
criteriaUpdate.set( "name", "abc" );
|
||||
final Session session = entityManager.unwrap( Session.class );
|
||||
session.createMutationQuery( criteriaUpdate ).executeUpdate();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateMutationQueryCriteriaDelete(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final CriteriaDelete<AnEntity> criteriaDelete = entityManager.getCriteriaBuilder()
|
||||
.createCriteriaDelete( AnEntity.class );
|
||||
final Session session = entityManager.unwrap( Session.class );
|
||||
session.createMutationQuery( criteriaDelete ).executeUpdate();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateMutationQueryCriteriaInsertSelect(EntityManagerFactoryScope scope) {
|
||||
scope.inTransaction( entityManager -> {
|
||||
final Session session = entityManager.unwrap( Session.class );
|
||||
final SqmCriteriaNodeBuilder criteriaBuilder = (SqmCriteriaNodeBuilder) session.getCriteriaBuilder();
|
||||
final SqmInsertSelectStatement<AnEntity> insertSelect = criteriaBuilder
|
||||
.createCriteriaInsertSelect( AnEntity.class );
|
||||
final SqmSelectStatement<Tuple> select = criteriaBuilder.createQuery( Tuple.class );
|
||||
select.multiselect( select.from( AnEntity.class ).get( "name" ) );
|
||||
insertSelect.addInsertTargetStateField( insertSelect.getTarget().get( "name" ) );
|
||||
insertSelect.setSelectQueryPart( select.getQuerySpec() );
|
||||
session.createMutationQuery( insertSelect ).executeUpdate();
|
||||
verifyQuerySetTimeoutWasCalled();
|
||||
} );
|
||||
}
|
||||
|
||||
private void verifyQuerySetTimeoutWasCalled() {
|
||||
try {
|
||||
verify(
|
||||
connectionProvider.getPreparedStatements().get( 0 ),
|
||||
times( 1 )
|
||||
).setQueryTimeout( 123 );
|
||||
}
|
||||
catch (SQLException e) {
|
||||
fail( "should not have thrown exception" );
|
||||
}
|
||||
}
|
||||
|
||||
@Entity(name = "AnEntity")
|
||||
@Table(name = "AnEntity")
|
||||
public static class AnEntity {
|
||||
@Id
|
||||
@GeneratedValue
|
||||
private Integer id;
|
||||
|
||||
private String name;
|
||||
|
||||
public Integer getId() {
|
||||
return id;
|
||||
}
|
||||
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
public void setName(String name) {
|
||||
this.name = name;
|
||||
}
|
||||
}
|
||||
|
||||
public static class SpyConnectionProviderSettingProvider
|
||||
implements SettingProvider.Provider<PreparedStatementSpyConnectionProvider> {
|
||||
@Override
|
||||
public PreparedStatementSpyConnectionProvider getSetting() {
|
||||
return new PreparedStatementSpyConnectionProvider( true, false );
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue