Added YANG typedef support in the YANG parser

[controller.git] / opendaylight / sal / yang-prototype / code-generator / yang-model-parser-impl / src / test / java / org / opendaylight / controller / yang / model / parser / impl / YangModelParserTest.java

diff --git a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/model/parser/impl/YangModelParserTest.java b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/model/parser/impl/YangModelParserTest.java
index 74cfed5fb13bbfaf8eeed80298216869bc3a8286..889290f65c1ce20dc1ffe0dd6e690217062305c5 100644 (file)
--- a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/model/parser/impl/YangModelParserTest.java
+++ b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/model/parser/impl/YangModelParserTest.java
@@ -9,17 +9,24 @@ package org.opendaylight.controller.yang.model.parser.impl;
 
 import static org.junit.Assert.*;
 
-import java.io.IOException;
+import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 
 import org.junit.Before;
 import org.junit.Test;
+import org.opendaylight.controller.model.api.type.IntegerTypeDefinition;
+import org.opendaylight.controller.model.api.type.PatternConstraint;
+import org.opendaylight.controller.model.api.type.RangeConstraint;
+import org.opendaylight.controller.model.util.Decimal64;
 import org.opendaylight.controller.model.util.Int32;
+import org.opendaylight.controller.model.util.StringType;
 import org.opendaylight.controller.yang.model.api.AugmentationSchema;
 import org.opendaylight.controller.yang.model.api.ContainerSchemaNode;
 import org.opendaylight.controller.yang.model.api.LeafSchemaNode;
 import org.opendaylight.controller.yang.model.api.ListSchemaNode;
 import org.opendaylight.controller.yang.model.api.Module;
