Expose URLYangTextSource
[yangtools.git] / parser / rfc6643-parser-support / src / test / java / org / opendaylight / yangtools / rfc6643 / parser / IetfYangSmiv2ExtensionPluginTest.java
index 8eb6bb24199562188a9fdae6a10236503680c4b6..fedaab441a79a5bf427bf5891c9b80091687fa3d 100644 (file)
@@ -24,7 +24,7 @@ import org.opendaylight.yangtools.yang.common.Uint32;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
-import org.opendaylight.yangtools.yang.model.spi.source.YangTextSource;
+import org.opendaylight.yangtools.yang.model.spi.source.URLYangTextSource;
 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YangStatementStreamSource;
@@ -54,10 +54,10 @@ class IetfYangSmiv2ExtensionPluginTest {
             .build();
         final var schema = reactor.newBuild()
             .addSources(
-                YangStatementStreamSource.create(YangTextSource.forResource(
-                    IetfYangSmiv2ExtensionPluginTest.class, "/foo.yang")),
-                YangStatementStreamSource.create(YangTextSource.forResource(
-                    IetfYangSmiv2ExtensionPluginTest.class, "/ietf-yang-smiv2.yang")))
+                YangStatementStreamSource.create(new URLYangTextSource(
+                    IetfYangSmiv2ExtensionPluginTest.class.getResource("/foo.yang"))),
+                YangStatementStreamSource.create(new URLYangTextSource(
+                    IetfYangSmiv2ExtensionPluginTest.class.getResource("/ietf-yang-smiv2.yang"))))
             .buildEffective();
 
         assertEquals(1, schema.getUnknownSchemaNodes().size());