commit 0ab06da133190bc3c90a456099261dfb686e2488
author VinodKumarS-Huawei <vinods.kumar@huawei.com> Sat Aug 20 22:04:59 2016 +0530
committer VinodKumarS-Huawei <vinods.kumar@huawei.com> Sat Aug 20 22:04:59 2016 +0530
tree 7a15dcbc50bd678517cae834eae6482167c48e27
parent 3b4cce06c26d0eaaa0886399827928cfb3fd2c04

YANG Schema node Interface update

Change-Id: Id99029c28f80117015566527faf54e13f4403d2a

plugin/src/test/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListenerTest.java

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
+}