added child test

Author: rrayst

annot/src/test/java/com/predic8/membrane/annot/YAMLParsingTest.java

@@ -5,8 +5,7 @@
 
 import static com.predic8.membrane.annot.SpringConfigurationXSDGeneratingAnnotationProcessorTest.MC_MAIN_DEMO;
 import static com.predic8.membrane.annot.util.CompilerHelper.*;
-import static com.predic8.membrane.annot.util.StructureAssertionUtil.assertStructure;
-import static com.predic8.membrane.annot.util.StructureAssertionUtil.clazz;
+import static com.predic8.membrane.annot.util.StructureAssertionUtil.*;
 
 public class YAMLParsingTest {
  @Test
@@ -30,6 +29,46 @@ public class DemoElement {
  );
  }
 
+ @Test
+ public void singleChild() {
+ var sources = splitSources(MC_MAIN_DEMO + """
+ package com.predic8.membrane.demo;
+ import com.predic8.membrane.annot.*;
+ import java.util.List;
+ @MCElement(name="demo")
+ public class DemoElement {
+ ChildElement child;
+ 
+ public ChildElement getChild() {
+ return child;
+ }
+ 
+ @MCChildElement
+ public void setChild(ChildElement child) {
+ this.child = child;
+ }
+ }
+ ---
+ package com.predic8.membrane.demo;
+ import com.predic8.membrane.annot.*;
+ @MCElement(name="child1", topLevel=false)
+ public class Child1Element {
+ }
+ """);
+ var result = CompilerHelper.compile(sources, false);
+ assertCompilerResult(true, result);
+
+ assertStructure(
+ parseYAML(result, """
+ demo:
+ child:
+ child1: {}
+ """),
+ clazz("DemoElement",
+ property("child", clazz("Child1Element")))
+ );
+ }
+
  @Test
  public void twoObjects() {
  var sources = splitSources(MC_MAIN_DEMO + """

annot/src/test/java/com/predic8/membrane/annot/util/StructureAssertionUtil.java

@@ -2,6 +2,8 @@
 
 import com.predic8.membrane.annot.yaml.BeanRegistry;
 
+import java.lang.reflect.Method;
+
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
@@ -14,14 +16,36 @@ public static void assertStructure(BeanRegistry registry, Asserter... asserter)
  }
 
  public interface Asserter {
- void assertStructure(Object o1);
+ void assertStructure(Object bean);
+ }
+
+ public interface Property {
+ void assertStructure(Object bean);
  }
 
- public static Asserter clazz(String clazzName) {
+ public static Asserter clazz(String clazzName, Property... properties) {
  return new Asserter() {
  @Override
- public void assertStructure(Object o1) {
- assertTrue(o1.getClass().getSimpleName().equals(clazzName));
+ public void assertStructure(Object bean) {
+ assertTrue(bean.getClass().getSimpleName().equals(clazzName));
+ for (Property p : properties) {
+ p.assertStructure(bean);
+ }
+ }
+ };
+ }
+
+ public static Property property(String name, Asserter asserter) {
+ return new Property() {
+ @Override
+ public void assertStructure(Object bean) {
+ try {
+ Method getter = bean.getClass().getMethod("get" + Character.toUpperCase(name.charAt(0)) + name.substring(1));
+ Object propertyValue = getter.invoke(bean);
+ asserter.assertStructure(propertyValue);
+ } catch (NoSuchMethodException | IllegalAccessException | java.lang.reflect.InvocationTargetException e) {
+ throw new RuntimeException(e);
+ }
  }
  };
  }