Add DataSchemaContextNode/SchemaInferenceStack integration

[yangtools.git] / data / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / AugmentationContextNode.java

diff --git a/data/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/AugmentationContextNode.java b/data/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/AugmentationContextNode.java
index 15fb6a2d5f37aacc550917030cb73410d6edd7c3..2482631c6f6015831ad22e2b96f3e37d56ef795f 100644 (file)
--- a/data/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/AugmentationContextNode.java
+++ b/data/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/AugmentationContextNode.java
@@ -15,10 +15,11 @@ import org.opendaylight.yangtools.yang.model.api.AugmentationTarget;
 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 import org.opendaylight.yangtools.yang.model.util.EffectiveAugmentationSchema;
+import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
 
 final class AugmentationContextNode extends DataContainerContextNode<AugmentationIdentifier> {
-    AugmentationContextNode(final AugmentationSchemaNode augmentation, final DataNodeContainer schema) {
-        super(augmentationIdentifierFrom(augmentation), EffectiveAugmentationSchema.create(augmentation, schema), null);
+    AugmentationContextNode(final AugmentationSchemaNode augmentation, final DataNodeContainer target) {
+        super(augmentationIdentifierFrom(augmentation), new EffectiveAugmentationSchema(augmentation, target), null);
     }
 
     @Override
@@ -33,11 +34,16 @@ final class AugmentationContextNode extends DataContainerContextNode<Augmentatio
         if (schema instanceof DataSchemaNode && result.isAugmenting()) {
             return fromAugmentation(schema, (AugmentationTarget) schema, result);
         }
-        return fromDataSchemaNode(result);
+        return lenientOf(result);
     }
 
     @Override
     protected Set<QName> getQNameIdentifiers() {
         return getIdentifier().getPossibleChildNames();
     }
+
+    @Override
+    protected void pushToStack(final SchemaInferenceStack stack) {
+        // No-op
+    }
 }