Binding generator v2 - uses statement - uses of list
[mdsal.git] / binding2 / mdsal-binding2-generator-impl / src / test / java / org / opendaylight / mdsal / binding / javav2 / generator / impl / BindingGeneratorImplTest.java
index c9adc7c6101b607666777197b0e4484d9096ba9b..674632cc3b726ff0da277043877eb4f3e50bb873 100644 (file)
@@ -15,16 +15,60 @@ import static org.junit.Assert.fail;
 import java.util.List;
 import org.junit.Test;
 import org.opendaylight.mdsal.binding.javav2.generator.api.BindingGenerator;
+import org.opendaylight.mdsal.binding.javav2.model.api.Enumeration;
 import org.opendaylight.mdsal.binding.javav2.model.api.GeneratedProperty;
 import org.opendaylight.mdsal.binding.javav2.model.api.GeneratedTransferObject;
 import org.opendaylight.mdsal.binding.javav2.model.api.GeneratedType;
 import org.opendaylight.mdsal.binding.javav2.model.api.MethodSignature;
 import org.opendaylight.mdsal.binding.javav2.model.api.Type;
+import org.opendaylight.mdsal.binding.javav2.model.api.type.builder.GeneratedTOBuilder;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
 
 public class BindingGeneratorImplTest {
 
+    @Test
+    public void genTypesTypeDefTest() throws Exception {
+        final BindingGeneratorImpl bg = new BindingGeneratorImpl(false);
+        final SchemaContext schemaContext = YangParserTestUtils.parseYangSource("/generator/test-typedef.yang");
+        final List<Type> generateTypes = bg.generateTypes(schemaContext);
+        assertNotNull(generateTypes);
+        for (final Type genType : generateTypes) {
+            if (genType.getName().equals("MyInnerCont") && genType.getPackageName()
+                    .equals("org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.rev170206.data.asteriskmy_cont")) {
+                final GeneratedType gt = (GeneratedType) genType;
+                for (final MethodSignature methodSignature : gt.getMethodDefinitions()) {
+                    if (methodSignature.getName().equals("getMyLeaf2")) {
+                        assertEquals(methodSignature.getReturnType().getName(), "MyType");
+                    }
+                }
+            }
+        }
+    }
+
+    @Test
+    public void generatedTypesEnumTest() throws Exception {
+        final BindingGenerator bg = new BindingGeneratorImpl(false);
+        final SchemaContext context = YangParserTestUtils.parseYangSource("/generator/apple-test.yang");
+        final List<Type> generateTypes = bg.generateTypes(context);
+        assertNotNull(generateTypes);
+        assertTrue(!generateTypes.isEmpty());
+        for (final Type type : generateTypes) {
+            if (type.getName().equals("Apple") && type.getPackageName()
+                    .equals("org.opendaylight.mdsal.gen.javav2.urn.test.simple.apple.rev170503.data")) {
+                final GeneratedType gt = (GeneratedType) type;
+                final Enumeration enumeration = gt.getEnumerations().get(0);
+                assertEquals("Apple1", enumeration.getName());
+                assertEquals("org.opendaylight.mdsal.gen.javav2.urn.test.simple.apple.rev170503.data.Apple",
+                        enumeration.getPackageName());
+                assertEquals("apple", enumeration.getValues().get(0).getName());
+                assertEquals("APPLE", enumeration.getValues().get(0).getMappedName());
+                assertEquals("apple1", enumeration.getValues().get(1).getName());
+                assertEquals("APPLE1", enumeration.getValues().get(1).getMappedName());
+            }
+        }
+    }
+
     @Test
     public void generatedTypesTest() throws Exception {
         final BindingGenerator bg = new BindingGeneratorImpl(false);
@@ -33,27 +77,27 @@ public class BindingGeneratorImplTest {
 
         assertNotNull(generateTypes);
         assertTrue(!generateTypes.isEmpty());
-        assertEquals(7, generateTypes.size());
+        assertEquals(3, generateTypes.size());
         final int[] test_i = { 0 };
         for (final Type t : generateTypes) {
             if (t instanceof GeneratedTransferObject) {
                 final GeneratedTransferObject genTransferObj = (GeneratedTransferObject) t;
                 assertBaseGeneratedType(genTransferObj, "MyListKey",
-                        "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list.wrapper");
+                        "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list");
                 assertEquals(3, genTransferObj.getProperties().size());
                 int test_j = 0;
                 for (final GeneratedProperty generatedProperty : genTransferObj.getProperties()) {
                     switch (generatedProperty.getName()) {
-                        case "key":
-                            assertEquals("MyListKey", generatedProperty.getReturnType().getName());
+                        case "keyReservedWord":
+                            assertEquals("String", generatedProperty.getReturnType().getName());
                             test_j++;
                             break;
                         case "key1":
-                            assertEquals("MyListKey1", generatedProperty.getReturnType().getName());
+                            assertEquals("String", generatedProperty.getReturnType().getName());
                             test_j++;
                             break;
                         case "key2":
-                            assertEquals("MyListKey2", generatedProperty.getReturnType().getName());
+                            assertEquals("String", generatedProperty.getReturnType().getName());
                             test_j++;
                             break;
                         default:
@@ -66,7 +110,63 @@ public class BindingGeneratorImplTest {
                 testActualType((GeneratedType) t, test_i);
             }
         }
-        assertEquals(7, test_i[0]);
+        assertEquals(3, test_i[0]);
+    }
+
+    @Test
+    public void generateTypesDescriptionsTest() throws Exception {
+        final BindingGenerator bg = new BindingGeneratorImpl(true);
+        final SchemaContext context = YangParserTestUtils.parseYangSources("/base/with_import/");
+        assertNotNull(context);
+
+        final List<Type> generateTypes = bg.generateTypes(context, context.getModules());
+        assertNotNull(generateTypes);
+        assertTrue(!generateTypes.isEmpty());
+
+        for (final Type type : generateTypes) {
+            if (type.getName().equals("TestData")) {
+                final String description = ((GeneratedType) type).getDescription().get();
+                description
+                        .contains("    import test-import { prefix \"imported-test\"; revision-date 2017-04-21; }\n\n");
+                description.contains("    revision 2017-02-06;\n\n");
+                description.contains("    typedef my-type {\n        type int8;\n    }");
+                description.contains("    container *my-cont {\n    }\n");
+            }
+        }
+    }
+
+    @Test
+    public void generateTypesIdentityTest() throws Exception {
+        final BindingGenerator bg = new BindingGeneratorImpl(true);
+        final SchemaContext context = YangParserTestUtils.parseYangSources("/identity/");
+        assertNotNull(context);
+
+        final List<Type> generateTypes = bg.generateTypes(context, context.getModules());
+        assertNotNull(generateTypes);
+        assertTrue(!generateTypes.isEmpty());
+        for (final Type type : generateTypes) {
+            if (type.getFullyQualifiedName()
+                    .equals("org.opendaylight.mdsal.gen.javav2.identity3.module.rev170708.ident.Iden1")) {
+                final GeneratedTransferObject genTO = (GeneratedTransferObject)type;
+                assertEquals("org.opendaylight.mdsal.gen.javav2.identity3.module.rev170708.ident.Iden2",
+                        genTO.getSuperType().getFullyQualifiedName());
+
+            }
+            if (type.getFullyQualifiedName()
+                    .equals("org.opendaylight.mdsal.gen.javav2.identity3.module.rev170708.ident.Iden2")) {
+                final GeneratedTransferObject genTO = (GeneratedTransferObject)type;
+                assertEquals("org.opendaylight.mdsal.gen.javav2.identity.import_.rev170602.ident.Iden1",
+                        genTO.getSuperType().getFullyQualifiedName());
+
+            }
+            if (type.getFullyQualifiedName()
+                    .equals("org.opendaylight.mdsal.gen.javav2.identity3.module.rev170708.ident.Iden3")) {
+                final GeneratedTransferObject genTO = (GeneratedTransferObject)type;
+                assertEquals("org.opendaylight.mdsal.gen.javav2.identity3.module.rev170708.ident.Iden1",
+                        genTO.getSuperType().getFullyQualifiedName());
+
+            }
+        }
     }
 
     private void testActualType(final GeneratedType t, final int[] test_i) {
@@ -116,32 +216,24 @@ public class BindingGeneratorImplTest {
                     switch (m.getName()) {
                         case "getKey":
                                 assertMethod(t, "getKey", "MyListKey",
-                                        "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list.wrapper",
+                                        "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list",
                                         m);
                             test_j++;
                             break;
-                        case "getMyListKey1":
-                            assertMethod(t, "getMyListKey1", "MyListKey1",
-                                    "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list",
-                                    m);
+                        case "getKey1":
+                            assertMethod(t, "getKey1", "String","java.lang", m);
                             test_j++;
                             break;
-                        case "getMyListKey2":
-                            assertMethod(t, "getMyListKey2", "MyListKey2",
-                                    "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list",
-                                    m);
+                        case "getKey2":
+                            assertMethod(t, "getKey2", "String","java.lang", m);
                             test_j++;
                             break;
-                        case "getMyListFoo":
-                            assertMethod(t, "getMyListFoo", "MyListFoo",
-                                    "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.data.my_list",
-                                    m);
+                        case "getFoo":
+                            assertMethod(t, "getFoo", "String","java.lang", m);
                             test_j++;
                             break;
-                        case "getMyListKey":
-                            assertMethod(t, "getMyListKey", "MyListKey",
-                                    "org.opendaylight.mdsal.gen.javav2.urn.test.simple.test.list.rev170314.key.my_list",
-                                    m);
+                        case "getKeyReservedWord":
+                            assertMethod(t, "getKeyReservedWord", "String","java.lang", m);
                             test_j++;
                             break;
                         default: