openjpa-persistence-jdbc/src/test/java/org/apache/openjpa/persistence/kernel/TestQueries.java

/*
 * TestQueries.java
 *
 * Created on October 13, 2006, 4:27 PM
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 */

/*
 * 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.kernel;

import java.util.Collection;
import java.util.Iterator;

import org.apache.openjpa.persistence.OpenJPAEntityManager;
import org.apache.openjpa.persistence.OpenJPAQuery;
import org.apache.openjpa.persistence.kernel.common.apps.RuntimeTest1;
import org.apache.openjpa.persistence.kernel.common.apps.RuntimeTest2;
import org.apache.openjpa.persistence.kernel.common.apps.RuntimeTest3;

public class TestQueries extends BaseKernelTest {

    private static final int CHILD_COUNT = 3;
    private int id = 10000;

    public TestQueries(String name) {
        super(name);
    }

    public TestQueries() {
    }

    @Override
    public void setUp()
        throws Exception {
        super.setUp(RuntimeTest1.class, RuntimeTest2.class, RuntimeTest3.class);
    }

    public void testSimpleQuery() {
        OpenJPAEntityManager pm = getPM();

        persist(newRuntimeTest1("SimpleQueryA", 50), pm);
        persist(newRuntimeTest2("SimpleQueryB", 50), pm);

        Collection results = runQuery(RuntimeTest1.class, false,
            "stringField = \'SimpleQueryA\'", pm);
        assertEquals(1, results.size());

        results = runQuery(RuntimeTest1.class, true, "intField1 = 50", pm);
        assertEquals(2, results.size());

        results = runQuery(RuntimeTest2.class, true, "intField1 = 50", pm);
        assertEquals(1, results.size());

        results = runQuery(RuntimeTest2.class, false, "intField1 = 50", pm);
        assertEquals(1, results.size());

        endEm(pm);
    }

    public void testAndQuery() {
        String query = "intField1 > 10 AND o.intField1 < 50";

        for (int currentCount = 1; currentCount < 15; currentCount++) {
            OpenJPAEntityManager pm = getPM();

            // make sure that none exist
            deleteByQuery(RuntimeTest1.class, false, query, pm);

            startTx(pm);
            for (int i = 1; i <= currentCount; i++)
                pm.persist(newRuntimeTest1("AndQueryTest", 30 + i));
            endTx(pm);

            // verify that the query works.
            Collection results = runQuery(RuntimeTest1.class, false,
                query, pm);
            assertEquals(currentCount, results.size());
            endEm(pm);
        }
    }

    public void testRelationQuery() {
        relationQuery(10, "JOE", 20);
        relationQuery(99, "BOB", 2);
        relationQuery(3, "MARTHA", 1);
        relationQuery(5, "brenda", 43);
        relationQuery(43, "SarA", 55);
    }

    public void relationQuery(int intField, String stringField, int count) {
        OpenJPAEntityManager pm = getPM();

        String query = "selfOneOne.intField1 = " + intField
            + " AND o.selfOneOne.stringField = '" + stringField + "'";
        deleteByQuery(RuntimeTest1.class, true, query, pm);

        // we go up the to max count, adding to the people, and validate
        // with a query each time.
        for (int currentcount = 0; currentcount <= count; currentcount++) {
            if (currentcount != 0)
                persist(newRuntimeTest2(stringField, intField), pm);

            Collection results = runQuery(RuntimeTest1.class, true, query, pm);
            assertEquals("query (" + query + ") failed to yield "
                + currentcount + " instances",
                currentcount * CHILD_COUNT, results.size());
        }

        endEm(pm);
    }

    public void testQueryExecuteThrowsExceptionWhenNoNTR() {
        OpenJPAEntityManager pm = getPM();
        startTx(pm);
        pm.persist(newRuntimeTest1("Query", 10));
        endTx(pm);

        pm.setNontransactionalRead(false);
//        OpenJPAQuery q = pm.createNativeQuery("",RuntimeTest1.class);
        OpenJPAQuery q = pm.createQuery("SELECT o FROM RuntimeTest1 o");
        try {
            q.getResultList();
            fail("Query.execute() should have thrown a JDOException when "
                + "PM is outside a Transaction and NTR==false");
        } catch (Exception jdoe) {
            // good
            startTx(pm);
            q.getResultList();
            rollbackTx(pm);
        }
    }

    /**
     * Delete the results of a query, so we can reinsert and test.
     */
    private void deleteByQuery(Class type, boolean subs, String filter,
        OpenJPAEntityManager pm) {
        startTx(pm);
        pm.createExtent(type, subs);
        //FIXME jthomas
//        OpenJPAQuery query = pm.newQuery(extent, filter);
//        Collection items = (Collection) query.execute();
        String cstrng = type.getSimpleName();

        OpenJPAQuery query =
            pm.createQuery("SELECT o FROM " + cstrng + " o WHERE o." + filter);
        query.setSubclasses(subs);
        Collection items = query.getResultList();
        for (Object item : items) pm.remove(item);

        //FIXME jthomas
        //query = pm.newQuery(extent, filter);

        query =
            pm.createQuery("SELECT o FROM " + cstrng + " o WHERE o." + filter);
        query.setSubclasses(subs);
        items = query.getResultList();
        endTx(pm);
        assertEquals("after deleting from query (" + query
            + "), there should have been zero items",
            0, items.size());
    }

    private Collection runQuery(Class type, boolean subs, String filter,
        OpenJPAEntityManager pm) {
        pm.createExtent(type, subs);
        //FIXME jthomas
        //Query query = pm.newQuery(extent, filter);
        String cstrng = type.getName();
        OpenJPAQuery query =
            pm.createQuery("SELECT o FROM " + cstrng + " o WHERE o." + filter);
        query.setSubclasses(subs);
        Collection results = query.getResultList();
        return results;
    }

    private void persist(RuntimeTest1 parent, OpenJPAEntityManager pm) {
        RuntimeTest1 child;
        for (int i = 0; i < CHILD_COUNT; i++) {
            child = newRuntimeTest1("CHILD" + i, i * 10);
            child.setSelfOneOne(parent);
            parent.getSelfOneMany().add(child);
        }

        startTx(pm);
        pm.persist(parent);
        endTx(pm);
    }

    private RuntimeTest1 newRuntimeTest1(String stringField, int intField) {
        RuntimeTest1 pc = new RuntimeTest1(stringField, intField);
        pc.setIntField(id++);
        return pc;
    }

    private RuntimeTest2 newRuntimeTest2(String stringField, int intField) {
        RuntimeTest2 pc = new RuntimeTest2(stringField, intField);
        pc.setIntField(id++);
        return pc;
    }
}