+import org.opendaylight.controller.yang.model.api.TypeDefinition;
 import org.opendaylight.controller.yang.model.parser.api.YangModelParser;
 
 public class YangModelParserTest {
@@ -29,7 +36,7 @@ public class YangModelParserTest {
     private YangModelParser tested;
 
     @Before
-    public void init() throws IOException {
+    public void init() {
         tested = new YangModelParserImpl();
     }
 
@@ -68,13 +75,14 @@ public class YangModelParserTest {
 
         ListSchemaNode list = (ListSchemaNode)container.getDataChildByName("ifEntry");
         assertNotNull(list);
+        assertEquals(1, list.getAvailableAugmentations().size());
 
         LeafSchemaNode leaf = (LeafSchemaNode)list.getDataChildByName("ds0ChannelNumber");
         assertNotNull(leaf);
     }
 
     @Test
-    public void testTypeDef() {
+    public void testTypedefRangesResolving() {
         Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
         assertEquals(2, modules.size());
 
@@ -87,7 +95,141 @@ public class YangModelParserTest {
         assertNotNull(m1);
 
         LeafSchemaNode testleaf = (LeafSchemaNode)m1.getDataChildByName("testleaf");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
+        assertTrue(testleaf.getType().getBaseType() instanceof Int32);
+        Int32 baseTypeCast = (Int32)baseType;
+        List<RangeConstraint> ranges = baseTypeCast.getRangeStatements();
+        assertEquals(2, ranges.size());
+        RangeConstraint range = ranges.get(0);
+        assertEquals(2L, range.getMin());
+        assertEquals(20L, range.getMax());
+    }
+
+    @Test
+    public void testTypedefPatternsResolving() {
+        Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
+        assertEquals(2, modules.size());
+
+        Module m1 = null;
+        for(Module m : modules) {
+            if(m.getName().equals("types1")) {
+                m1 = m;
+            }
+        }
+        assertNotNull(m1);
+
+        LeafSchemaNode testleaf = (LeafSchemaNode)m1.getDataChildByName("test-string-leaf");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
+        assertTrue(testleaf.getType().getBaseType() instanceof StringType);
+        StringType baseTypeCast = (StringType)baseType;
+
+        Set<String> expectedRegularExpressions = new HashSet<String>();
+        expectedRegularExpressions.add("[a-k]*");
+        expectedRegularExpressions.add("[b-u]*");
+        expectedRegularExpressions.add("[e-z]*");
+
+        Set<String> actualRegularExpressions = new HashSet<String>();
+        List<PatternConstraint> patterns = baseTypeCast.getPatterns();
+        for(PatternConstraint pc : patterns) {
+            actualRegularExpressions.add(pc.getRegularExpression());
+        }
+
+        assertEquals(expectedRegularExpressions, actualRegularExpressions);
+    }
+
+    @Test
+    public void testTypedefLengthsResolving() {
+        Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
+        assertEquals(2, modules.size());
+
+        Module m1 = null;
+        for(Module m : modules) {
+            if(m.getName().equals("types1")) {
+                m1 = m;
+            }
+        }
+        assertNotNull(m1);
+
+        LeafSchemaNode testleaf = (LeafSchemaNode)m1.getDataChildByName("test-int-leaf");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
+        assertTrue(testleaf.getType().getBaseType() instanceof IntegerTypeDefinition);
+        Int32 baseTypeCast = (Int32)baseType;
+
+        Long[][] expectedRanges = new Long[3][2];
+        expectedRanges[0] = new Long[]{10L, 20L};
+        expectedRanges[1] = new Long[]{12L, 18L};
+        expectedRanges[2] = new Long[]{14L, 16L};
+
+        List<RangeConstraint> actualRanges = baseTypeCast.getRangeStatements();
+        assertEquals(3, actualRanges.size());
+        for(int i = 0; i < actualRanges.size(); i++) {
+            assertEquals(expectedRanges[i][0], actualRanges.get(i).getMin());
+            assertEquals(expectedRanges[i][1], actualRanges.get(i).getMax());
+        }
+    }
+
+    @Test
+    public void testTypeDef() {
+        Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
+        assertEquals(2, modules.size());
+
+        Module m2 = null;
+        for(Module m : modules) {
+            if(m.getName().equals("types2")) {
+                m2 = m;
+            }
+        }
+        assertNotNull(m2);
+
+        LeafSchemaNode testleaf = (LeafSchemaNode)m2.getDataChildByName("nested-type-leaf");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
         assertTrue(testleaf.getType().getBaseType() instanceof Int32);
+        Int32 baseTypeCast = (Int32)baseType;
+        List<RangeConstraint> ranges = baseTypeCast.getRangeStatements();
+        assertEquals(2, ranges.size());
+        RangeConstraint range = ranges.get(0);
+        assertEquals(2L, range.getMin());
+        assertEquals(20L, range.getMax());
+    }
+
+    @Test
+    public void testTypedefDecimal1() {
+        Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
+        assertEquals(2, modules.size());
+
+        Module m1 = null;
+        for(Module m : modules) {
+            if(m.getName().equals("types1")) {
+                m1 = m;
+            }
+        }
+        assertNotNull(m1);
+
+        LeafSchemaNode testleaf = (LeafSchemaNode)m1.getDataChildByName("test-decimal-leaf");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
+        assertTrue(testleaf.getType().getBaseType() instanceof Decimal64);
+        Decimal64 baseTypeCast = (Decimal64)baseType;
+        assertEquals(4, (int)baseTypeCast.getFractionDigits());
+    }
+
+    @Test
+    public void testTypedefDecimal2() {
+        Set<Module> modules = tested.parseYangModels(testFile1, testFile2);
+        assertEquals(2, modules.size());
+
+        Module m1 = null;
+        for(Module m : modules) {
+            if(m.getName().equals("types1")) {
+                m1 = m;
+            }
+        }
+        assertNotNull(m1);
+
+        LeafSchemaNode testleaf = (LeafSchemaNode)m1.getDataChildByName("test-decimal-leaf2");
+        TypeDefinition<?> baseType = testleaf.getType().getBaseType();
+        assertTrue(testleaf.getType().getBaseType() instanceof Decimal64);
+        Decimal64 baseTypeCast = (Decimal64)baseType;
+        assertEquals(5, (int)baseTypeCast.getFractionDigits());
     }
 
 }