HBX-2743: Create Interface for CriteriaWrapper

jbt/src/main/java/org/hibernate/tool/orm/jbt/api/CriteriaWrapper.java

@@ -0,0 +1,14 @@
+package org.hibernate.tool.orm.jbt.api;
+
+import java.util.List;
+
+import org.hibernate.tool.orm.jbt.wrp.Wrapper;
+
+import jakarta.persistence.Query;
+
+public interface CriteriaWrapper extends Wrapper {
+
+	default void setMaxResults(int intValue) { ((Query)getWrappedObject()).setMaxResults(intValue); }
+	default List<?> list() { return ((Query)getWrappedObject()).getResultList(); }
+
+}

jbt/src/main/java/org/hibernate/tool/orm/jbt/internal/factory/CriteriaWrapperFactory.java

@@ -0,0 +1,15 @@
+package org.hibernate.tool.orm.jbt.internal.factory;
+
+import org.hibernate.tool.orm.jbt.api.CriteriaWrapper;
+
+import jakarta.persistence.Query;
+
+public class CriteriaWrapperFactory {
+
+	public static CriteriaWrapper createCriteriaWrapper(final Query wrappedCriteria) {
+		return new CriteriaWrapper() {
+			@Override public Query getWrappedObject() { return wrappedCriteria; }
+		};
+	}
+
+}

jbt/src/test/java/org/hibernate/tool/orm/jbt/api/CriteriaWrapperTest.java

@@ -0,0 +1,67 @@
+package org.hibernate.tool.orm.jbt.api;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertSame;
+
+import java.lang.reflect.InvocationHandler;
+import java.lang.reflect.Method;
+import java.lang.reflect.Proxy;
+import java.util.Arrays;
+import java.util.List;
+
+import org.hibernate.tool.orm.jbt.internal.factory.CriteriaWrapperFactory;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import jakarta.persistence.Query;
+
+public class CriteriaWrapperTest {
+
+	private static final List<String> RESULT_LIST = Arrays.asList("foo", "bar");
+
+	private CriteriaWrapper criteriaWrapper = null; 
+	private Query wrappedCriteria = null;
+
+	@BeforeEach
+	public void beforeEach() throws Exception {
+		wrappedCriteria = createQuery();
+		criteriaWrapper = CriteriaWrapperFactory.createCriteriaWrapper(wrappedCriteria);
+	}
+
+	@Test
+	public void testConstruction() {
+		assertNotNull(wrappedCriteria);
+		assertNotNull(criteriaWrapper);
+	}
+
+	@Test
+	public void testSetMaxResults() {
+		assertNotEquals(wrappedCriteria.getMaxResults(), Integer.MAX_VALUE);
+		criteriaWrapper.setMaxResults(Integer.MAX_VALUE);
+		assertEquals(wrappedCriteria.getMaxResults(), Integer.MAX_VALUE);
+	}
+
+	@Test
+	public void testList() {
+		assertSame(criteriaWrapper.list(), RESULT_LIST);
+	}
+
+	private Query createQuery() {
+		return (Query)Proxy.newProxyInstance(
+				getClass().getClassLoader(), 
+				new Class[] { Query.class }, 
+				new InvocationHandler () {
+					private int maxResults = 0;
+					@Override
+					public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
+						if (method.getName().equals("getMaxResults")) { return maxResults; }
+						else if (method.getName().equals("setMaxResults")) { maxResults = (int)args[0]; }
+						else if (method.getName().equals("getResultList")) { return RESULT_LIST; }
+						return null;
+					}				
+				});
+	}
+
+}