Remove TestUtils.findModule()

[yangtools.git] / parser / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / YangParserTest.java

diff --git a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
index 5135b130992a5826848c4fceeb42d558d4cd37c9..b59425acb88dc55d3e696f89ed8ff92e815da653 100644 (file)
--- a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
+++ b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
@@ -86,9 +86,9 @@ public class YangParserTest {
     @Before
     public void init() throws Exception {
         context = TestUtils.loadModules(getClass().getResource("/model").toURI());
-        foo = TestUtils.findModule(context, "foo").get();
-        bar = TestUtils.findModule(context, "bar").get();
-        baz = TestUtils.findModule(context, "baz").get();
+        foo = context.findModules("foo").iterator().next();
+        bar = context.findModules("bar").iterator().next();
+        baz = context.findModules("baz").iterator().next();
     }
 
     @Test