/* | |
* 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()); | |
assertContainsSQL(" optimize for 8 row"); | |
} | |
} | |
else { | |
em.find(AllFieldTypes.class, 0); | |
if (dict instanceof DB2Dictionary ) { | |
assertEquals(1, sql.size()); | |
assertContainsSQL(" 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(); | |
} | |
} | |
} |