From 9eaec94118049ad189d237519d9dc366a4a7733c Mon Sep 17 00:00:00 2001
From: "David J. Wisneski" <wisneskid@apache.org>
Date: Fri, 11 May 2007 15:44:13 +0000
Subject: [PATCH] OpenJPA 168  testcase

git-svn-id: https://svn.apache.org/repos/asf/incubator/openjpa/trunk@537213 13f79535-47bb-0310-9956-ffa450edef68
---
 .../jdbc/TestOptimizeForClause.java           | 112 ++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 openjpa-persistence-jdbc/src/test/java/org/apache/openjpa/persistence/jdbc/TestOptimizeForClause.java

diff --git a/openjpa-persistence-jdbc/src/test/java/org/apache/openjpa/persistence/jdbc/TestOptimizeForClause.java b/openjpa-persistence-jdbc/src/test/java/org/apache/openjpa/persistence/jdbc/TestOptimizeForClause.java
new file mode 100644
index 000000000..7ff53dae3
--- /dev/null
+++ b/openjpa-persistence-jdbc/src/test/java/org/apache/openjpa/persistence/jdbc/TestOptimizeForClause.java
@@ -0,0 +1,112 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.    
+ */
+package org.apache.openjpa.persistence.jdbc;
+
+import javax.persistence.Query;
+
+import org.apache.openjpa.persistence.test.SQLListenerTestCase;
+import org.apache.openjpa.persistence.simple.AllFieldTypes;
+import org.apache.openjpa.persistence.OpenJPAPersistence;
+import org.apache.openjpa.persistence.OpenJPAEntityManager;
+import org.apache.openjpa.persistence.OpenJPAQuery;
+import org.apache.openjpa.persistence.InvalidStateException;
+import org.apache.openjpa.jdbc.sql.DBDictionary;
+import org.apache.openjpa.jdbc.sql.DB2Dictionary;
+import org.apache.openjpa.jdbc.sql.HSQLDictionary;
+import org.apache.openjpa.jdbc.conf.JDBCConfiguration;
+
+public class TestOptimizeForClause
+    extends SQLListenerTestCase {
+
+    public void setUp() {
+        setUp(AllFieldTypes.class);
+    }
+
+    public void testOptimizeForClauseViaHint() {
+        testOptimizeForClause(true,false,false);
+    }
+
+    public void testOptimizeForClauseViaFind() {
+        testOptimizeForClause(false,true,false);
+    }
+    public void testOptimizeForClauseViaQueryHint() {
+        testOptimizeForClause(false,true,true);
+    }
+    public void testOptimizeForClause(boolean hint,
+        boolean find, boolean queryHint) {
+        OpenJPAEntityManager em =
+            OpenJPAPersistence.cast(emf.createEntityManager());
+        DBDictionary dict = ((JDBCConfiguration) em.getConfiguration())
+            .getDBDictionaryInstance();
+
+        // hsql doesn't support optimizing; circumvent the test
+        if (dict instanceof HSQLDictionary)
+            return;
+
+        sql.clear();
+        try {
+            em.getTransaction().begin();
+            if (hint || queryHint) {
+                if (hint) {
+                    Query q = em.createQuery(
+                    "select o from AllFieldTypes o where o.intField = :p");
+                    q.setParameter("p", 0);
+                    q.setHint("openjpa.hint.OptimizeResultCount"
+                         ,new Integer(8));
+                    q.getResultList();
+                }    
+                else {
+                    OpenJPAQuery q =  OpenJPAPersistence.cast (em.createQuery 
+                        ("select o from AllFieldTypes o where o.intField " +
+                         "= :p"));
+                    q.setParameter("p", 0);
+                    q.setHint(q.HINT_RESULT_COUNT, new Integer(8)); 
+                    q.getResultList();
+               }     
+               if (dict instanceof DB2Dictionary) {
+                   assertEquals(1, sql.size());
+                   assertSQL("SELECT t0.id, t0.booleanField, t0.byteField," +
+                       " t0.charField, t0.dateField, t0.doubleField, " +
+                       "t0.floatField, t0.intField, t0.longField, " +
+                       "t0.shortField, t0.stringField FROM AllFieldTypes " +
+                       "t0 WHERE \\(t0.intField = \\?\\)  optimize for 8 row");
+               }
+            }
+            else {
+                 em.find(AllFieldTypes.class, 0);
+                 if (dict instanceof DB2Dictionary ) {
+                    assertEquals(1, sql.size());
+                    assertSQL("SELECT t0.booleanField, t0.byteField, " +
+                        "t0.charField, t0.dateField, t0.doubleField, " +
+                        "t0.floatField, t0.intField, t0.longField, " +
+                        "t0.shortField, t0.stringField FROM AllFieldTypes" +
+                        " t0 WHERE t0.id = \\?  optimize for 1 row");
+                }
+                   
+            }
+        } catch (InvalidStateException pe) {
+            // if we're not using DB2, we expect an InvalidStateException.
+            if (dict instanceof DB2Dictionary)
+                throw pe;
+        } finally {
+            em.getTransaction().rollback();
+            em.close();
+        }
+    }
+}