| /* |
| * 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.inheritance.jointable; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.List; |
| |
| import javax.persistence.EntityManager; |
| import javax.persistence.Query; |
| |
| import org.apache.openjpa.persistence.ArgumentException; |
| import org.apache.openjpa.persistence.test.SQLListenerTestCase; |
| |
| |
| public class TestInheritanceTypeJoinedQuery extends SQLListenerTestCase { |
| private int numPTEmployees = 1; |
| private int numFTEmployees = 2; |
| private int numContractors = 3; |
| |
| @Override |
| public void setUp() { |
| setUp(Contractor.class, Employee.class, Department.class, Person.class, |
| FulltimeEmployee.class, ParttimeEmployee.class, |
| CLEAR_TABLES); |
| populate(); |
| } |
| |
| public void populate() { |
| EntityManager em = emf.createEntityManager(); |
| |
| Department d = new Department("IT"); |
| for (int i = 0; i < numContractors; i++) { |
| Contractor c = new Contractor("ctr" + i); |
| c.setDept(d); |
| em.persist(c); |
| } |
| em.persist(d); |
| |
| for (int i = 0; i < numFTEmployees; i++) { |
| FulltimeEmployee f = new FulltimeEmployee("ftemp" + i); |
| f.setDept(d); |
| em.persist(f); |
| } |
| |
| for (int i = 0; i < numPTEmployees; i++) { |
| ParttimeEmployee p = new ParttimeEmployee("ptemp" + i); |
| p.setDept(d); |
| em.persist(p); |
| } |
| |
| em.getTransaction().begin(); |
| em.getTransaction().commit(); |
| em.close(); |
| } |
| |
| public void testInheritanceJoinedTypeOperator() { |
| EntityManager em = emf.createEntityManager(); |
| Query q = null; |
| String qS = null; |
| List rs = null; |
| |
| qS = "SELECT p FROM Person p where TYPE(p) <> Contractor"; |
| q = em.createQuery(qS); |
| rs = q.getResultList(); |
| assertEquals(numPTEmployees + numFTEmployees, rs.size()); |
| for (int i = 0; i < rs.size(); i++){ |
| Object obj = rs.get(i); |
| assertTrue((obj instanceof ParttimeEmployee) || (obj instanceof FulltimeEmployee)); |
| } |
| |
| qS = "SELECT p FROM Person p where TYPE(p) = Contractor"; |
| q = em.createQuery(qS); |
| rs = q.getResultList(); |
| assertEquals(numContractors, rs.size()); |
| for (int i = 0; i < rs.size(); i++) |
| assertTrue(rs.get(i) instanceof Contractor); |
| |
| qS = "select p from Person p where TYPE(p) in (?1) order by p.name"; |
| q = em.createQuery(qS).setParameter(1, Contractor.class); |
| rs = q.getResultList(); |
| assertEquals(numContractors, rs.size()); |
| for (int i = 0; i < rs.size(); i++) |
| assertTrue(rs.get(i) instanceof Contractor); |
| |
| qS = "select p from Person p where TYPE(p) in ?1 order by p.name"; |
| Collection<Class<?>> params = new ArrayList<>(2); |
| params.add(Contractor.class); |
| params.add(Employee.class); |
| try { |
| q = em.createQuery(qS).setParameter(1, params); |
| rs = q.getResultList(); |
| } catch (ArgumentException e) { |
| // as expected |
| //System.out.println(e.getMessage()); |
| } |
| |
| qS = "SELECT p FROM Person p where TYPE(p) = Contractor AND p.name = 'Name ctr0'"; |
| q = em.createQuery(qS); |
| rs = q.getResultList(); |
| assertEquals(1, rs.size()); |
| for (int i = 0; i < rs.size(); i++) |
| assertTrue(rs.get(i) instanceof Contractor); |
| |
| qS = "select p from Person p where TYPE(p) in (?1, ?2) and p.name = ?3 order by p.name"; |
| q = em.createQuery(qS); |
| q.setParameter(1, Contractor.class); |
| q.setParameter(2, FulltimeEmployee.class); |
| q.setParameter(3, "Name ctr0"); |
| |
| rs = q.getResultList(); |
| assertEquals(1, rs.size()); |
| for (int i = 0; i < rs.size(); i++) { |
| Object obj = rs.get(i); |
| assertTrue(obj instanceof Contractor || obj instanceof FulltimeEmployee); |
| } |
| |
| qS = "select p from Person p where TYPE(p) in (?1, ?2) order by p.name"; |
| q = em.createQuery(qS); |
| q.setParameter(1, Contractor.class); |
| q.setParameter(2, FulltimeEmployee.class); |
| |
| rs = q.getResultList(); |
| assertEquals(numContractors + numFTEmployees, rs.size()); |
| for (int i = 0; i < rs.size(); i++) { |
| Object obj = rs.get(i); |
| assertTrue(obj instanceof Contractor || obj instanceof FulltimeEmployee); |
| } |
| |
| qS = "select p from Person p where TYPE(p) not in (Contractor) order by p.name"; |
| q = em.createQuery(qS); |
| rs = q.getResultList(); |
| assertEquals(numPTEmployees + numFTEmployees, rs.size()); |
| for (int i = 0; i < rs.size(); i++){ |
| Object obj = rs.get(i); |
| assertTrue((obj instanceof ParttimeEmployee) || (obj instanceof FulltimeEmployee)); |
| } |
| |
| qS = "select p from Person p where TYPE(p) not in (?1) order by p.name"; |
| q = em.createQuery(qS); |
| q.setParameter(1, Contractor.class); |
| rs = q.getResultList(); |
| assertEquals(numPTEmployees + numFTEmployees, rs.size()); |
| for (int i = 0; i < rs.size(); i++){ |
| Object obj = rs.get(i); |
| assertTrue((obj instanceof ParttimeEmployee) || (obj instanceof FulltimeEmployee)); |
| } |
| |
| qS = "select p from Person p where TYPE(p) not in (?1, ?2) order by p.name"; |
| q = em.createQuery(qS); |
| q.setParameter(1, Contractor.class); |
| q.setParameter(2, FulltimeEmployee.class); |
| rs = q.getResultList(); |
| assertEquals(numPTEmployees, rs.size()); |
| for (int i = 0; i < rs.size(); i++){ |
| Object obj = rs.get(i); |
| assertTrue((obj instanceof ParttimeEmployee) || (obj instanceof FulltimeEmployee)); |
| } |
| em.close(); |
| } |
| |
| public void testInheritanceTypeJoinedQuery() { |
| EntityManager em = emf.createEntityManager(); |
| Query q = null; |
| String qS = null; |
| Department dept = null; |
| |
| qS = "SELECT c.OID, c.dept FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| List<Object[]> lResult = q.getResultList(); |
| for (Object[] resultElement : lResult) { |
| Long oid = (Long)resultElement[0]; |
| dept = (Department)resultElement[1]; |
| } |
| |
| qS = "SELECT c.OID FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| for (Object resultElement : q.getResultList()) { |
| Long oid = (Long)resultElement; |
| } |
| |
| qS = "SELECT d FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| for (Department aResult: (List <Department>) q.getResultList()) { |
| assertEquals(dept.getOID(), aResult.getOID()); |
| } |
| |
| qS = "SELECT c FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| for (Contractor aResult: (List <Contractor>) q.getResultList()) { |
| //System.out.println(aResult.getDescription() + ", " + aResult.getOID()); |
| assertEquals(dept.getOID(), aResult.getDept().getOID()); |
| } |
| qS = "SELECT c FROM Contractor c, Department d where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| for (Contractor aResult: (List <Contractor>) q.getResultList()) { |
| assertEquals(dept.getOID(), aResult.getDept().getOID()); |
| } |
| |
| qS = "SELECT c, c.OID FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| List<Object[]> cResult = q.getResultList(); |
| Contractor contractor = null; |
| for (Object[] resultElement : cResult) { |
| contractor = (Contractor)resultElement[0]; |
| Long oid = (Long)resultElement[1]; |
| assertTrue(contractor.getOID() == oid); |
| assertEquals(dept.getOID(), contractor.getDept().getOID()); |
| } |
| |
| qS = "SELECT c.OID, c FROM Contractor c, Department d where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| List<Object[]> dResult = q.getResultList(); |
| for (Object[] resultElement : dResult) { |
| Long oid = (Long)resultElement[0]; |
| contractor = (Contractor)resultElement[1]; |
| assertTrue(contractor.getOID() == oid); |
| assertEquals(dept.getOID(), contractor.getDept().getOID()); |
| } |
| |
| qS = "SELECT c, c.OID FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| List<Object[]> eResult = q.getResultList(); |
| for (Object[] resultElement : eResult) { |
| Long oid = (Long)resultElement[1]; |
| contractor = (Contractor)resultElement[0]; |
| assertTrue(contractor.getOID() == oid); |
| assertEquals(dept.getOID(), contractor.getDept().getOID()); |
| } |
| |
| qS = "SELECT c.OID, c FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT'"; |
| q = em.createQuery(qS); |
| List<Object[]> fResult = q.getResultList(); |
| for (Object[] resultElement : fResult) { |
| Long oid = (Long)resultElement[0]; |
| Contractor c = (Contractor)resultElement[1]; |
| assertTrue(oid == c.getOID()); |
| assertEquals(dept.getOID(), c.getDept().getOID()); |
| } |
| |
| qS = "SELECT d,c FROM Department d, Contractor c where d.OID = c.dept.OID and d.description = 'IT' " + |
| " and c = ?1"; |
| q = em.createQuery(qS); |
| q.setParameter(1, contractor); |
| for (Object[] aResult: (List <Object[]>) q.getResultList()) { |
| System.out.println(((Department)aResult[0]).getOID() + ", " + ((Contractor)aResult[1]).getOID()); |
| assertTrue(contractor.equals(aResult[1])); |
| } |
| |
| qS = "SELECT c,d FROM Contractor c, Department d where d.OID = c.dept.OID and d.description = 'IT' " + |
| " and c = ?1"; |
| q = em.createQuery(qS); |
| q.setParameter(1, contractor); |
| for (Object[] aResult: (List <Object[]>) q.getResultList()) { |
| System.out.println(((Contractor)aResult[0]).getOID() + ", " + ((Department)aResult[1]).getOID()); |
| assertTrue(contractor.equals(aResult[0])); |
| } |
| |
| qS = "SELECT p FROM Person p "; |
| q = em.createQuery(qS); |
| List rs = (List<Object>) q.getResultList(); |
| assertEquals(numPTEmployees + numFTEmployees + numContractors, rs.size()); |
| for (Object aResult: rs) { |
| assertTrue(aResult instanceof Person); |
| } |
| |
| em.close(); |
| } |
| } |
| |