Package org.apache.openjpa.persistence.jdbc

Source Code of org.apache.openjpa.persistence.jdbc.TestOptimizeForClause

/*
* 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.persistence.OpenJPAEntityManagerSPI;
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) {
        OpenJPAEntityManagerSPI em = 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();
        }
    }
}
TOP

Related Classes of org.apache.openjpa.persistence.jdbc.TestOptimizeForClause

TOP
Copyright © 2015 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.