BUG-4688: Rework SchemaContext module lookups

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug7440Test.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug7440Test.java
index 34221ace427b76def7628243fafaa36751de3501..70fc2d67e9a1d7650219892584c78a1301943b83 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug7440Test.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug7440Test.java
@@ -31,11 +31,8 @@ public class Bug7440Test {
         assertNotNull(schemaContext);
 
         final Date revision = QName.parseRevision("2016-12-23");
-
-        final Module foo = schemaContext.findModuleByName("foo", revision);
-        assertNotNull(foo);
-        final Module bar = schemaContext.findModuleByName("bar", revision);
-        assertNotNull(bar);
+        final Module foo = schemaContext.findModule("foo", revision).get();
+        final Module bar = schemaContext.findModule("bar", revision).get();
 
         final Set<Deviation> deviations = foo.getDeviations();
         assertEquals(1, deviations.size());