binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/yangtools/binding/data/codec/test/InstanceIdentifierTest.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.yangtools.binding.data.codec.test;
   9
  10 import javassist.ClassPool;
  11 import org.junit.Test;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.yangtools.test.augment.rev140709.TreeComplexUsesAugment;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.yangtools.test.augment.rev140709.TreeLeafOnlyAugment;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.yangtools.test.binding.rev140701.Top;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.yangtools.test.binding.rev140701.two.level.list.TopLevelList;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.yangtools.test.binding.rev140701.two.level.list.TopLevelListKey;
  17 import org.opendaylight.yangtools.binding.data.codec.gen.impl.StreamWriterGenerator;
  18 import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
  19 import org.opendaylight.yangtools.sal.binding.generator.util.JavassistUtils;
  20 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  21 import org.opendaylight.yangtools.yang.common.QName;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
  24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  25
  26 import static org.junit.Assert.assertEquals;
  27 import static org.junit.Assert.assertTrue;
  28
  29
  30 public class InstanceIdentifierTest extends AbstractBindingRuntimeTest {
  31
  32     private static final TopLevelListKey TOP_FOO_KEY = new TopLevelListKey("foo");
  33     private static final InstanceIdentifier<TopLevelList> BA_TOP_LEVEL_LIST = InstanceIdentifier.builder(Top.class)
  34             .child(TopLevelList.class, TOP_FOO_KEY).build();
  35     private static final InstanceIdentifier<TreeLeafOnlyAugment> BA_TREE_LEAF_ONLY = BA_TOP_LEVEL_LIST
  36             .augmentation(TreeLeafOnlyAugment.class);
  37     private static final InstanceIdentifier<TreeComplexUsesAugment> BA_TREE_COMPLEX_USES = BA_TOP_LEVEL_LIST
  38             .augmentation(TreeComplexUsesAugment.class);
  39     private static final QName SIMPLE_VALUE_QNAME = QName.create(TreeComplexUsesAugment.QNAME, "simple-value");
  40
  41     private BindingNormalizedNodeCodecRegistry registry;
  42
  43     @Override
  44     public void setup() {
  45         super.setup();
  46         JavassistUtils utils = JavassistUtils.forClassPool(ClassPool.getDefault());
  47         registry = new BindingNormalizedNodeCodecRegistry(StreamWriterGenerator.create(utils));
  48         registry.onBindingRuntimeContextUpdated(getRuntimeContext());
  49     }
  50
  51     @Test
  52     public void testComplexAugmentationSerialization() {
  53         YangInstanceIdentifier yangII = registry.toYangInstanceIdentifier(BA_TREE_COMPLEX_USES);
  54         PathArgument lastArg = yangII.getLastPathArgument();
  55         assertTrue("Last argument should be AugmentationIdentifier", lastArg instanceof AugmentationIdentifier);
  56         InstanceIdentifier<?> bindingII = registry.fromYangInstanceIdentifier(yangII);
  57         assertEquals(BA_TREE_COMPLEX_USES, bindingII);
  58     }
  59
  60     @Test
  61     public void testLeafOnlyAugmentationSerialization() {
  62         PathArgument leafOnlyLastArg = registry.toYangInstanceIdentifier(BA_TREE_LEAF_ONLY).getLastPathArgument();
  63         assertTrue("Last argument should be AugmentationIdentifier", leafOnlyLastArg instanceof AugmentationIdentifier);
  64         assertTrue(((AugmentationIdentifier) leafOnlyLastArg).getPossibleChildNames().contains(SIMPLE_VALUE_QNAME));
  65     }
  66
  67 }