Implemented use of ChildOf interface in generated classes.

[yangtools.git] / code-generator / binding-generator-impl / src / test / java / org / opendaylight / yangtools / sal / binding / generator / impl / GeneratedTypesTest.java

diff --git a/code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/GeneratedTypesTest.java b/code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/GeneratedTypesTest.java
index 0658adaec3469117577cf2373432e5fd56200413..2e10a4a1be7003ab3fa3134648ddbb8607b47c56 100644 (file)
--- a/code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/GeneratedTypesTest.java
+++ b/code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/GeneratedTypesTest.java
@@ -264,11 +264,17 @@ public class GeneratedTypesTest {
         assertNotNull(genTypes);\r
         assertEquals(3, genTypes.size());\r
 \r
-        final GeneratedType simpleContainer = (GeneratedType) genTypes.get(1);\r
-        final GeneratedType nestedContainer = (GeneratedType) genTypes.get(2);\r
-\r
-        assertEquals("SimpleContainer", simpleContainer.getName());\r
-        assertEquals("NestedContainer", nestedContainer.getName());\r
+        GeneratedType simpleContainer = (GeneratedType) genTypes.get(1);\r
+        GeneratedType nestedContainer = (GeneratedType) genTypes.get(2);\r
+        for (Type t : genTypes) {\r
+            if ("SimpleContainer".equals(t.getName())) {\r
+                simpleContainer = (GeneratedType)t;\r
+            } else if ("NestedContainer".equals(t.getName())) {\r
+                nestedContainer = (GeneratedType)t;\r
+            }\r
+        }\r
+        assertNotNull(simpleContainer);\r
+        assertNotNull(nestedContainer);\r
         assertEquals(3, simpleContainer.getMethodDefinitions().size());\r
         assertEquals(2, nestedContainer.getMethodDefinitions().size());\r
 \r
@@ -343,11 +349,17 @@ public class GeneratedTypesTest {
         assertNotNull(genTypes);\r
         assertEquals(3, genTypes.size());\r
 \r
-        final GeneratedType simpleContainer = (GeneratedType) genTypes.get(1);\r
-        final GeneratedType nestedContainer = (GeneratedType) genTypes.get(2);\r
-\r
-        assertEquals("SimpleContainer", simpleContainer.getName());\r
-        assertEquals("NestedContainer", nestedContainer.getName());\r
+        GeneratedType simpleContainer = (GeneratedType) genTypes.get(1);\r
+        GeneratedType nestedContainer = (GeneratedType) genTypes.get(2);\r
+        for (Type t : genTypes) {\r
+            if ("SimpleContainer".equals(t.getName())) {\r
+                simpleContainer = (GeneratedType)t;\r
+            } else if ("NestedContainer".equals(t.getName())) {\r
+                nestedContainer = (GeneratedType)t;\r
+            }\r
+        }\r
+        assertNotNull(simpleContainer);\r
+        assertNotNull(nestedContainer);\r
         assertEquals(3, simpleContainer.getMethodDefinitions().size());\r
         assertEquals(2, nestedContainer.getMethodDefinitions().size());\r
 \r