UT for YANG translator utils

Change-Id: I1da6f7d6201f880c27885659e5e52edc3fccbdd6

+/*
+ * Copyright 2016 Open Networking Laboratory
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.onosproject.yangutils.translator.tojava.utils;
+
+import org.junit.Test;
+import org.onosproject.yangutils.translator.GeneratedFileType;
+import org.onosproject.yangutils.translator.tojava.GeneratedMethodTypes;
+import org.onosproject.yangutils.translator.tojava.TraversalType;
+import org.onosproject.yangutils.utils.UtilConstants;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
+import static org.junit.Assert.assertNotNull;
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+
+/**
+ * Unit tests for class definition generator for generated files.
+ */
+public final class ClassDefinitionGeneratorTest {
+
+    /**
+     * Unit test for private constructor.
+     *
+     * @throws SecurityException if any security violation is observed.
+     * @throws NoSuchMethodException if when the method is not found.
+     * @throws IllegalArgumentException if there is illegal argument found.
+     * @throws InstantiationException if instantiation is provoked for the private constructor.
+     * @throws IllegalAccessException if instance is provoked or a method is provoked.
+     * @throws InvocationTargetException when an exception occurs by the method or constructor.
+     */
+    @Test
+    public void callPrivateConstructors() throws SecurityException, NoSuchMethodException,
+    IllegalArgumentException, InstantiationException, IllegalAccessException, InvocationTargetException {
+        Class<?>[] classesToConstruct = {ClassDefinitionGenerator.class };
+        for (Class<?> clazz : classesToConstruct) {
+            Constructor<?> constructor = clazz.getDeclaredConstructor();
+            constructor.setAccessible(true);
+            assertNotNull(constructor.newInstance());
+        }
+    }
+
+    /**
+     * Unit test for builder class definition.
+     */
+    @Test
+    public void generateBuilderClassDefinitionTest() {
+
+        String builderClassDefinition = ClassDefinitionGenerator
+                .generateClassDefinition(GeneratedFileType.BUILDER_CLASS, "BuilderClass");
+        assertThat(true, is(builderClassDefinition.contains(UtilConstants.BUILDER)));
+        assertThat(true, is(builderClassDefinition.contains(UtilConstants.CLASS)));
+    }
+
+    /**
+     * Unit test for builder interface definition.
+     */
+    @Test
+    public void generateBuilderInterfaceDefinitionTest() {
+
+        String builderInterfaceDefinition = ClassDefinitionGenerator
+                .generateClassDefinition(GeneratedFileType.BUILDER_INTERFACE, "BuilderInterfaceClass");
+        assertThat(true, is(builderInterfaceDefinition.contains(UtilConstants.BUILDER)));
+    }
+
+    /**
+     * Unit test for impl class definition.
+     */
+    @Test
+    public void generateImplDefinitionTest() {
+
+        String implDefinition = ClassDefinitionGenerator.generateClassDefinition(GeneratedFileType.IMPL, "ImplClass");
+        assertThat(true, is(implDefinition.contains(UtilConstants.IMPL)));
+    }
+
+    /**
+     * Unit test for interface definition.
+     */
+    @Test
+    public void generateinterfaceDefinitionTest() {
+
+        String interfaceDefinition = ClassDefinitionGenerator.generateClassDefinition(GeneratedFileType.INTERFACE,
+                "InterfaceClass");
+        assertThat(true, is(interfaceDefinition.contains(UtilConstants.INTERFACE)));
+    }
+
+    /**
+     * Unit test for invalid generated type.
+     */
+    @Test
+    public void generateInvalidDefinitionTest() {
+
+        String invalidDefinition = ClassDefinitionGenerator.generateClassDefinition(GeneratedFileType.ALL, "invalid");
+        assertThat(true, is(invalidDefinition == null));
+    }
+
+    /**
+     * Unit test for enum data types.
+     */
+    @Test
+    public void enumDataTypesTest() {
+
+        TraversalType.valueOf(TraversalType.CHILD.toString());
+        GeneratedMethodTypes.valueOf(GeneratedMethodTypes.CONSTRUCTOR.toString());
+    }
+}

