BUG-4688: Rework SchemaContext module lookups

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/GroupingAndUsesStmtTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/GroupingAndUsesStmtTest.java
index ea23c159de546795d5a341fa8780c7dc3e86128e..8492782a584698b0883b80b4a3744acc6a1441d3 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/GroupingAndUsesStmtTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/GroupingAndUsesStmtTest.java
@@ -53,7 +53,7 @@ public class GroupingAndUsesStmtTest {
         final SchemaContext result = reactor.buildEffective();
         assertNotNull(result);
 
-        final Module testModule = result.findModuleByName("baz", null);
+        final Module testModule = result.findModules("baz").iterator().next();
         assertNotNull(testModule);
 
         final Set<GroupingDefinition> groupings = testModule.getGroupings();
@@ -95,7 +95,7 @@ public class GroupingAndUsesStmtTest {
         final SchemaContext result = reactor.buildEffective();
         assertNotNull(result);
 
-        final Module testModule = result.findModuleByName("foo", null);
+        final Module testModule = result.findModules("foo").iterator().next();
         assertNotNull(testModule);
 
         final Set<UsesNode> usesNodes = testModule.getUses();