BUG-4688: Rework SchemaContext module lookups

[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / CaseStmtTest.java

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/CaseStmtTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/CaseStmtTest.java
index 23ad0d971024b289335e067d7c666bde87bfa2f8..cd47bf687f929c26f08f74d5165cada4a4653bfa 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/CaseStmtTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/CaseStmtTest.java
@@ -46,8 +46,8 @@ public class CaseStmtTest {
         String date_s = "2015-09-09 00:00:00.0";
         SimpleDateFormat dt = SimpleDateFormatUtil.getRevisionFormat();
         Date date = dt.parse(date_s);
-        rootFoo = schema.findModuleByName("foo", date);
-        rootBar = schema.findModuleByName("bar", date);
+        rootFoo = schema.findModule("foo", date).get();
+        rootBar = schema.findModule("bar", date).get();
         assertNotNull(rootFoo);
         assertNotNull(rootBar);
         qnameFoo = QNameModule.create(URI.create("foo"), date);