X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fsal%2Fyang-prototype%2Fcode-generator%2Fyang-model-parser-impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fyang%2Fparser%2Fimpl%2FYangParserTest.java;h=f6240c9a3f9f2d203711a6e4d3f25053367614d8;hb=6f9523a8243defc7e0c9534274e2c41a4b89dc9e;hp=b2a93ea07ec8f1749d6523788d8d51afe00c25b4;hpb=39ff6ed2000232d607877311770c74cef36cf96f;p=controller.git

diff --git a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/YangParserTest.java b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/YangParserTest.java
index b2a93ea07e..f6240c9a3f 100644
--- a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/YangParserTest.java
+++ b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/YangParserTest.java
@@ -212,9 +212,15 @@ public class YangParserTest {
                 .getDataChildByName("ifEntry");
         ContainerSchemaNode augmentedContainer = (ContainerSchemaNode) ifEntry
                 .getDataChildByName("augment-holder");
-        // Set<AugmentationSchema> augmentedContainerAugments =
-        // augmentedContainer
-        // .getAvailableAugmentations();
+
+        // augmentation defined in testfile1 and augmentation returned from
+        // augmented container have to be same
+        Set<AugmentationSchema> augmentedContainerAugments = augmentedContainer
+                .getAvailableAugmentations();
+        AugmentationSchema augmentDefinition = augmentedContainerAugments
+                .iterator().next();
+        assertEquals(augment1, augmentDefinition);
+
         LeafSchemaNode augmentedLeaf = (LeafSchemaNode) augmentedContainer
                 .getDataChildByName("ds0ChannelNumber");
         assertTrue(augmentedLeaf.isAugmenting());
@@ -230,18 +236,22 @@ public class YangParserTest {
         Set<AugmentationSchema> module3Augmentations = module3
                 .getAugmentations();
         assertEquals(2, module3Augmentations.size());
-        // AugmentationSchema augment3 = module3Augmentations.iterator().next();
-        // ContainerSchemaNode augmentedContainerDefinition =
-        // (ContainerSchemaNode) augment3
-        // .getDataChildByName("augment-holder");
-        // assertTrue(augmentedContainerDefinition.isAugmenting());
-        //
-        // // check
-        // assertEquals(augmentedContainer, augmentedContainerDefinition);
-        // assertEquals(augmentedContainerAugments.iterator().next(), augment1);
-        //
-        // assertEquals(augmentedLeaf, augmentedLeafDefinition);
-        // assertEquals(ifEntryAugments.iterator().next(), augment3);
+        AugmentationSchema augment3 = null;
+        for (AugmentationSchema as : module3Augmentations) {
+            if ("if:ifType='ds0'".equals(as.getWhenCondition().toString())) {
+                augment3 = as;
+            }
+        }
+        ContainerSchemaNode augmentedContainerDefinition = (ContainerSchemaNode) augment3
+                .getDataChildByName("augment-holder");
+        assertTrue(augmentedContainerDefinition.isAugmenting());
+
+        // check
+        assertEquals(augmentedContainer, augmentedContainerDefinition);
+        assertEquals(augmentedContainerAugments.iterator().next(), augment1);
+
+        assertEquals(augmentedLeaf, augmentedLeafDefinition);
+        assertEquals(ifEntryAugments.iterator().next(), augment3);
     }
 
     @Test
@@ -256,17 +266,21 @@ public class YangParserTest {
                 .getAvailableAugmentations();
         assertEquals(2, augmentations.size());
 
-        // AugmentationSchema augment = augmentations.iterator().next();
-
-        // ContainerSchemaNode augmentHolder = (ContainerSchemaNode) augment
-        // .getDataChildByName("augment-holder");
-        // assertNotNull(augmentHolder);
-        // assertTrue(augmentHolder.isAugmenting());
-        // QName augmentHolderQName = augmentHolder.getQName();
-        // assertEquals("augment-holder", augmentHolderQName.getLocalName());
-        // assertEquals("t3", augmentHolderQName.getPrefix());
-        // assertEquals("Description for augment holder",
-        // augmentHolder.getDescription());
+        AugmentationSchema augment = null;
+        for (AugmentationSchema as : augmentations) {
+            if ("if:ifType='ds0'".equals(as.getWhenCondition().toString())) {
+                augment = as;
+            }
+        }
+        ContainerSchemaNode augmentHolder = (ContainerSchemaNode) augment
+                .getDataChildByName("augment-holder");
+        assertNotNull(augmentHolder);
+        assertTrue(augmentHolder.isAugmenting());
+        QName augmentHolderQName = augmentHolder.getQName();
+        assertEquals("augment-holder", augmentHolderQName.getLocalName());
+        assertEquals("t3", augmentHolderQName.getPrefix());
+        assertEquals("Description for augment holder",
+                augmentHolder.getDescription());
     }
 
     @Test
@@ -285,7 +299,7 @@ public class YangParserTest {
         List<RangeConstraint> ranges = leafType.getRanges();
         assertEquals(1, ranges.size());
         RangeConstraint range = ranges.get(0);
-        assertEquals(11L, range.getMin());
+        assertEquals(3L, range.getMin());
         assertEquals(20L, range.getMax());
     }