BUG-4688: Rework SchemaContext module lookups

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9242Test.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9242Test.java
index 94cc63d7c694609ef2aef3b0735280bcd57f0241..ec7158245a1acb6b8142e9ec2653d98183acf2d4 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9242Test.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9242Test.java
@@ -29,9 +29,8 @@ public class Bug9242Test {
 
         final Date revision = QName.parseRevision("2017-10-13");
 
-        final Module rootModule = schemaContext.findModuleByName("root-module", revision);
-        final Module impModule = schemaContext.findModuleByName("imp-module", revision);
-        assertNotNull(impModule);
+        final Module rootModule = schemaContext.findModule("root-module", revision).get();
+        final Module impModule = schemaContext.findModule("imp-module", revision).get();
 
         TypeDefinition<?> deviatedMyLeafType = null;
         TypeDefinition<?> deviatedMyLeaf2Type = null;