| /* |
| * 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.maps.spec_10_1_26_ex3; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| |
| import javax.persistence.EntityManager; |
| import javax.persistence.EntityManagerFactory; |
| import javax.persistence.EntityTransaction; |
| import javax.persistence.Query; |
| |
| import org.apache.openjpa.kernel.QueryImpl; |
| import org.apache.openjpa.persistence.test.AllowFailure; |
| import org.apache.openjpa.persistence.test.SQLListenerTestCase; |
| |
| import junit.framework.Assert; |
| |
| public class TestSpec10_1_26_Ex3 extends SQLListenerTestCase { |
| |
| public int numDepartments = 2; |
| public int numEmployeesPerDept = 2; |
| public List<String> namedQueries = new ArrayList<>(); |
| |
| public int deptId = 1; |
| public int empId = 1; |
| public List rsAllDepartments = null; |
| |
| @Override |
| public void setUp() { |
| super.setUp(DROP_TABLES, |
| Department.class, |
| Employee.class, |
| EmployeeName.class); |
| createObj(emf); |
| rsAllDepartments = getAll(Department.class); |
| } |
| |
| @AllowFailure |
| public void testQueryInMemoryQualifiedId() throws Exception { |
| queryQualifiedId(true); |
| } |
| |
| public void testQueryQualifiedId() throws Exception { |
| queryQualifiedId(false); |
| } |
| |
| public void setCandidate(Query q, Class clz) |
| throws Exception { |
| org.apache.openjpa.persistence.QueryImpl q1 = |
| (org.apache.openjpa.persistence.QueryImpl) q; |
| org.apache.openjpa.kernel.Query q2 = q1.getDelegate(); |
| org.apache.openjpa.kernel.QueryImpl qi = (QueryImpl) q2; |
| if (clz == Department.class) |
| qi.setCandidateCollection(rsAllDepartments); |
| } |
| |
| public void queryQualifiedId(boolean inMemory) throws Exception { |
| EntityManager em = emf.createEntityManager(); |
| String query = "select KEY(e) from Department d, " + |
| " in (d.emps) e order by d.deptId, e.empId"; |
| Query q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| List rs = q.getResultList(); |
| EmployeeName d = (EmployeeName) rs.get(0); |
| |
| em.clear(); |
| query = "select ENTRY(e) from Department d, " + |
| " in (d.emps) e order by d.deptId, e.empId"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| Map.Entry me = (Map.Entry) rs.get(0); |
| |
| assertEquals(d, me.getKey()); |
| |
| // test ORDER BY qualified path |
| query = "select KEY(e), KEY(e).fName from Department d, " + |
| " in (d.emps) e order by KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| d = (EmployeeName) ((Object[]) rs.get(0))[0]; |
| String fname = (String) ((Object[]) rs.get(0))[1]; |
| |
| assertEquals(d.getFName(), fname); |
| |
| // test GROUP BY qualified path |
| sql.clear(); |
| query = "select COUNT(KEY(e).fName) from Department d " + |
| " left join d.emps e GROUP BY KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertTrue(sql.get(0).toUpperCase().indexOf(" GROUP BY ") != -1); |
| |
| query = "select KEY(e) from Department d, " + |
| " in (d.emps) e where VALUE(e).department.deptId = 1" + |
| " ORDER BY KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertEquals(((EmployeeName) rs.get(0)).getFName(), "f1"); |
| |
| query = "select KEY(e) from Department d, " + |
| " in (d.emps) e where VALUE(e).name.fName = 'f1'" + |
| " ORDER BY KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertEquals(((EmployeeName) rs.get(0)).getFName(), "f1"); |
| |
| query = "select KEY(e) from Department d, " + |
| " in (d.emps) e where SUBSTRING(VALUE(e).name.fName, 1) = 'f1'" + |
| " ORDER BY KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertEquals(((EmployeeName) rs.get(0)).getFName(), "f1"); |
| |
| query = "select KEY(e) from Department d, " + |
| " in (d.emps) e where LOCATE(VALUE(e).name.fName, 'f1') <> 0" + |
| " ORDER BY KEY(e).fName"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Department.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertEquals(((EmployeeName) rs.get(0)).getFName(), "f1"); |
| |
| em.close(); |
| } |
| |
| public void testQueryObject() { |
| queryObj(emf); |
| } |
| |
| public void createObj(EntityManagerFactory emf) { |
| EntityManager em = emf.createEntityManager(); |
| EntityTransaction tran = em.getTransaction(); |
| for (int i = 0; i < numDepartments; i++) |
| createDepartment(em, deptId++); |
| tran.begin(); |
| em.flush(); |
| tran.commit(); |
| em.close(); |
| } |
| |
| public void createDepartment(EntityManager em, int id) { |
| Department d = new Department(); |
| d.setDeptId(id); |
| for (int i = 0; i < numEmployeesPerDept; i++) { |
| Employee e = createEmployee(em, empId++); |
| d.addEmployee(e); |
| e.setDepartment(d); |
| em.persist(e); |
| } |
| em.persist(d); |
| } |
| |
| public Employee createEmployee(EntityManager em, int id) { |
| Employee e = new Employee(); |
| EmployeeName name = new EmployeeName("f" + id, "l" + id); |
| e.setEmpId(id); |
| e.setName(name); |
| return e; |
| } |
| |
| public void findObj(EntityManagerFactory emf) { |
| EntityManager em = emf.createEntityManager(); |
| Department d = em.find(Department.class, 1); |
| assertDepartment(d); |
| |
| //Employee e = em.find(Employee.class, 1); |
| //assertEmployee(e); |
| |
| updateObj(em, d); |
| deleteObj(em, d); |
| em.close(); |
| } |
| |
| public void updateObj(EntityManager em, Department d) { |
| EntityTransaction tran = em.getTransaction(); |
| // add an element |
| tran.begin(); |
| Employee e = createEmployee( |
| em, numDepartments * numEmployeesPerDept + 1); |
| d.addEmployee(e); |
| e.setDepartment(d); |
| em.persist(d); |
| em.persist(e); |
| em.flush(); |
| tran.commit(); |
| |
| // remove an element |
| tran.begin(); |
| d.removeEmployee(e.getName()); |
| e.setDepartment(null); |
| em.persist(d); |
| em.flush(); |
| tran.commit(); |
| } |
| |
| public void deleteObj(EntityManager em, Department d) { |
| EntityTransaction tran = em.getTransaction(); |
| tran.begin(); |
| em.remove(d); |
| tran.commit(); |
| } |
| |
| public void assertDepartment(Department d) { |
| int id = d.getDeptId(); |
| Map<EmployeeName, Employee> es = d.getEmployees(); |
| Assert.assertEquals(2,es.size()); |
| Set keys = es.keySet(); |
| for (Object obj : keys) { |
| EmployeeName empName = (EmployeeName) obj; |
| Employee e = (Employee)es.get(empName); |
| Assert.assertEquals(empName, e.getName()); |
| } |
| } |
| |
| public void queryObj(EntityManagerFactory emf) { |
| queryDepartment(emf); |
| queryEmployee(emf); |
| } |
| |
| public void queryDepartment(EntityManagerFactory emf) { |
| EntityManager em = emf.createEntityManager(); |
| EntityTransaction tran = em.getTransaction(); |
| tran.begin(); |
| Query q = em.createQuery("select d from Department d"); |
| List<Department> ds = q.getResultList(); |
| for (Department d : ds){ |
| assertDepartment(d); |
| } |
| tran.commit(); |
| em.close(); |
| } |
| |
| public void queryEmployee(EntityManagerFactory emf) { |
| EntityManager em = emf.createEntityManager(); |
| EntityTransaction tran = em.getTransaction(); |
| tran.begin(); |
| Query q = em.createQuery("select e from Employee e"); |
| List<Employee> es = q.getResultList(); |
| for (Employee e : es){ |
| assertEmployee(e); |
| } |
| tran.commit(); |
| em.close(); |
| } |
| |
| public void assertEmployee(Employee e) { |
| int id = e.getEmpId(); |
| Department d = e.getDepartment(); |
| assertDepartment(d); |
| } |
| } |