data/yang-data-tree-ri/src/test/java/org/opendaylight/yangtools/yang/data/tree/impl/Bug2690Test.java

   1 /*
   2  * Copyright (c) 2015 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.yangtools.yang.data.tree.impl;
   9
  10 import static org.junit.jupiter.api.Assertions.assertEquals;
  11 import static org.junit.jupiter.api.Assertions.assertTrue;
  12
  13 import org.junit.jupiter.api.BeforeEach;
  14 import org.junit.jupiter.api.Test;
  15 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  16 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
  17 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  18 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  19 import org.opendaylight.yangtools.yang.data.tree.api.DataTree;
  20 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeConfiguration;
  21 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeModification;
  22 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeSnapshot;
  23 import org.opendaylight.yangtools.yang.data.tree.api.DataValidationFailedException;
  24 import org.opendaylight.yangtools.yang.data.tree.impl.di.InMemoryDataTreeFactory;
  25
  26 class Bug2690Test extends AbstractTestModelTest {
  27     private DataTree inMemoryDataTree;
  28
  29     @BeforeEach
  30     void prepare() {
  31         inMemoryDataTree = new InMemoryDataTreeFactory().create(DataTreeConfiguration.DEFAULT_OPERATIONAL,
  32             SCHEMA_CONTEXT);
  33     }
  34
  35     @Test
  36     void testWriteMerge1() throws DataValidationFailedException {
  37         final var fooEntryNode = ImmutableNodes.mapEntry(TestModel.OUTER_LIST_QNAME, TestModel.ID_QNAME, 1);
  38         final var barEntryNode = ImmutableNodes.mapEntry(TestModel.OUTER_LIST_QNAME, TestModel.ID_QNAME, 2);
  39         final var mapNode1 = ImmutableNodes.mapNodeBuilder()
  40                 .withNodeIdentifier(new NodeIdentifier(TestModel.OUTER_LIST_QNAME))
  41                 .withChild(fooEntryNode).build();
  42         final var mapNode2 = ImmutableNodes.mapNodeBuilder()
  43                 .withNodeIdentifier(new NodeIdentifier(TestModel.OUTER_LIST_QNAME))
  44                 .withChild(barEntryNode).build();
  45
  46         final var cont1 = Builders.containerBuilder()
  47                 .withNodeIdentifier(new NodeIdentifier(TestModel.TEST_QNAME))
  48                 .withChild(mapNode1).build();
  49
  50         final var cont2 = Builders.containerBuilder()
  51                 .withNodeIdentifier(new NodeIdentifier(TestModel.TEST_QNAME))
  52                 .withChild(mapNode2).build();
  53
  54         final var modificationTree = inMemoryDataTree.takeSnapshot().newModification();
  55         modificationTree.write(TestModel.TEST_PATH, cont1);
  56         modificationTree.merge(TestModel.TEST_PATH, cont2);
  57         commit(modificationTree);
  58
  59         final var snapshotAfterTx = inMemoryDataTree.takeSnapshot();
  60         final var modificationAfterTx = snapshotAfterTx.newModification();
  61         final var readNode = modificationAfterTx.readNode(TestModel.OUTER_LIST_PATH);
  62         assertTrue(readNode.isPresent());
  63         assertEquals(2, ((NormalizedNodeContainer<?>)readNode.orElseThrow()).size());
  64     }
  65
  66     @Test
  67     void testDeleteStructuralAndWriteChild() throws DataValidationFailedException {
  68         final var modificationTree = setupTestDeleteStructuralAndWriteChild();
  69         verifyTestDeleteStructuralAndWriteChild(modificationTree);
  70     }
  71
  72     @Test
  73     void testDeleteStructuralAndWriteChildWithCommit() throws DataValidationFailedException {
  74         final var modificationTree = setupTestDeleteStructuralAndWriteChild();
  75         commit(modificationTree);
  76         verifyTestDeleteStructuralAndWriteChild(inMemoryDataTree.takeSnapshot());
  77     }
  78
  79     private DataTreeModification setupTestDeleteStructuralAndWriteChild() {
  80         final var modificationTree = inMemoryDataTree.takeSnapshot().newModification();
  81         modificationTree.delete(TestModel.NON_PRESENCE_PATH);
  82         modificationTree.write(TestModel.NAME_PATH, ImmutableNodes.leafNode(TestModel.NAME_QNAME, "abc"));
  83         return modificationTree;
  84     }
  85
  86     private static void verifyTestDeleteStructuralAndWriteChild(final DataTreeSnapshot snapshot) {
  87         final var readNode = snapshot.readNode(TestModel.NAME_PATH);
  88         assertTrue(readNode.isPresent());
  89     }
  90
  91     private void commit(final DataTreeModification modificationTree) throws DataValidationFailedException {
  92         modificationTree.ready();
  93
  94         inMemoryDataTree.validate(modificationTree);
  95         final var prepare = inMemoryDataTree.prepare(modificationTree);
  96         inMemoryDataTree.commit(prepare);
  97     }
  98 }