| /* |
| * 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.m2mmapex10; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Date; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| import javax.persistence.EntityManager; |
| import javax.persistence.EntityTransaction; |
| import javax.persistence.Query; |
| |
| import org.apache.openjpa.kernel.QueryImpl; |
| import org.apache.openjpa.persistence.ArgumentException; |
| import org.apache.openjpa.persistence.test.AllowFailure; |
| import org.apache.openjpa.persistence.test.SQLListenerTestCase; |
| |
| import junit.framework.Assert; |
| |
| public class TestMany2ManyMapEx10 extends SQLListenerTestCase { |
| |
| public int numEmployees = 2; |
| public int numPhoneNumbers = numEmployees + 1; |
| public int numEmployeesPerPhoneNumber = 2; |
| public int numPhoneNumbersPerEmployee = 2; |
| public List<String> namedQueries = new ArrayList<>(); |
| public List<EmployeePK> empPKs = new ArrayList<>(); |
| public List<PhonePK> phonePKs = new ArrayList<>(); |
| |
| public Map<String, Employee> empMap = |
| new HashMap<>(); |
| public Map<PhonePK, PhoneNumber> phoneMap = |
| new HashMap<>(); |
| |
| public int empId = 1; |
| public int phoneId = 1; |
| public int divId = 1; |
| public List rsAllPhones = null; |
| public List rsAllEmps = null; |
| |
| @Override |
| public void setUp() { |
| super.setUp(DROP_TABLES, |
| "openjpa.jdbc.DBDictionary", "(defaultFractionLength=6)", |
| EmployeePK.class, PhonePK.class, |
| Employee.class, PhoneNumber.class); |
| createObj(); |
| rsAllPhones = getAll(PhoneNumber.class); |
| rsAllEmps = getAll(Employee.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 == PhoneNumber.class) |
| qi.setCandidateCollection(rsAllPhones); |
| else if (clz == Employee.class) |
| qi.setCandidateCollection(rsAllEmps); |
| } |
| |
| public void queryQualifiedId(boolean inMemory) throws Exception { |
| EntityManager em = emf.createEntityManager(); |
| EmployeePK ekey = empPKs.get(0); |
| String query = "select KEY(e) from PhoneNumber p, " + |
| " in (p.emps) e where e.empPK.name = ?1 and e.empPK.bDay = ?2"; |
| Query q = em.createQuery(query).setParameter(1, ekey.getName()). |
| setParameter(2, ekey.getBDay()); |
| if (inMemory) |
| setCandidate(q, PhoneNumber.class); |
| List rs = q.getResultList(); |
| EmployeePK d = (EmployeePK) rs.get(0); |
| |
| query = "select KEY(e) from PhoneNumber p, " + |
| " in (p.emps) e where e.empPK = ?1"; |
| try { |
| rs = em.createQuery(query).setParameter(1, ekey).getResultList(); |
| } catch (Exception e) { |
| // as excepted: conditional expressional expression over embeddable |
| assertException(e, ArgumentException.class); |
| } |
| |
| query = "select KEY(p) from Employee e, " + |
| " in (e.phones) p"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Employee.class); |
| rs = q.getResultList(); |
| PhonePK k = (PhonePK) rs.get(0); |
| |
| em.clear(); |
| query = "select ENTRY(e) from PhoneNumber p, " + |
| " in (p.emps) e where e.empPK.name = ?1 and e.empPK.bDay = ?2"; |
| q = em.createQuery(query).setParameter(1, ekey.getName()). |
| setParameter(2, ekey.getBDay()); |
| if (inMemory) |
| setCandidate(q, PhoneNumber.class); |
| rs = q.getResultList(); |
| Map.Entry me = (Map.Entry) rs.get(0); |
| |
| assertEquals(d, me.getKey()); |
| |
| // test navigation thru KEY |
| em.clear(); |
| query = "select KEY(e), KEY(e).name from PhoneNumber p, " + |
| " in (p.emps) e"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, PhoneNumber.class); |
| rs = q.getResultList(); |
| EmployeePK d0 = (EmployeePK) ((Object[]) rs.get(0))[0]; |
| String name = (String)((Object[]) rs.get(0))[1]; |
| assertEquals(d0.getName(), name); |
| |
| em.clear(); |
| query = "select KEY(p), KEY(p).phoneNum from Employee e, " + |
| " in (e.phones) p"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Employee.class); |
| rs = q.getResultList(); |
| k = (PhonePK) ((Object[]) rs.get(0))[0]; |
| String phoneNum = (String) ((Object[]) rs.get(0))[1]; |
| assertEquals(k.getPhoneNum(), phoneNum); |
| |
| // test ORDER BY qualified path |
| query = "select KEY(p), KEY(p).phoneNum as pno from Employee e " + |
| " left join e.phones p ORDER BY pno "; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Employee.class); |
| rs = q.getResultList(); |
| |
| // test GROUP BY qualified path |
| sql.clear(); |
| query = "select COUNT(KEY(p).phoneNum) from Employee e " + |
| " left join e.phones p GROUP BY KEY(p).phoneNum"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Employee.class); |
| rs = q.getResultList(); |
| if (!inMemory) |
| assertTrue(sql.get(0).toUpperCase().indexOf(" GROUP BY ") != -1); |
| |
| query = "select LENGTH(KEY(e).name) from PhoneNumber p, " + |
| " in (p.emps) e where KEY(e).bDay = CURRENT_TIMESTAMP"; |
| q = em.createQuery(query); |
| if (inMemory) |
| setCandidate(q, Employee.class); |
| rs = q.getResultList(); |
| assertEquals(rs.size(), 0); |
| |
| em.close(); |
| } |
| |
| public void testQueryObject() throws Exception { |
| queryObj(); |
| findObj(); |
| } |
| |
| public void createObj() { |
| EntityManager em = emf.createEntityManager(); |
| EntityTransaction tran = em.getTransaction(); |
| for (int i = 0; i < numEmployees; i++) { |
| Employee e = createEmployee(em, empId++); |
| empMap.put(e.getEmpPK().getName(), e); |
| } |
| tran.begin(); |
| em.flush(); |
| tran.commit(); |
| em.close(); |
| } |
| |
| public Employee createEmployee(EntityManager em, int id) { |
| Employee e = new Employee(); |
| Date bDay = new Date(System.currentTimeMillis() - 1000000); |
| EmployeePK empPK = new EmployeePK("e" + id, bDay); |
| empPKs.add(empPK); |
| e.setEmpPK(empPK); |
| e.setSalary(1000); |
| for (int i = 0; i < numPhoneNumbersPerEmployee; i++) { |
| PhoneNumber phoneNumber = new PhoneNumber(); |
| PhonePK phonePK = new PhonePK("areaCode" + phoneId, "phoneNum" + |
| phoneId); |
| phoneNumber.setRoom(phoneId); |
| phoneId++; |
| phonePKs.add(phonePK); |
| phoneNumber.setPhonePK(phonePK); |
| phoneNumber.addEmployees(empPK, e); |
| e.addPhoneNumber(phonePK, phoneNumber); |
| em.persist(phoneNumber); |
| phoneMap.put(phoneNumber.getPhonePK(), phoneNumber); |
| } |
| em.persist(e); |
| return e; |
| } |
| |
| public void findObj() throws Exception { |
| EntityManager em = emf.createEntityManager(); |
| Employee e = em.find(Employee.class, empPKs.get(1)); |
| assertEmployee(e); |
| |
| PhoneNumber p = em.find(PhoneNumber.class, phonePKs.get(1)); |
| assertPhoneNumber(p); |
| em.close(); |
| } |
| |
| public void queryObj() throws Exception { |
| queryEmployee(); |
| queryPhoneNumber(); |
| } |
| |
| public void queryPhoneNumber() throws Exception { |
| EntityManager em = emf.createEntityManager(); |
| EntityTransaction tran = em.getTransaction(); |
| tran.begin(); |
| Query q = em.createQuery("select p from PhoneNumber p"); |
| List<PhoneNumber> ps = q.getResultList(); |
| for (PhoneNumber p : ps) { |
| assertPhoneNumber(p); |
| } |
| tran.commit(); |
| em.close(); |
| } |
| |
| public void queryEmployee() throws Exception { |
| 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) throws Exception { |
| EmployeePK empPK = e.getEmpPK(); |
| Employee e0 = empMap.get(empPK.getName()); |
| Map<PhonePK, PhoneNumber> phones = e.getPhoneNumbers(); |
| Map<PhonePK, PhoneNumber> phones0 = e0.getPhoneNumbers(); |
| Assert.assertEquals(phones0.size(), phones.size()); |
| checkPhoneMap(phones0, phones); |
| } |
| |
| public void assertPhoneNumber(PhoneNumber p) throws Exception { |
| PhonePK phonePK = p.getPhonePK(); |
| PhoneNumber p0 = phoneMap.get(phonePK); |
| |
| Map<EmployeePK, Employee> es = p.getEmployees(); |
| Map<EmployeePK, Employee> es0 = p0.getEmployees(); |
| Assert.assertEquals(es0.size(), es.size()); |
| checkEmpMap(es0, es); |
| } |
| |
| public void checkPhoneMap(Map<PhonePK, PhoneNumber> es0, |
| Map<PhonePK, PhoneNumber> es) throws Exception { |
| Collection<Map.Entry<PhonePK, PhoneNumber>> entrySets0 = es0.entrySet(); |
| for (Map.Entry<PhonePK, PhoneNumber> entry0 : entrySets0) { |
| PhonePK d0 = entry0.getKey(); |
| PhoneNumber p0 = entry0.getValue(); |
| PhoneNumber p = es.get(d0); |
| if (!p0.equals(p)) |
| throw new Exception("Assertion failure"); |
| } |
| } |
| |
| public void checkEmpMap(Map<EmployeePK, Employee> es0, |
| Map<EmployeePK, Employee> es) |
| throws Exception { |
| Collection<Map.Entry<EmployeePK, Employee>> entrySets0 = es0.entrySet(); |
| for (Map.Entry<EmployeePK, Employee> entry0 : entrySets0) { |
| EmployeePK key0 = entry0.getKey(); |
| Employee e0 = entry0.getValue(); |
| //Employee e = es.get(key0); |
| Employee e = Employee.findEmpl(es, key0); |
| if (!e0.equals(e)) |
| throw new Exception("Assertion failure"); |
| } |
| } |
| } |