BUG-4688: Rework SchemaContext module lookups

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9005Test.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9005Test.java
index 852c4bf4e535e6e5f913241c570e37859e693916..15b3f3e5ca809f95f04bb154325c2b82ef2294d4 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9005Test.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug9005Test.java
@@ -23,7 +23,7 @@ public class Bug9005Test {
         final SchemaContext context = StmtTestUtils.parseYangSources("/bugs/bug9005");
         assertNotNull(context);
 
-        final Module foo = context.findModuleByName("foo", QName.parseRevision("2017-07-07"));
+        final Module foo = context.findModule("foo", QName.parseRevision("2017-07-07")).get();
 
         final Set<ModuleImport> imports = foo.getImports();
         assertEquals(1, imports.size());