Fix YANG snippet escaping
[mdsal.git] / binding / mdsal-binding-java-api-generator / src / test / java / org / opendaylight / mdsal / binding / java / api / generator / BuilderGeneratorTest.java
index 16f1b1df91e384c176b3bed2eefae9985fd06a15..bdcda6be6ab7a44cef778c76609e98a15878464a 100644 (file)
@@ -12,20 +12,17 @@ import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.spy;
 
-import java.lang.reflect.Field;
 import java.util.ArrayList;
-import java.util.Collection;
 import java.util.List;
 import org.junit.Test;
-import org.opendaylight.mdsal.binding.model.api.GeneratedProperty;
 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
+import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
 import org.opendaylight.mdsal.binding.model.api.MethodSignature;
 import org.opendaylight.mdsal.binding.model.api.Type;
 
 public class BuilderGeneratorTest {
-
-    private static final String PROPERTIES_FIELD_NAME = "properties";
     private static final String TEST = "test";
+    private static final JavaTypeName TYPE_NAME = JavaTypeName.create(TEST, TEST);
 
     @Test
     public void basicTest() {
@@ -36,65 +33,65 @@ public class BuilderGeneratorTest {
     public void builderTemplateGenerateToStringWithPropertyTest() {
         final GeneratedType genType = mockGenType("get" + TEST);
 
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    CodeHelpers.appendValue(helper, \"_test\", _test);\n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(genType).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    CodeHelpers.appendValue(helper, \"_test\", _test);\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(genType).toString());
     }
 
     @Test
     public void builderTemplateGenerateToStringWithoutAnyPropertyTest() throws Exception {
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(mockGenType(TEST)).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(mockGenType(TEST)).toString());
     }
 
     @Test
     public void builderTemplateGenerateToStringWithMorePropertiesTest() throws Exception {
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    CodeHelpers.appendValue(helper, \"_test1\", _test1);\n" +
-                "    CodeHelpers.appendValue(helper, \"_test2\", _test2);\n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(mockGenTypeMoreMeth("get" + TEST)).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    CodeHelpers.appendValue(helper, \"_test1\", _test1);\n"
+                + "    CodeHelpers.appendValue(helper, \"_test2\", _test2);\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(mockGenTypeMoreMeth("get" + TEST)).toString());
     }
 
     @Test
     public void builderTemplateGenerateToStringWithoutPropertyWithAugmentTest() throws Exception {
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    CodeHelpers.appendValue(helper, \"augmentation\", augmentation.values()); \n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(mockAugment(mockGenType(TEST))).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    CodeHelpers.appendValue(helper, \"augmentation\", augmentations().values());\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(mockAugment(mockGenType(TEST))).toString());
     }
 
     @Test
     public void builderTemplateGenerateToStringWithPropertyWithAugmentTest() throws Exception {
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    CodeHelpers.appendValue(helper, \"_test\", _test);\n" +
-                "    CodeHelpers.appendValue(helper, \"augmentation\", augmentation.values()); \n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(mockAugment(mockGenType("get" + TEST))).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    CodeHelpers.appendValue(helper, \"_test\", _test);\n"
+                + "    CodeHelpers.appendValue(helper, \"augmentation\", augmentations().values());\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(mockAugment(mockGenType("get" + TEST))).toString());
     }
 
     @Test
     public void builderTemplateGenerateToStringWithMorePropertiesWithAugmentTest() throws Exception {
-        assertEquals("@Override\n" +
-                "public java.lang.String toString() {\n" +
-                "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n" +
-                "    CodeHelpers.appendValue(helper, \"_test1\", _test1);\n" +
-                "    CodeHelpers.appendValue(helper, \"_test2\", _test2);\n" +
-                "    CodeHelpers.appendValue(helper, \"augmentation\", augmentation.values()); \n" +
-                "    return helper.toString();\n" +
-                "}\n", genToString(mockAugment(mockGenTypeMoreMeth("get" + TEST))).toString());
+        assertEquals("@Override\n"
+                + "public String toString() {\n"
+                + "    final MoreObjects.ToStringHelper helper = MoreObjects.toStringHelper(\"test\");\n"
+                + "    CodeHelpers.appendValue(helper, \"_test1\", _test1);\n"
+                + "    CodeHelpers.appendValue(helper, \"_test2\", _test2);\n"
+                + "    CodeHelpers.appendValue(helper, \"augmentation\", augmentations().values());\n"
+                + "    return helper.toString();\n"
+                "}\n", genToString(mockAugment(mockGenTypeMoreMeth("get" + TEST))).toString());
     }
 
     private static GeneratedType mockAugment(final GeneratedType genType) {
@@ -108,6 +105,7 @@ public class BuilderGeneratorTest {
 
     private static GeneratedType mockGenTypeMoreMeth(final String methodeName) {
         final GeneratedType genType = spy(GeneratedType.class);
+        doReturn(TYPE_NAME).when(genType).getIdentifier();
         doReturn(TEST).when(genType).getName();
         doReturn(TEST).when(genType).getPackageName();
 
@@ -123,20 +121,14 @@ public class BuilderGeneratorTest {
         return genType;
     }
 
-    @SuppressWarnings("unchecked")
     private static CharSequence genToString(final GeneratedType genType) {
-        try {
-            final BuilderTemplate bt = new BuilderTemplate(genType);
-            final Field propertiesField = bt.getClass().getDeclaredField(PROPERTIES_FIELD_NAME);
-            propertiesField.setAccessible(true);
-            return bt.generateToString((Collection<GeneratedProperty>) propertiesField.get(bt));
-        } catch (IllegalArgumentException | IllegalAccessException | NoSuchFieldException | SecurityException e) {
-            throw new RuntimeException(e);
-        }
+        final BuilderTemplate bt = BuilderGenerator.templateForType(genType);
+        return bt.generateToString(bt.properties);
     }
 
     private static GeneratedType mockGenType(final String methodeName) {
         final GeneratedType genType = spy(GeneratedType.class);
+        doReturn(TYPE_NAME).when(genType).getIdentifier();
         doReturn(TEST).when(genType).getName();
         doReturn(TEST).when(genType).getPackageName();
 
@@ -154,8 +146,7 @@ public class BuilderGeneratorTest {
         final MethodSignature methSign = mock(MethodSignature.class);
         doReturn(methodeName).when(methSign).getName();
         final Type methType = mock(Type.class);
-        doReturn(TEST).when(methType).getName();
-        doReturn(TEST).when(methType).getPackageName();
+        doReturn(TYPE_NAME).when(methType).getIdentifier();
         doReturn(methType).when(methSign).getReturnType();
         return methSign;
     }