commit 620b6e9d5b23d1a376bc3c111e7aab23960344a8
author Vidyashree Rama <vidyashree.rama@huawei.com> Mon Mar 28 11:59:27 2016 +0530
committer Gerrit Code Review <gerrit@onlab.us> Mon Mar 28 18:18:16 2016 +0000
tree 3fdaa0d12edaa1ac53d4d2010ad561ec51be5394
parent c34b9441e65e170cbdb2dcc2da5ee5effd54515a

yang Notification Listener

Change-Id: I1073be0342bee083495e2c103bad8351f492441c

src/main/java/org/onosproject/yangutils/parser/impl/listeners/ContainerListener.java

diff --git a/src/main/java/org/onosproject/yangutils/parser/impl/listeners/ContainerListener.java b/src/main/java/org/onosproject/yangutils/parser/impl/listeners/ContainerListener.java
index 06dd0d5..1677cf6 100644
--- a/src/main/java/org/onosproject/yangutils/parser/impl/listeners/ContainerListener.java
+++ b/src/main/java/org/onosproject/yangutils/parser/impl/listeners/ContainerListener.java
@@ -21,6 +21,10 @@
 import org.onosproject.yangutils.datamodel.YangList;
 import org.onosproject.yangutils.datamodel.YangModule;
 import org.onosproject.yangutils.datamodel.YangNode;
+import org.onosproject.yangutils.datamodel.YangNotification;
+import org.onosproject.yangutils.datamodel.YangInput;
+import org.onosproject.yangutils.datamodel.YangOutput;
+import org.onosproject.yangutils.datamodel.YangAugment;
 import org.onosproject.yangutils.datamodel.exceptions.DataModelException;
 import org.onosproject.yangutils.parser.Parsable;
 import org.onosproject.yangutils.parser.antlrgencode.GeneratedYangParser;
@@ -127,7 +131,10 @@
 
         Parsable curData = listener.getParsedDataStack().peek();
         if (curData instanceof YangModule || curData instanceof YangContainer
-                || curData instanceof YangList || curData instanceof YangCase) {
+                || curData instanceof YangList || curData instanceof YangCase
+                || curData instanceof YangNotification
+                || curData instanceof YangInput || curData instanceof YangOutput
+                || curData instanceof YangAugment) {
             YangNode curNode = (YangNode) curData;
             try {
                 curNode.addChild(container);