Remove TestUtils.findModule()

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

diff --git a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileAugmentStmtTest.java b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileAugmentStmtTest.java
index f6abe8edc7783ab09b0a7c510f90169b0b4c6aa4..7dded3c98a8de6793efedfd932cdb6174fe71b4e 100644 (file)
--- a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileAugmentStmtTest.java
+++ b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileAugmentStmtTest.java
@@ -21,13 +21,11 @@ import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.stmt.TestUtils;
 
 public class YinFileAugmentStmtTest extends AbstractYinModulesTest {
-
     @Test
     public void testAugment() {
-        final Module testModule = TestUtils.findModule(context, "main-impl").get();
+        final Module testModule = context.findModules("main-impl").iterator().next();
         assertNotNull(testModule);
 
         final Collection<? extends AugmentationSchemaNode> augmentations = testModule.getAugmentations();