| /* |
| * Copyright (c) 1998, 2021 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 v. 2.0 which is available at |
| * http://www.eclipse.org/legal/epl-2.0, |
| * or the Eclipse Distribution License v. 1.0 which is available at |
| * http://www.eclipse.org/org/documents/edl-v10.php. |
| * |
| * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause |
| */ |
| |
| // Contributors: |
| // Oracle - initial API and implementation from Oracle TopLink |
| package org.eclipse.persistence.testing.tests.returning.model; |
| |
| import java.util.Iterator; |
| |
| import org.eclipse.persistence.descriptors.ClassDescriptor; |
| import org.eclipse.persistence.internal.helper.Helper; |
| import org.eclipse.persistence.mappings.DatabaseMapping; |
| import org.eclipse.persistence.sessions.SessionEvent; |
| import org.eclipse.persistence.sessions.SessionEventAdapter; |
| import org.eclipse.persistence.sessions.UnitOfWork; |
| import org.eclipse.persistence.sessions.changesets.ChangeRecord; |
| import org.eclipse.persistence.sessions.changesets.ObjectChangeSet; |
| import org.eclipse.persistence.sessions.changesets.UnitOfWorkChangeSet; |
| import org.eclipse.persistence.testing.framework.*; |
| |
| public class ReturningUpdateTestCase extends TestCase { |
| |
| Class1 originalObject; |
| Class1 changedObject; |
| Class1 workingObject; |
| Class1 cloneWorkingObject; |
| Class1 objectBeforeChange; |
| boolean useUOW; |
| ReturnObjectControl control; |
| UnitOfWorkChangeSet uowChangeSet; |
| |
| class Listener extends SessionEventAdapter { |
| @Override |
| public void postCalculateUnitOfWorkChangeSet(SessionEvent event) { |
| uowChangeSet = (UnitOfWorkChangeSet)event.getProperty("UnitOfWorkChangeSet"); |
| } |
| } |
| Listener listener = new Listener(); |
| |
| public ReturningUpdateTestCase(Class1 originalObject, Class1 changedObject, boolean useUOW, ReturnObjectControl control) { |
| this.originalObject = originalObject; |
| this.changedObject = changedObject; |
| this.useUOW = useUOW; |
| setName(getName() + " " + originalObject); |
| setName(getName() + " changed=" + changedObject); |
| if (useUOW) { |
| setName(getName() + " useUOW"); |
| } |
| this.control = control; |
| } |
| |
| @Override |
| protected void setup() { |
| workingObject = (Class1)originalObject.clone(); |
| UnitOfWork uow = getSession().acquireUnitOfWork(); |
| uow.registerObject(workingObject); |
| uow.commit(); |
| objectBeforeChange = (Class1)workingObject.clone(); |
| } |
| |
| @Override |
| protected void test() { |
| if (useUOW) { |
| getSession().getEventManager().addListener(listener); |
| UnitOfWork uow = getSession().acquireUnitOfWork(); |
| cloneWorkingObject = (Class1)uow.registerObject(workingObject); |
| cloneWorkingObject.updateWith(changedObject); |
| uow.commit(); |
| } else { |
| workingObject.updateWith(changedObject); |
| getAbstractSession().writeObject(workingObject); |
| } |
| } |
| |
| @Override |
| protected void verify() { |
| Class1 controlObject = (Class1)control.getObjectForUpdate(getSession(), objectBeforeChange, changedObject, useUOW); |
| if (!workingObject.isValid()) { |
| throw new TestErrorException("Object is invalid"); |
| } |
| if (!controlObject.compareWithoutId(workingObject)) { |
| throw new TestErrorException("Object is wrong"); |
| } |
| |
| if(useUOW) { |
| ClassDescriptor descriptor = getSession().getDescriptor(Class1.class); |
| ObjectChangeSet changeSet = uowChangeSet.getObjectChangeSetForClone(cloneWorkingObject); |
| Iterator<ChangeRecord> it = changeSet.getChanges().iterator(); |
| while(it.hasNext()) { |
| ChangeRecord changeRecord = it.next(); |
| String attributeName = changeRecord.getAttribute(); |
| DatabaseMapping mapping = descriptor.getMappingForAttributeName(attributeName); |
| Object beforeChangeValue = mapping.getAttributeValueFromObject(objectBeforeChange); |
| Object changeRecordOldValue = changeRecord.getOldValue(); |
| if(beforeChangeValue == null) { |
| if(changeRecordOldValue != null) { |
| throw new TestErrorException(attributeName + ": before change value was null, not " + changeRecordOldValue); |
| } |
| } else { |
| if(beforeChangeValue.getClass().isArray()) { |
| if(!Helper.compareArrays((Object[]) beforeChangeValue, (Object[]) changeRecordOldValue)) { |
| throw new TestErrorException(attributeName + ": arrays are not equal"); |
| } |
| } else { |
| if(!beforeChangeValue.equals(changeRecordOldValue)) { |
| throw new TestErrorException(attributeName + ": before change value was " + beforeChangeValue + ", not " + changeRecordOldValue); |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| @Override |
| public void reset() { |
| if (useUOW) { |
| getSession().getEventManager().removeListener(listener); |
| } |
| } |
| } |