+/*
+ * Copyright 2016 Open Networking Laboratory
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.onosproject.yangutils.translator.tojava.utils;
+
+import org.junit.Test;
+import static org.junit.Assert.assertNotNull;
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
+
+/**
+ * Unit tests for java identifier syntax.
+ */
+public final class JavaIdentifierSyntaxTest {
+
+    /**
+     * Unit test for private constructor.
+     *
+     * @throws SecurityException if any security violation is observed.
+     * @throws NoSuchMethodException if when the method is not found.
+     * @throws IllegalArgumentException if there is illegal argument found.
+     * @throws InstantiationException if instantiation is provoked for the private constructor.
+     * @throws IllegalAccessException if instance is provoked or a method is provoked.
+     * @throws InvocationTargetException when an exception occurs by the method or constructor.
+     */
+    public void callPrivateConstructors() throws SecurityException, NoSuchMethodException, IllegalArgumentException,
+    InstantiationException, IllegalAccessException, InvocationTargetException {
+        Class<?>[] classesToConstruct = {JavaIdentifierSyntax.class };
+        for (Class<?> clazz : classesToConstruct) {
+            Constructor<?> constructor = clazz.getDeclaredConstructor();
+            constructor.setAccessible(true);
+            assertNotNull(constructor.newInstance());
+        }
+    }
+
+    /**
+     * Unit test for testing the package path generation from a parent package.
+     */
+    @Test
+    public void getPackageFromParentTest() {
+        String pkgFromParent = JavaIdentifierSyntax.getPackageFromParent("test5.test6.test7", "test1:test2:test3");
+        assertThat(pkgFromParent.equals("test5.test6.test7.test1.test2.test3"), is(true));
+    }
+
+    /**
+     * Unit test for root package generation with revision complexity.
+     */
+    @Test
+    public void getRootPackageTest() {
+
+        String rootPackage = JavaIdentifierSyntax.getRootPackage((byte) 0, "test1:test2:test3", "5-1-2000");
+        assertThat(rootPackage.equals("org.onosproject.yang.gen.v0.test1.test2.test3.rev05012000"), is(true));
+    }
+
+    /**
+     * Unit test for root package generation without revision complexity.
+     */
+    @Test
+    public void getRootPackageWithRevTest() {
+
+        String rootPkgWithRev = JavaIdentifierSyntax.getRootPackage((byte) 0, "test1:test2:test3", "25-01-1992");
+        assertThat(rootPkgWithRev.equals("org.onosproject.yang.gen.v0.test1.test2.test3.rev25011992"), is(true));
+    }
+
+    /**
+     * Unit test for capitalizing the incoming string.
+     */
+    @Test
+    public void getCapitalCaseTest() {
+
+        String capitalCase = JavaIdentifierSyntax.getCaptialCase("test_this");
+        assertThat(capitalCase.equals("Test_this"), is(true));
+    }
+
+    /**
+     * Unit test for getting the camel case for the received string.
+     */
+    @Test
+    public void getCamelCaseTest() {
+        String camelCase = JavaIdentifierSyntax.getCamelCase("test-camel-case-identifier");
+        assertThat(camelCase.equals("testCamelCaseIdentifier"), is(true));
+    }
+}

