HHH-19383 - Add a few more tests and put them together

Signed-off-by: Jan Schatteman <[email protected]>

Author: jrenaat

hibernate-core/src/test/java/org/hibernate/orm/test/query/sql/NativeQueryResultCheckingTests.java

@@ -0,0 +1,198 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.query.sql;
+
+import java.util.List;
+import java.util.Map;
+
+import jakarta.persistence.Tuple;
+import org.hibernate.orm.test.sql.hand.Person;
+import org.hibernate.query.NativeQuery;
+import org.hibernate.testing.orm.domain.StandardDomainModel;
+import org.hibernate.testing.orm.junit.DomainModel;
+import org.hibernate.testing.orm.junit.Jira;
+import org.hibernate.testing.orm.junit.SessionFactory;
+import org.hibernate.testing.orm.junit.SessionFactoryScope;
+import org.hibernate.testing.orm.domain.library.Book;
+import org.hibernate.type.StandardBasicTypes;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+
+/**
+ * @author Jan Schatteman
+ */
+@DomainModel(
+		standardModels = StandardDomainModel.LIBRARY
+)
+@SessionFactory
+@Jira("https://hibernate.atlassian.net/browse/HHH-19383")
+public class NativeQueryResultCheckingTests {
+
+	@Test
+	@Jira("https://hibernate.atlassian.net/browse/HHH-19376")
+	public void testForHHH19376(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					String sql = "SELECT p.*, COUNT(*) OVER() AS total_count "
+								+ "FROM Person p "
+								+ "WHERE p.name ILIKE :name "
+								+ "ORDER BY p.id";
+					// Declare Person as result type
+					NativeQuery<Person> query = session.createNativeQuery(sql, Person.class);
+					query.setParameter("name", "Ga%");
+					query.setMaxResults(2);
+					query.setFirstResult(0);
+					// Now mutate the result set mapping and verify an Exception is thrown
+					assertThrows( IllegalArgumentException.class,
+							() -> query.addScalar( "total_count", StandardBasicTypes.LONG).list() );
+				}
+		);
+	}
+
+	@Test
+	public void testOkMutateResultSetMappingWithString(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> session.createNativeQuery( "select isbn from Book", String.class )
+										.addScalar( "isbn", String.class )
+										.getResultList()
+				)
+		);
+	}
+
+	@Test
+	public void testNokMutateResultSetMappingWithString(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertThrows( IllegalArgumentException.class,
+								() -> session.createNativeQuery( "select title, isbn from Book", String.class )
+										.addScalar( "title", String.class )
+										.addScalar( "isbn", String.class )
+										.getResultList()
+						)
+		);
+	}
+
+	@Test
+	public void testOkMutateResultSetMappingWithBook(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> session.createNativeQuery( "select id, name from Book", Book.class )
+										.addScalar( "id", Integer.class )
+										.addScalar( "name", String.class )
+										.getResultList()
+						)
+		);
+	}
+
+	@Test
+	public void testNokMutateResultSetMappingWithBook(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertThrows( IllegalArgumentException.class,
+								() -> session.createNativeQuery( "select title, isbn from Book", Book.class )
+										// this mapping doesn't have an appropriate constructor in Book, should throw error
+										.addScalar( "title", String.class )
+										.addScalar( "isbn", String.class )
+										.getResultList()
+						)
+		);
+	}
+
+	@Test
+	public void testMutateResultSetMappingWithObjectArray(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> {
+									session.createNativeQuery( "select * from Book", Object[].class )
+											.addScalar( "id", Integer.class )
+											.addScalar( "name", String.class )
+											.getResultList();
+								}
+						)
+		);
+	}
+
+	@Test
+	public void testMutateResultSetMappingWithTuple(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> {
+									session.createNativeQuery( "select * from Book", Tuple.class )
+											.addScalar( "id", Integer.class )
+											.addScalar( "name", String.class )
+											.getResultList();
+								}
+						)
+		);
+	}
+
+	@Test
+	public void testMutateResultSetMappingWithMap(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> {
+									session.createNativeQuery( "select * from Book", Map.class )
+											.addScalar( "id", Integer.class )
+											.addScalar( "name", String.class )
+											.getResultList();
+								}
+						)
+		);
+	}
+
+	@Test
+	public void testMutateResultSetMappingWithList(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+						assertDoesNotThrow(
+								() -> {
+									session.createNativeQuery( "select * from Book", List.class )
+											.addScalar( "id", Integer.class )
+											.addScalar( "name", String.class )
+											.getResultList();
+								}
+						)
+		);
+	}
+
+	@Test
+	public void testRecordWithPrimitiveField(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session ->
+					assertDoesNotThrow(
+							() -> session.createNativeQuery( "select id, name from Person", Record.class)
+									// map a Long onto a primitive long, shouldn't throw an exception
+									.addScalar("id", Long.class)
+									.addScalar("name", String.class)
+									.getResultList()
+					)
+		);
+	}
+
+	static class Record {
+		long id;
+		String name;
+		public Record(long id, String name) {
+			this.id = id;
+			this.name = name;
+		}
+		long id() {
+			return id;
+		}
+		String name() {
+			return name;
+		}
+	}
+
+}