BUG-4688: Rework SchemaContext module lookups

[yangtools.git] / yang / yang-model-util / src / test / java / org / opendaylight / yangtools / yang / model / util / SchemaContextProxyTest.java

diff --git a/yang/yang-model-util/src/test/java/org/opendaylight/yangtools/yang/model/util/SchemaContextProxyTest.java b/yang/yang-model-util/src/test/java/org/opendaylight/yangtools/yang/model/util/SchemaContextProxyTest.java
index fe99d93de6d1dbe90ecf3d51ecb7d415c2fc70d0..8cd8bf4d8ca1c6230673fa0e4d34f1a6bd1bcebc 100644 (file)
--- a/yang/yang-model-util/src/test/java/org/opendaylight/yangtools/yang/model/util/SchemaContextProxyTest.java
+++ b/yang/yang-model-util/src/test/java/org/opendaylight/yangtools/yang/model/util/SchemaContextProxyTest.java
@@ -545,14 +545,13 @@ public class SchemaContextProxyTest {
         //asserting collections
         if (expected != null) {
             for (final Module module : expected) {
-                assertEquals(module, filteringSchemaContextProxy.findModuleByName(module.getName(),
-                            module.getRevision()));
+                assertEquals(module, filteringSchemaContextProxy.findModule(module.getName(), module.getRevision())
+                    .get());
 
-                Set<Module> mod = filteringSchemaContextProxy.findModuleByNamespace(module.getNamespace());
+                Set<Module> mod = filteringSchemaContextProxy.findModules(module.getNamespace());
                 assertTrue(mod.contains(module));
-
-                assertEquals(module, filteringSchemaContextProxy.findModuleByNamespaceAndRevision(module.getNamespace(),
-                            module.getRevision()));
+                assertEquals(module, filteringSchemaContextProxy.findModule(module.getNamespace(),
+                    module.getRevision()).get());
             }
         }
     }