+/*
+ * Copyright 2016 Open Networking Laboratory
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.onosproject.yangutils.translator.tojava.utils;
+
+import org.junit.Test;
+import static org.hamcrest.core.Is.is;
+import static org.hamcrest.Matchers.nullValue;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertNotNull;
+import org.onosproject.yangutils.datamodel.YangDataTypes;
+import org.onosproject.yangutils.datamodel.YangType;
+import org.onosproject.yangutils.translator.GeneratedFileType;
+import org.onosproject.yangutils.translator.tojava.AttributeInfo;
+import org.onosproject.yangutils.translator.tojava.GeneratedMethodTypes;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Unit tests for generated methods from the file type.
+ */
+public final class MethodsGeneratorTest {
+
+    public static AttributeInfo testAttr = new AttributeInfo();
+    public static YangType<?> attrType = new YangType<>();
+
+    /**
+     * Unit test for private constructor.
+     *
+     * @throws SecurityException if any security violation is observed.
+     * @throws NoSuchMethodException if when the method is not found.
+     * @throws IllegalArgumentException if there is illegal argument found.
+     * @throws InstantiationException if instantiation is provoked for the private constructor.
+     * @throws IllegalAccessException if instance is provoked or a method is provoked.
+     * @throws InvocationTargetException when an exception occurs by the method or constructor.
+     */
+    @Test
+    public void callPrivateConstructors() throws SecurityException, NoSuchMethodException, IllegalArgumentException,
+    InstantiationException, IllegalAccessException, InvocationTargetException {
+
+        Class<?>[] classesToConstruct = {MethodsGenerator.class };
+        for (Class<?> clazz : classesToConstruct) {
+            Constructor<?> constructor = clazz.getDeclaredConstructor();
+            constructor.setAccessible(true);
+            assertNotNull(constructor.newInstance());
+        }
+    }
+
+    /**
+     * Unit test for checking the generated builder class method.
+     */
+    @Test
+    public void getMethodBuilderClassTest() {
+
+        attrType.setDataTypeName("integer");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.INT8);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeBuilderClassTest");
+        testAttr.setAttributeType(attrType);
+        String builderClassMethod = MethodsGenerator.getMethodString(testAttr, GeneratedFileType.BUILDER_CLASS);
+        assertThat(builderClassMethod.contains("public Byte getAttributeBuilderClassTest() {"), is(true));
+        assertThat(builderClassMethod.contains(
+                "public testnameof setAttributeBuilderClassTest(Byte attributeBuilderClassTest) {"), is(true));
+    }
+
+    /**
+     * Unit test for checking the generated builder interface method.
+     */
+    @Test
+    public void getMethodBuilderInterfaceTest() {
+
+        attrType.setDataTypeName("integer16");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.INT16);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeBuilderInterfaceTest");
+        testAttr.setAttributeType(attrType);
+        String builderInterfaceMethod = MethodsGenerator.getMethodString(testAttr, GeneratedFileType.BUILDER_INTERFACE);
+        assertThat(builderInterfaceMethod.contains("Returns the attribute attributeBuilderInterfaceTest.")
+                && builderInterfaceMethod.contains("Short getAttributeBuilderInterfaceTest();")
+                && builderInterfaceMethod.contains("Returns the builder object of attributeBuilderInterfaceTest.")
+                && builderInterfaceMethod
+                .contains("Builder setAttributeBuilderInterfaceTest(Short attributeBuilderInterfaceTest);"),
+                is(true));
+    }
+
+    /**
+     * Unit test for checking the generated impl method.
+     */
+    @Test
+    public void getMethodImplTest() {
+
+        attrType.setDataTypeName("integer16");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.INT16);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeImplTest");
+        testAttr.setAttributeType(attrType);
+        String implMethod = MethodsGenerator.getMethodString(testAttr, GeneratedFileType.IMPL);
+        assertThat(implMethod.contains("public Short getAttributeImplTest() {")
+                && implMethod.contains("return attributeImplTest;"), is(true));
+    }
+
+    /**
+     * Unit test for checking the generated interface method.
+     */
+    @Test
+    public void getMethodInterfaceTest() {
+
+        attrType.setDataTypeName("binary");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.INT32);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeInterfaceTest");
+        testAttr.setAttributeType(attrType);
+        String interfaceMethod = MethodsGenerator.getMethodString(testAttr, GeneratedFileType.INTERFACE);
+        assertThat(interfaceMethod.contains("Returns the attribute attributeInterfaceTest.")
+                && interfaceMethod.contains("@return attributeInterfaceTest")
+                && interfaceMethod.contains("Int getAttributeInterfaceTest();"), is(true));
+    }
+
+    /**
+     * Unit test for checking the response for an invalid input.
+     */
+    @Test
+    public void getMethodInvalidTest() {
+
+        attrType.setDataTypeName("decimal64");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.DECIMAL64);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeInvalidTest");
+        testAttr.setAttributeType(attrType);
+        String invalidMethod = MethodsGenerator.getMethodString(testAttr, GeneratedFileType.ALL);
+        assertThat(invalidMethod, is(nullValue()));
+    }
+
+    /**
+     * Unit test for checking the generated construct method info.
+     */
+    @Test
+    public void constructMethodInfoTest() {
+
+        attrType.setDataTypeName("decimal64");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.DECIMAL64);
+        attrType.getDataType();
+        MethodsGenerator.setBuilderClassName("testnameof");
+        String builderClassName = MethodsGenerator.getBuilderClassName();
+        assertThat(builderClassName.equals("testnameof"), is(true));
+        String implTypenullMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL, "testname",
+                GeneratedMethodTypes.GETTER, null);
+        assertThat(implTypenullMethod.contains("public Testname getTestname() {")
+                && implTypenullMethod.contains("return testname;"), is(true));
+        String implTypeGetterMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL, "testname",
+                GeneratedMethodTypes.GETTER, attrType);
+        assertThat(implTypeGetterMethod.contains("public Decimal64 getTestname()")
+                && implTypeGetterMethod.contains("return testname;"), is(true));
+        String implTypeConstructorMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL, "testname",
+                GeneratedMethodTypes.CONSTRUCTOR, attrType);
+        assertThat(implTypeConstructorMethod.contains("public testnameImpl(testnameBuilder testnameObject) {")
+                && implTypeConstructorMethod.contains("}"), is(true));
+        String implTypeDefaultConstructorMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL,
+                "testname", GeneratedMethodTypes.DEFAULT_CONSTRUCTOR, attrType);
+        assertThat(implTypeDefaultConstructorMethod.contains("public testnameImpl() {")
+                && implTypeDefaultConstructorMethod.contains("}"), is(true));
+        String implTypeSetterMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL, "testname",
+                GeneratedMethodTypes.SETTER, attrType);
+        assertThat(implTypeSetterMethod, is(nullValue()));
+        String builderInterfaceTypeSetterMethod = MethodsGenerator.constructMethodInfo(
+                GeneratedFileType.BUILDER_INTERFACE, "testname2", GeneratedMethodTypes.SETTER, attrType);
+        assertThat(builderInterfaceTypeSetterMethod.contains("Builder setTestname2(Decimal64 testname2);"), is(true));
+        String builderInterfaceTypeGetterMethod = MethodsGenerator.constructMethodInfo(
+                GeneratedFileType.BUILDER_INTERFACE, "testname2", GeneratedMethodTypes.GETTER, attrType);
+        assertThat(builderInterfaceTypeGetterMethod.contains("Decimal64 getTestname2();"), is(true));
+        String builderInterfaceTypeBuildMethod = MethodsGenerator.constructMethodInfo(
+                GeneratedFileType.BUILDER_INTERFACE, "testname2", GeneratedMethodTypes.BUILD, attrType);
+        assertThat(builderInterfaceTypeBuildMethod.contains("testname2 build();"), is(true));
+        String builderInterfaceTypeConstructorMethod = MethodsGenerator.constructMethodInfo(
+                GeneratedFileType.BUILDER_INTERFACE, "testname2", GeneratedMethodTypes.CONSTRUCTOR, attrType);
+        assertThat(builderInterfaceTypeConstructorMethod, is(nullValue()));
+        String builderClassTypeBuildMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.BUILDER_CLASS,
+                "testname2", GeneratedMethodTypes.BUILD, attrType);
+        assertThat(builderClassTypeBuildMethod.contains("public testname2 build() {")
+                && builderClassTypeBuildMethod.contains("return new testname2Impl(this);"), is(true));
+        String builderClassTypeGetterMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.BUILDER_CLASS,
+                "testname2", GeneratedMethodTypes.GETTER, attrType);
+        assertThat(builderClassTypeGetterMethod.contains("public Decimal64 getTestname2() {")
+                && builderClassTypeGetterMethod.contains("return testname2;"), is(true));
+        String builderClassTypeSetterMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.BUILDER_CLASS,
+                "testname2", GeneratedMethodTypes.SETTER, attrType);
+        assertThat(builderClassTypeSetterMethod.contains("public testnameof setTestname2(Decimal64 testname2) {")
+                && builderClassTypeSetterMethod.contains("this.testname2 = testname2;"), is(true));
+        String builderClassTypeDefaultConstructorMethod = MethodsGenerator.constructMethodInfo(
+                GeneratedFileType.BUILDER_CLASS, "testname2", GeneratedMethodTypes.DEFAULT_CONSTRUCTOR, attrType);
+        assertThat(builderClassTypeDefaultConstructorMethod.contains("public testname2Builder() {"), is(true));
+        String builderClassTypeConstructorMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.BUILDER_CLASS,
+                "testname2", GeneratedMethodTypes.CONSTRUCTOR, attrType);
+        assertThat(builderClassTypeConstructorMethod, is(nullValue()));
+        String invalidMethod = MethodsGenerator.constructMethodInfo(GeneratedFileType.ALL, "testname2",
+                GeneratedMethodTypes.CONSTRUCTOR, attrType);
+        assertThat(invalidMethod, is(nullValue()));
+    }
+
+    /**
+     * Unit test for checking the method constructor.
+     */
+    @Test
+    public void getMethodConstructorTest() {
+
+        MethodsGenerator.parseBuilderInterfaceBuildMethodString("testname7");
+        attrType.setDataTypeName("binary");
+        attrType.getDataTypeName();
+        attrType.setDataType(YangDataTypes.BINARY);
+        attrType.getDataType();
+        testAttr.setAttributeName("attributeTest");
+        testAttr.setAttributeType(attrType);
+        List<AttributeInfo> settingAttributes = new ArrayList<AttributeInfo>();
+        settingAttributes.add(testAttr);
+        MethodsGenerator.setAttrInfo(settingAttributes);
+        String methodConstructor = MethodsGenerator.constructMethodInfo(GeneratedFileType.IMPL, "testname",
+                GeneratedMethodTypes.CONSTRUCTOR, attrType);
+        assertThat(
+                methodConstructor.contains("public testnameImpl(testnameBuilder testnameObject) {")
+                && methodConstructor.contains("this.attributeTest = testnameObject.getAttributeTest();"),
+                is(true));
+    }
+
+    /**
+     * Unit test for checking the values received from constructor, default constructor and build string formation.
+     */
+    @Test
+    public void getValuesTest() {
+        String stringConstructor = MethodsGenerator.getConstructorString("testname");
+        assertThat(
+                stringConstructor.contains("Construct the object of testnameImpl.")
+                && stringConstructor.contains("@param testnameObject builder object of  testname")
+                && stringConstructor.contains("public testnameImpl(testnameBuilder testnameObject) {"),
+                is(true));
+        String stringDefaultConstructor = MethodsGenerator.getDefaultConstructorString(GeneratedFileType.BUILDER_CLASS,
+                "testname");
+        assertThat(stringDefaultConstructor.contains("Default Constructor.")
+                && stringDefaultConstructor.contains("public testnameBuilder() {")
+                && stringDefaultConstructor.contains("}"), is(true));
+        String stringBuild = MethodsGenerator.getBuildString("testname");
+        assertThat(
+                stringBuild.contains("public testname build() {")
+                && stringBuild.contains("return new testnameImpl(this);") && stringBuild.contains("}"),
+                is(true));
+    }
+}
\ No newline at end of file