binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/mdsal/binding/dom/codec/impl/AugmentationSubstitutionTest.java

   1 /*
   2  * Copyright (c) 2014 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.mdsal.binding.dom.codec.impl;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.util.Map.Entry;
  13 import org.junit.Test;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.augment.rev140709.RpcComplexUsesAugment;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.augment.rev140709.RpcComplexUsesAugmentBuilder;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.augment.rev140709.TreeComplexUsesAugmentBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.augment.rev140709.complex.from.grouping.ContainerWithUsesBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.binding.rev140701.Top;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.binding.rev140701.two.level.list.TopLevelList;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.binding.rev140701.two.level.list.TopLevelListBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.test.binding.rev140701.two.level.list.TopLevelListKey;
  22 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  24 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  25
  26 public class AugmentationSubstitutionTest extends AbstractBindingCodecTest {
  27     private static final TopLevelListKey TOP_FOO_KEY = new TopLevelListKey("foo");
  28     private static final InstanceIdentifier<TopLevelList> BA_TOP_LEVEL_LIST = InstanceIdentifier.builder(Top.class)
  29             .child(TopLevelList.class, TOP_FOO_KEY).build();
  30
  31     @Test
  32     public void augmentationInGroupingSubstituted() {
  33         final TopLevelList baRpc = new TopLevelListBuilder()
  34             .withKey(TOP_FOO_KEY)
  35             .addAugmentation(new RpcComplexUsesAugmentBuilder(createComplexData()).build())
  36             .build();
  37         final TopLevelList baTree = new TopLevelListBuilder()
  38             .withKey(TOP_FOO_KEY)
  39             .addAugmentation(new TreeComplexUsesAugmentBuilder(createComplexData()).build())
  40             .build();
  41         final NormalizedNode domTreeEntry = codecContext.toNormalizedNode(BA_TOP_LEVEL_LIST, baTree).getValue();
  42         final NormalizedNode domRpcEntry = codecContext.toNormalizedNode(BA_TOP_LEVEL_LIST, baRpc).getValue();
  43         assertEquals(domTreeEntry, domRpcEntry);
  44     }
  45
  46     @Test
  47     public void copyBuilderWithAugmenationsTest() {
  48         final TopLevelList manuallyConstructed = new TopLevelListBuilder()
  49             .withKey(TOP_FOO_KEY)
  50             .addAugmentation(new TreeComplexUsesAugmentBuilder(createComplexData()).build())
  51             .build();
  52
  53         final Entry<YangInstanceIdentifier, NormalizedNode> entry = codecContext.toNormalizedNode(
  54             BA_TOP_LEVEL_LIST, manuallyConstructed);
  55         final TopLevelList deserialized = (TopLevelList) codecContext.fromNormalizedNode(entry.getKey(),
  56             entry.getValue()).getValue();
  57         assertEquals(manuallyConstructed, deserialized);
  58         final TopLevelList copiedFromDeserialized = new TopLevelListBuilder(deserialized).build();
  59         assertEquals(manuallyConstructed, copiedFromDeserialized);
  60     }
  61
  62     private static RpcComplexUsesAugment createComplexData() {
  63         return new RpcComplexUsesAugmentBuilder()
  64                 .setContainerWithUses(new ContainerWithUsesBuilder().setLeafFromGrouping("foo").build())
  65                 .build();
  66     }
  67 }