/*******************************************************************************
* Copyright (c) 1998, 2015 Oracle and/or its affiliates. All rights reserved.
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0
* which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* Contributors:
* Oracle - initial API and implementation from Oracle TopLink
******************************************************************************/
package org.eclipse.persistence.testing.tests.jpa.advanced;
import java.util.*;
import org.eclipse.persistence.jpa.JpaEntityManager;
import org.eclipse.persistence.internal.jpa.EntityManagerImpl;
import org.eclipse.persistence.testing.framework.*;
import org.eclipse.persistence.testing.models.jpa.advanced.*;
import org.eclipse.persistence.testing.tests.jpa.EntityContainerTestBase;
public class EMRemoveAndCommitTests extends EntityContainerTestBase {
public EMRemoveAndCommitTests() {
setDescription("Test remove and commit in EntityManager");
}
//reset gets called twice on error
protected boolean reset = false;
public Integer[] empIDs = new Integer[3];
public Integer[] projIDs = new Integer[2];
public Integer[] addrIDs = new Integer[1];
public ArrayList<PhoneNumberPK> phoneIDs = new ArrayList<PhoneNumberPK>();
public void setup (){
super.setup();
this.reset = true;
Employee empClone1 = ModelExamples.employeeExample1();
empClone1.setAddress(ModelExamples.addressExample1());
empClone1.addPhoneNumber(ModelExamples.phoneExample1());
empClone1.addPhoneNumber(ModelExamples.phoneExample9());
Employee empClone2 = ModelExamples.employeeExample2();
empClone2.setAddress(ModelExamples.addressExample2());
empClone2.addPhoneNumber(ModelExamples.phoneExample2());
empClone2.addPhoneNumber(ModelExamples.phoneExample8());
Employee empClone3 = ModelExamples.employeeExample3();
empClone3.setAddress(ModelExamples.addressExample3());
empClone3.addPhoneNumber(ModelExamples.phoneExample3());
empClone3.addPhoneNumber(ModelExamples.phoneExample7());
empClone1.addManagedEmployee(empClone2);
empClone1.addManagedEmployee(empClone3);
Project projClone1 = ModelExamples.projectExample1();
Project projClone2 = ModelExamples.projectExample2();
projClone1.setTeamLeader(empClone1);
projClone1.addTeamMember(empClone1);
projClone1.addTeamMember(empClone2);
projClone1.addTeamMember(empClone3);
empClone1.addProject(projClone1);
empClone2.addProject(projClone1);
empClone3.addProject(projClone1);
Address addrClone = ModelExamples.addressExample4();
try {
beginTransaction();
getEntityManager().persist(empClone1);
getEntityManager().persist(empClone2);
getEntityManager().persist(empClone3);
getEntityManager().persist(projClone1);
getEntityManager().persist(projClone2);
getEntityManager().persist(addrClone);
commitTransaction();
} catch (Exception ex) {
rollbackTransaction();
throw new TestException("Unable to setup Test" + ex);
}
((EntityManagerImpl)getEntityManager()).getActiveSession().getIdentityMapAccessor().initializeAllIdentityMaps();
empIDs[0] = empClone1.getId();
empIDs[1] = empClone2.getId();
empIDs[2] = empClone3.getId();
projIDs[0] = projClone1.getId();
projIDs[1] = projClone2.getId();
addrIDs[0] = addrClone.getID();
getSession().getIdentityMapAccessor().initializeAllIdentityMaps();
}
public void reset (){
if (reset){
phoneIDs.clear();
reset = false;
}
super.reset();
}
public void test(){
try {
beginTransaction();
Employee employee = getEntityManager().find(Employee.class, empIDs[1]);
ArrayList projects = new ArrayList(employee.getProjects());
for (Iterator projs = projects.iterator(); projs.hasNext();){
Project project = (Project) projs.next();
if (project.getTeamLeader() == employee){
project.setTeamLeader(null);
}
employee.getProjects().remove(project);
}
ArrayList managedEmps = new ArrayList(employee.getManagedEmployees());
for (Iterator reports = managedEmps.iterator(); reports.hasNext();){
Employee report = (Employee) reports.next();
if (report.getManager() == employee){
report.setManager(null);
}
employee.getManagedEmployees().remove(report);
}
for (Iterator phones = employee.getPhoneNumbers().iterator(); phones.hasNext();){
this.phoneIDs.add(((PhoneNumber)phones.next()).buildPK());
}
getEntityManager().remove(employee);
Address address = getEntityManager().find(Address.class, addrIDs[0]);
getEntityManager().remove(address);
commitTransaction();
} catch (Exception ex) {
rollbackTransaction();
throw new TestErrorException("Exception thrown during assignment of employee to a project" + ex);
}
}
public void verify(){
Employee employee = getEntityManager().find(Employee.class, empIDs[1]);
if ( employee != null){
throw new TestErrorException("Employee ID :" + empIDs[1] + " Was not Deleted");
}
for (Iterator<PhoneNumberPK> ids = this.phoneIDs.iterator(); ids.hasNext();){
PhoneNumber phone = getEntityManager().find(PhoneNumber.class, ids.next());
if (phone != null){
throw new TestErrorException("Employee ID :" + empIDs[1] + " Related Phones were not deleted");
}
}
//lets initialize the identity map to make sure they were persisted
((JpaEntityManager)getEntityManager()).getActiveSession().getIdentityMapAccessor().initializeAllIdentityMaps();
employee = getEntityManager().find(Employee.class, empIDs[1]);
if ( employee != null){
throw new TestErrorException("Employee ID :" + empIDs[1] + " Was not Deleted on Database");
}
for (Iterator<PhoneNumberPK> ids = this.phoneIDs.iterator(); ids.hasNext();){
PhoneNumber phone = getEntityManager().find(PhoneNumber.class, ids.next());
if (phone != null){
throw new TestErrorException("Employee ID :" + empIDs[1] + " Related Phones were not deleted on Database");
}
}
Address address = getEntityManager().find(Address.class, addrIDs[0]);
if ( address != null){
throw new TestErrorException("Address ID :" + addrIDs[0] + " Was not Deleted on Database");
}
}
}