| /* |
| * 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: |
| // tware - initial implementation |
| package org.eclipse.persistence.testing.tests.collections.map; |
| |
| import org.eclipse.persistence.expressions.Expression; |
| import org.eclipse.persistence.expressions.ExpressionBuilder; |
| import org.eclipse.persistence.internal.queries.MappedKeyMapContainerPolicy; |
| import org.eclipse.persistence.mappings.DirectMapMapping; |
| import org.eclipse.persistence.mappings.ForeignReferenceMapping; |
| import org.eclipse.persistence.queries.ReadObjectQuery; |
| import org.eclipse.persistence.sessions.UnitOfWork; |
| import org.eclipse.persistence.testing.framework.TestErrorException; |
| import org.eclipse.persistence.testing.models.collections.map.EntityDirectMapHolder; |
| import org.eclipse.persistence.testing.models.collections.map.EntityMapKey; |
| |
| public class TestUpdateEntityDirectMapMapping extends TestReadEntityDirectMapMapping { |
| |
| private boolean usePrivateOwned = false; |
| protected ForeignReferenceMapping keyMapping = null; |
| private boolean oldKeyPrivateOwnedValue = false; |
| protected EntityDirectMapHolder changedHolder = null; |
| |
| public TestUpdateEntityDirectMapMapping(){ |
| super(); |
| } |
| |
| public TestUpdateEntityDirectMapMapping(boolean usePrivateOwned){ |
| this(); |
| this.usePrivateOwned = usePrivateOwned; |
| setName("TestUpdateEntityDirectMapMapping privateOwned=" + usePrivateOwned); |
| } |
| |
| @Override |
| public void setup(){ |
| DirectMapMapping mapping = (DirectMapMapping)getSession().getProject().getDescriptor(EntityDirectMapHolder.class).getMappingForAttributeName("entityToDirectMap"); |
| keyMapping = (ForeignReferenceMapping)((MappedKeyMapContainerPolicy)mapping.getContainerPolicy()).getKeyMapping(); |
| oldKeyPrivateOwnedValue = keyMapping.isPrivateOwned(); |
| keyMapping.setIsPrivateOwned(usePrivateOwned); |
| super.setup(); |
| } |
| |
| @Override |
| public void test(){ |
| UnitOfWork uow = getSession().acquireUnitOfWork(); |
| holders = uow.readAllObjects(EntityDirectMapHolder.class, holderExp); |
| changedHolder = (EntityDirectMapHolder)holders.get(0); |
| EntityMapKey mapKey = new EntityMapKey(); |
| mapKey.setId(1); |
| changedHolder.removeEntityToDirectMapItem(mapKey); |
| mapKey = new EntityMapKey(); |
| mapKey.setId(3); |
| mapKey.setData("testData"); |
| mapKey = (EntityMapKey)uow.registerObject(mapKey); |
| changedHolder.addEntityDirectMapItem(mapKey, 3); |
| uow.commit(); |
| Object holderForComparison = uow.readObject(changedHolder); |
| if (!compareObjects(changedHolder, holderForComparison)){ |
| throw new TestErrorException("Objects do not match after write"); |
| } |
| } |
| |
| @Override |
| public void verify(){ |
| getSession().getIdentityMapAccessor().initializeIdentityMaps(); |
| holders = getSession().readAllObjects(EntityDirectMapHolder.class, holderExp); |
| EntityDirectMapHolder holder = (EntityDirectMapHolder)holders.get(0); |
| if (!compareObjects(holder, changedHolder)){ |
| throw new TestErrorException("Objects do not match reinitialize"); |
| } |
| EntityMapKey mapKey = new EntityMapKey(); |
| mapKey.setId(1); |
| if (holder.getEntityToDirectMap().containsKey(mapKey)){ |
| throw new TestErrorException("Item that was removed is still present in map."); |
| } |
| mapKey = new EntityMapKey(); |
| mapKey.setId(3); |
| Integer value = (Integer)holder.getEntityToDirectMap().get(mapKey); |
| if (value != 3){ |
| throw new TestErrorException("Item was not correctly added to map"); |
| } |
| if (keyMapping.isPrivateOwned()){ |
| ReadObjectQuery query = new ReadObjectQuery(EntityMapKey.class); |
| ExpressionBuilder keys = new ExpressionBuilder(); |
| Expression keycriteria = keys.get("id").equal(1); |
| query.setSelectionCriteria(keycriteria); |
| mapKey = (EntityMapKey)getSession().executeQuery(query); |
| if (mapKey != null){ |
| throw new TestErrorException("PrivateOwned EntityMapKey was not deleted."); |
| } |
| } |
| } |
| |
| @Override |
| public void reset(){ |
| super.reset(); |
| keyMapping.setIsPrivateOwned(oldKeyPrivateOwnedValue); |
| } |
| } |