Fix for Bug 140.

Signed-off-by: Martin Vitez <mvitez@cisco.com>

diff --git a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
index 8a6d638c5d584bd30db6cad5e4a25717d2f5a26d..aeba4b82d8d5d7c6e3d04ea5945ddb1f7c0660f8 100644 (file)
--- a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
+++ b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
@@ -846,12 +846,14 @@ public class BindingGeneratorImpl implements BindingGenerator {
                 tmpPath.add(currentName);
                 val dataNodeParent = parent as DataNodeContainer;
                 for (u : dataNodeParent.uses) {
-                    var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);
-                    if (!(targetGrouping instanceof GroupingDefinition)) {
-                        throw new IllegalArgumentException("Failed to generate code for augment in " + u);
+                    if (result == null) {\r
+                        var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);
+                        if (!(targetGrouping instanceof GroupingDefinition)) {
+                            throw new IllegalArgumentException("Failed to generate code for augment in " + u);
+                        }
+                        var gr = targetGrouping as GroupingDefinition;
+                        result = gr.getDataChildByName(currentName);\r
                     }
-                    var gr = targetGrouping as GroupingDefinition;
-                    result = gr.getDataChildByName(currentName);
                 }
                 if (result == null) {
                     currentName = (parent as SchemaNode).QName.localName;