YANG Schema node Interface update
Change-Id: Id99029c28f80117015566527faf54e13f4403d2a
diff --git a/plugin/src/test/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListenerTest.java b/plugin/src/test/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListenerTest.java
index ee418c2..fbe961b 100644
--- a/plugin/src/test/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListenerTest.java
+++ b/plugin/src/test/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListenerTest.java
@@ -20,6 +20,7 @@
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.onosproject.yangutils.datamodel.YangModule;
+import org.onosproject.yangutils.datamodel.javadatamodel.YangJavaModule;
import org.onosproject.yangutils.parser.antlrgencode.GeneratedYangParser;
import org.onosproject.yangutils.parser.exceptions.ParserException;
import org.onosproject.yangutils.parser.impl.TreeWalkListener;
@@ -41,7 +42,7 @@
thrown.expect(ParserException.class);
thrown.expectMessage("Internal parser error detected: Invalid holder for yangbase before processing.");
- YangModule tmpModule = new YangModule();
+ YangModule tmpModule = new YangJavaModule();
TreeWalkListener listener = new TreeWalkListener();
listener.getParsedDataStack().push(tmpModule);
GeneratedYangParser.YangfileContext ctx = null;
@@ -95,12 +96,12 @@
thrown.expect(ParserException.class);
thrown.expectMessage("Internal parser error detected: Invalid holder for yangbase after processing.");
- YangModule tmpModule = new YangModule();
- YangModule tmpModule2 = new YangModule();
+ YangModule tmpModule = new YangJavaModule();
+ YangModule tmpModule2 = new YangJavaModule();
TreeWalkListener listener = new TreeWalkListener();
listener.getParsedDataStack().push(tmpModule);
listener.getParsedDataStack().push(tmpModule2);
GeneratedYangParser.YangfileContext ctx = null;
BaseFileListener.processYangFileExit(listener, ctx);
}
-}
\ No newline at end of file
+}