restconf/restconf-nb-rfc8040/src/test/java/org/opendaylight/restconf/nb/rfc8040/rests/utils/PatchDataTransactionUtilTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.restconf.nb.rfc8040.rests.utils;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertFalse;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.mockito.ArgumentMatchers.any;
  14 import static org.mockito.Mockito.doReturn;
  15 import static org.mockito.Mockito.mock;
  16 import static org.opendaylight.restconf.common.patch.PatchEditOperation.CREATE;
  17 import static org.opendaylight.restconf.common.patch.PatchEditOperation.DELETE;
  18 import static org.opendaylight.restconf.common.patch.PatchEditOperation.MERGE;
  19 import static org.opendaylight.restconf.common.patch.PatchEditOperation.REMOVE;
  20 import static org.opendaylight.restconf.common.patch.PatchEditOperation.REPLACE;
  21 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateFalseFluentFuture;
  22 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateFluentFuture;
  23 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateTrueFluentFuture;
  24
  25 import java.util.ArrayList;
  26 import java.util.List;
  27 import java.util.Optional;
  28 import org.junit.Before;
  29 import org.junit.Test;
  30 import org.junit.runner.RunWith;
  31 import org.mockito.Mock;
  32 import org.mockito.Mockito;
  33 import org.mockito.junit.MockitoJUnitRunner;
  34 import org.opendaylight.mdsal.common.api.CommitInfo;
  35 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  36 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
  37 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadWriteTransaction;
  38 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
  39 import org.opendaylight.netconf.dom.api.NetconfDataTreeService;
  40 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
  41 import org.opendaylight.restconf.common.errors.RestconfError;
  42 import org.opendaylight.restconf.common.patch.PatchContext;
  43 import org.opendaylight.restconf.common.patch.PatchEntity;
  44 import org.opendaylight.restconf.common.patch.PatchStatusContext;
  45 import org.opendaylight.restconf.common.patch.PatchStatusEntity;
  46 import org.opendaylight.restconf.nb.rfc8040.TestRestconfUtils;
  47 import org.opendaylight.restconf.nb.rfc8040.handlers.TransactionChainHandler;
  48 import org.opendaylight.restconf.nb.rfc8040.rests.transactions.MdsalRestconfStrategy;
  49 import org.opendaylight.restconf.nb.rfc8040.rests.transactions.NetconfRestconfStrategy;
  50 import org.opendaylight.restconf.nb.rfc8040.rests.transactions.RestconfStrategy;
  51 import org.opendaylight.yangtools.yang.common.QName;
  52 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  53 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  54 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  55 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  56 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
  57 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  58 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
  59 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  60 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  61 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  62 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  63 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  64
  65 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  66 public class PatchDataTransactionUtilTest {
  67     private static final String PATH_FOR_NEW_SCHEMA_CONTEXT = "/jukebox";
  68     @Mock
  69     private DOMTransactionChain transactionChain;
  70     @Mock
  71     private DOMDataTreeReadWriteTransaction rwTransaction;
  72     @Mock
  73     private DOMDataBroker mockDataBroker;
  74     @Mock
  75     private NetconfDataTreeService netconfService;
  76
  77     private TransactionChainHandler transactionChainHandler;
  78     private EffectiveModelContext refSchemaCtx;
  79     private YangInstanceIdentifier instanceIdContainer;
  80     private YangInstanceIdentifier instanceIdCreateAndDelete;
  81     private YangInstanceIdentifier instanceIdMerge;
  82     private ContainerNode buildBaseContainerForTests;
  83     private YangInstanceIdentifier targetNodeForCreateAndDelete;
  84     private YangInstanceIdentifier targetNodeMerge;
  85     private MapNode buildArtistList;
  86
  87     @Before
  88     public void setUp() throws Exception {
  89         doReturn(transactionChain).when(mockDataBroker).createTransactionChain(any());
  90         transactionChainHandler = new TransactionChainHandler(mockDataBroker);
  91
  92         this.refSchemaCtx = YangParserTestUtils.parseYangFiles(
  93             TestRestconfUtils.loadFiles(PATH_FOR_NEW_SCHEMA_CONTEXT));
  94         final QName baseQName = QName.create("http://example.com/ns/example-jukebox", "2015-04-04", "jukebox");
  95         final QName containerPlayerQName = QName.create(baseQName, "player");
  96         final QName leafGapQName = QName.create(baseQName, "gap");
  97         final QName containerLibraryQName = QName.create(baseQName, "library");
  98         final QName listArtistQName = QName.create(baseQName, "artist");
  99         final QName leafNameQName = QName.create(baseQName, "name");
 100         final NodeIdentifierWithPredicates nodeWithKey = NodeIdentifierWithPredicates.of(listArtistQName, leafNameQName,
 101             "name of artist");
 102
 103         /* instance identifier for accessing container node "player" */
 104         this.instanceIdContainer = YangInstanceIdentifier.builder()
 105                 .node(baseQName)
 106                 .node(containerPlayerQName)
 107                 .build();
 108
 109         /* instance identifier for accessing leaf node "gap" */
 110         this.instanceIdCreateAndDelete = instanceIdContainer.node(leafGapQName);
 111
 112         /* values that are used for creating leaf for testPatchDataCreateAndDelete test */
 113         final LeafNode<?> buildGapLeaf = Builders.leafBuilder()
 114                 .withNodeIdentifier(new NodeIdentifier(leafGapQName))
 115                 .withValue(0.2)
 116                 .build();
 117
 118         final ContainerNode buildPlayerContainer = Builders.containerBuilder()
 119                 .withNodeIdentifier(new NodeIdentifier(containerPlayerQName))
 120                 .withChild(buildGapLeaf)
 121                 .build();
 122
 123         this.buildBaseContainerForTests = Builders.containerBuilder()
 124                 .withNodeIdentifier(new NodeIdentifier(baseQName))
 125                 .withChild(buildPlayerContainer)
 126                 .build();
 127
 128         this.targetNodeForCreateAndDelete = YangInstanceIdentifier.builder(this.instanceIdCreateAndDelete)
 129                 .node(containerPlayerQName)
 130                 .node(leafGapQName)
 131                 .build();
 132
 133         /* instance identifier for accessing leaf node "name" in list "artist" */
 134         this.instanceIdMerge = YangInstanceIdentifier.builder()
 135                 .node(baseQName)
 136                 .node(containerLibraryQName)
 137                 .node(listArtistQName)
 138                 .nodeWithKey(listArtistQName, QName.create(listArtistQName, "name"), "name of artist")
 139                 .node(leafNameQName)
 140                 .build();
 141
 142         /* values that are used for creating leaf for testPatchDataReplaceMergeAndRemove test */
 143         final LeafNode<Object> contentName = Builders.leafBuilder()
 144                 .withNodeIdentifier(new NodeIdentifier(QName.create(baseQName, "name")))
 145                 .withValue("name of artist")
 146                 .build();
 147
 148         final LeafNode<Object> contentDescription = Builders.leafBuilder()
 149                 .withNodeIdentifier(new NodeIdentifier(QName.create(baseQName, "description")))
 150                 .withValue("description of artist")
 151                 .build();
 152
 153         final MapEntryNode mapEntryNode = Builders.mapEntryBuilder()
 154                 .withNodeIdentifier(nodeWithKey)
 155                 .withChild(contentName)
 156                 .withChild(contentDescription)
 157                 .build();
 158
 159         this.buildArtistList = Builders.mapBuilder()
 160                 .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(listArtistQName))
 161                 .withChild(mapEntryNode)
 162                 .build();
 163
 164         this.targetNodeMerge = YangInstanceIdentifier.builder()
 165                 .node(baseQName)
 166                 .node(containerLibraryQName)
 167                 .node(listArtistQName)
 168                 .nodeWithKey(listArtistQName, leafNameQName, "name of artist")
 169                 .build();
 170
 171         /* Mocks */
 172         doReturn(this.rwTransaction).when(this.transactionChain).newReadWriteTransaction();
 173         doReturn(CommitInfo.emptyFluentFuture()).when(this.rwTransaction).commit();
 174         doReturn(CommitInfo.emptyFluentFuture()).when(this.netconfService).commit(Mockito.any());
 175     }
 176
 177     @Test
 178     public void testPatchDataReplaceMergeAndRemove() {
 179         final PatchEntity entityReplace =
 180                 new PatchEntity("edit1", REPLACE, this.targetNodeMerge, this.buildArtistList);
 181         final PatchEntity entityMerge = new PatchEntity("edit2", MERGE, this.targetNodeMerge, this.buildArtistList);
 182         final PatchEntity entityRemove = new PatchEntity("edit3", REMOVE, this.targetNodeMerge);
 183         final List<PatchEntity> entities = new ArrayList<>();
 184
 185         entities.add(entityReplace);
 186         entities.add(entityMerge);
 187         entities.add(entityRemove);
 188
 189         final InstanceIdentifierContext<? extends SchemaNode> iidContext =
 190                 new InstanceIdentifierContext<>(this.instanceIdMerge, null, null, this.refSchemaCtx);
 191         final PatchContext patchContext = new PatchContext(iidContext, entities, "patchRMRm");
 192
 193         patch(patchContext, new MdsalRestconfStrategy(transactionChainHandler), false);
 194         patch(patchContext, new NetconfRestconfStrategy(netconfService), false);
 195     }
 196
 197     @Test
 198     public void testPatchDataCreateAndDelete() {
 199         doReturn(immediateFalseFluentFuture()).when(this.rwTransaction).exists(LogicalDatastoreType.CONFIGURATION,
 200             this.instanceIdContainer);
 201         Mockito.when(this.netconfService.getConfig(this.instanceIdContainer))
 202                 .thenReturn(immediateFluentFuture(Optional.empty()));
 203         doReturn(immediateTrueFluentFuture()).when(this.rwTransaction).exists(LogicalDatastoreType.CONFIGURATION,
 204             this.targetNodeForCreateAndDelete);
 205         Mockito.when(this.netconfService.getConfig(this.targetNodeForCreateAndDelete))
 206                 .thenReturn(immediateFluentFuture(Optional.of(mock(NormalizedNode.class))));
 207
 208         final PatchEntity entityCreate =
 209                 new PatchEntity("edit1", CREATE, this.instanceIdContainer, this.buildBaseContainerForTests);
 210         final PatchEntity entityDelete =
 211                 new PatchEntity("edit2", DELETE, this.targetNodeForCreateAndDelete);
 212         final List<PatchEntity> entities = new ArrayList<>();
 213
 214         entities.add(entityCreate);
 215         entities.add(entityDelete);
 216
 217         final InstanceIdentifierContext<? extends SchemaNode> iidContext =
 218                 new InstanceIdentifierContext<>(this.instanceIdCreateAndDelete, null, null, this.refSchemaCtx);
 219         final PatchContext patchContext = new PatchContext(iidContext, entities, "patchCD");
 220         patch(patchContext, new MdsalRestconfStrategy(transactionChainHandler), true);
 221         patch(patchContext, new NetconfRestconfStrategy(netconfService), true);
 222     }
 223
 224     @Test
 225     public void deleteNonexistentDataTest() {
 226         doReturn(immediateFalseFluentFuture()).when(this.rwTransaction).exists(LogicalDatastoreType.CONFIGURATION,
 227             this.targetNodeForCreateAndDelete);
 228         Mockito.when(this.netconfService.getConfig(this.targetNodeForCreateAndDelete))
 229                 .thenReturn(immediateFluentFuture(Optional.empty()));
 230
 231         final PatchEntity entityDelete = new PatchEntity("edit", DELETE, this.targetNodeForCreateAndDelete);
 232         final List<PatchEntity> entities = new ArrayList<>();
 233
 234         entities.add(entityDelete);
 235
 236         final InstanceIdentifierContext<? extends SchemaNode> iidContext =
 237                 new InstanceIdentifierContext<>(this.instanceIdCreateAndDelete, null, null, this.refSchemaCtx);
 238         final PatchContext patchContext = new PatchContext(iidContext, entities, "patchD");
 239         delete(patchContext, new MdsalRestconfStrategy(transactionChainHandler));
 240         delete(patchContext, new NetconfRestconfStrategy(netconfService));
 241     }
 242
 243     @Test
 244     public void testPatchMergePutContainer() {
 245         final PatchEntity entityMerge =
 246                 new PatchEntity("edit1", MERGE, this.instanceIdContainer, this.buildBaseContainerForTests);
 247         final List<PatchEntity> entities = new ArrayList<>();
 248
 249         entities.add(entityMerge);
 250
 251         final InstanceIdentifierContext<? extends SchemaNode> iidContext =
 252                 new InstanceIdentifierContext<>(this.instanceIdCreateAndDelete, null, null, this.refSchemaCtx);
 253         final PatchContext patchContext = new PatchContext(iidContext, entities, "patchM");
 254         patch(patchContext, new MdsalRestconfStrategy(transactionChainHandler), false);
 255         patch(patchContext, new NetconfRestconfStrategy(netconfService), false);
 256     }
 257
 258     private void patch(final PatchContext patchContext, final RestconfStrategy strategy,
 259                        final boolean failed) {
 260         final PatchStatusContext patchStatusContext =
 261                 PatchDataTransactionUtil.patchData(patchContext, strategy, this.refSchemaCtx);
 262         for (final PatchStatusEntity entity : patchStatusContext.getEditCollection()) {
 263             if (failed) {
 264                 assertTrue("Edit " + entity.getEditId() + " failed", entity.isOk());
 265             } else {
 266                 assertTrue(entity.isOk());
 267             }
 268         }
 269         assertTrue(patchStatusContext.isOk());
 270     }
 271
 272     private void delete(final PatchContext patchContext, final RestconfStrategy strategy) {
 273         final PatchStatusContext patchStatusContext =
 274                 PatchDataTransactionUtil.patchData(patchContext, strategy, this.refSchemaCtx);
 275
 276         assertFalse(patchStatusContext.isOk());
 277         assertEquals(RestconfError.ErrorType.PROTOCOL,
 278                 patchStatusContext.getEditCollection().get(0).getEditErrors().get(0).getErrorType());
 279         assertEquals(RestconfError.ErrorTag.DATA_MISSING,
 280                 patchStatusContext.getEditCollection().get(0).getEditErrors().get(0).getErrorTag());
 281     }
 282 }