HHH-10950 - Add test for issue

hibernate-core/src/test/java/org/hibernate/test/criteria/SessionCreateQueryFromCriteriaTest.java

@@ -0,0 +1,149 @@
+/*
+ * 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.test.criteria;
+
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.Id;
+import javax.persistence.Table;
+import javax.persistence.criteria.CriteriaBuilder;
+import javax.persistence.criteria.CriteriaQuery;
+import javax.persistence.criteria.Root;
+import java.util.Optional;
+import java.util.stream.Stream;
+
+import org.hibernate.ScrollableResults;
+import org.hibernate.Session;
+
+import org.junit.After;
+import org.junit.Test;
+
+import org.hibernate.testing.TestForIssue;
+import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
+
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+
+/**
+ * @author Andrea Boriero
+ */
+@TestForIssue(jiraKey = "HHH-10950")
+public class SessionCreateQueryFromCriteriaTest extends BaseCoreFunctionalTestCase {
+
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class[] {TestEntity.class};
+	}
+
+	@After
+	public void tearDown() {
+		try (Session s = openSession()) {
+			session.getTransaction().begin();
+			try {
+				s.createQuery( "delete from TestEntity" ).executeUpdate();
+				s.getTransaction().commit();
+			}
+			catch (Exception e) {
+				if ( s.getTransaction().isActive() ) {
+					s.getTransaction().rollback();
+				}
+				throw e;
+			}
+		}
+	}
+
+	@Test
+	public void testUniqueResult() {
+		final String entityName = "expected";
+		try (Session session = openSession()) {
+			final CriteriaQuery<TestEntity> query = createTestEntityCriteriaQuery( entityName, session );
+			final Optional<TestEntity> result = session.createQuery( query ).uniqueResultOptional();
+			assertThat( result.isPresent(), is( false ) );
+		}
+	}
+
+	private CriteriaQuery<TestEntity> createTestEntityCriteriaQuery(
+			String entityName,
+			Session session) {
+		final CriteriaBuilder builder = session.getCriteriaBuilder();
+		final CriteriaQuery<TestEntity> query =
+				builder.createQuery( TestEntity.class );
+		final Root<TestEntity> fromTestEntity = query.from( TestEntity.class );
+		query.select( fromTestEntity );
+		query.where( builder.equal(
+				fromTestEntity.get( "name" ),
+				entityName
+		) );
+		return query;
+	}
+
+	@Test
+	public void testStreamMethod() {
+		final String entityName = "expected";
+		insertTestEntity( entityName );
+		try (Session session = openSession()) {
+			final CriteriaQuery<TestEntity> query = createTestEntityCriteriaQuery(
+					entityName,
+					session
+			);
+			final Stream<TestEntity> stream = session.createQuery( query ).stream();
+			assertThat( stream.count(), is( 1L ) );
+		}
+	}
+
+	@Test
+	public void testScrollMethod() {
+		final String entityName = "expected";
+		insertTestEntity( entityName );
+		try (Session session = openSession()) {
+			final CriteriaQuery<TestEntity> query = createTestEntityCriteriaQuery(
+					entityName,
+					session
+			);
+			try (final ScrollableResults scroll = session.createQuery( query ).scroll()) {
+				assertThat( scroll.first(), is( true ) );
+			}
+
+		}
+	}
+
+	private void insertTestEntity(String name) {
+		final TestEntity entity = new TestEntity();
+		entity.setName( name );
+		try (Session s = openSession()) {
+			session.getTransaction().begin();
+			try {
+				s.save( entity );
+				s.getTransaction().commit();
+			}
+			catch (Exception e) {
+				if ( s.getTransaction().isActive() ) {
+					s.getTransaction().rollback();
+				}
+				throw e;
+			}
+		}
+	}
+
+	@Entity(name = "TestEntity")
+	@Table(name = "TEST_ENTITY")
+	public static class TestEntity {
+		@Id
+		@GeneratedValue
+		private long id;
+
+		private String name;
+
+		public String getName() {
+			return name;
+		}
+
+		public void setName(String name) {
+			this.name = name;
+		}
+	}
+}