Do not use NamedInputStream
[netconf.git] / restconf / sal-rest-connector / src / test / java / org / opendaylight / controller / sal / restconf / impl / test / TestUtils.java
index 23c18f18f581cbad24e9529dff903b159fed6f43..2c2e1f4d593716de0ab1ce35c9d69ae4ac119205 100644 (file)
@@ -8,6 +8,7 @@
 package org.opendaylight.controller.sal.restconf.impl.test;
 
 import static org.junit.Assert.assertNotNull;
+
 import com.google.common.base.Preconditions;
 import java.io.BufferedReader;
 import java.io.ByteArrayOutputStream;
@@ -19,24 +20,26 @@ import java.io.InputStream;
 import java.io.OutputStreamWriter;
 import java.net.URI;
 import java.net.URISyntaxException;
+import java.nio.charset.StandardCharsets;
 import java.sql.Date;
 import java.text.ParseException;
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
-import javax.xml.parsers.DocumentBuilder;
-import javax.xml.parsers.DocumentBuilderFactory;
-import javax.xml.parsers.ParserConfigurationException;
 import javax.xml.transform.OutputKeys;
 import javax.xml.transform.Transformer;
 import javax.xml.transform.TransformerException;
 import javax.xml.transform.TransformerFactory;
 import javax.xml.transform.dom.DOMSource;
 import javax.xml.transform.stream.StreamResult;
+import org.opendaylight.yangtools.util.xml.UntrustedXML;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
@@ -46,10 +49,7 @@ import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.Module;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
-import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
-import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
-import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
-import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
+import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.w3c.dom.Document;
@@ -61,25 +61,24 @@ public final class TestUtils {
 
     public static SchemaContext loadSchemaContext(final String... yangPath)
             throws FileNotFoundException, ReactorException {
-        final CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR.newBuild();
-
-        for (int i = 0; i < yangPath.length; i++) {
-            final String path = yangPath[i];
+        final List<File> files = new ArrayList<>();
+        for (final String path : yangPath) {
             final String pathToFile = TestUtils.class.getResource(path).getPath();
             final File testDir = new File(pathToFile);
             final String[] fileList = testDir.list();
             if (fileList == null) {
                 throw new FileNotFoundException(pathToFile);
             }
-            for (int j = 0; j < fileList.length; j++) {
-                final String fileName = fileList[j];
+
+            for (final String fileName : fileList) {
                 final File file = new File(testDir, fileName);
                 if (file.isDirectory() == false) {
-                    reactor.addSource(new YangStatementSourceImpl(new NamedFileInputStream(file, file.getPath())));
+                    files.add(file);
                 }
             }
         }
-        return reactor.buildEffective();
+
+        return YangParserTestUtils.parseYangSources(files);
     }
 
     public static Module findModule(final Set<Module> modules, final String moduleName) {
@@ -93,10 +92,8 @@ public final class TestUtils {
 
     public static Document loadDocumentFrom(final InputStream inputStream) {
         try {
-            final DocumentBuilderFactory dbfac = DocumentBuilderFactory.newInstance();
-            final DocumentBuilder docBuilder = dbfac.newDocumentBuilder();
-            return docBuilder.parse(inputStream);
-        } catch (SAXException | IOException | ParserConfigurationException e) {
+            return UntrustedXML.newDocumentBuilder().parse(inputStream);
+        } catch (SAXException | IOException e) {
             LOG.error("Error during loading Document from XML", e);
             return null;
         }
@@ -114,10 +111,11 @@ public final class TestUtils {
             transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
             transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
 
-            transformer.transform(new DOMSource(doc), new StreamResult(new OutputStreamWriter(out, "UTF-8")));
+            transformer.transform(new DOMSource(doc), new StreamResult(new OutputStreamWriter(out,
+                StandardCharsets.UTF_8)));
             final byte[] charData = out.toByteArray();
-            return new String(charData, "UTF-8");
-        } catch (IOException | TransformerException e) {
+            return new String(charData, StandardCharsets.UTF_8);
+        } catch (final TransformerException e) {
             final String msg = "Error during transformation of Document into String";
             LOG.error(msg, e);
             return msg;
@@ -210,43 +208,40 @@ public final class TestUtils {
         return matcher.matches();
     }
 
-    public static YangInstanceIdentifier.NodeIdentifier getNodeIdentifier(final String localName, final String namespace,
+    public static NodeIdentifier getNodeIdentifier(final String localName, final String namespace,
             final String revision) throws ParseException {
-        return new YangInstanceIdentifier.NodeIdentifier(QName.create(namespace, revision, localName));
+        return new NodeIdentifier(QName.create(namespace, revision, localName));
     }
 
-    public static YangInstanceIdentifier.NodeIdentifierWithPredicates getNodeIdentifierPredicate(final String localName,
+    public static NodeIdentifierWithPredicates getNodeIdentifierPredicate(final String localName,
             final String namespace, final String revision, final Map<String, Object> keys) throws ParseException {
         final Map<QName, Object> predicate = new HashMap<>();
         for (final String key : keys.keySet()) {
             predicate.put(QName.create(namespace, revision, key), keys.get(key));
         }
 
-        return new YangInstanceIdentifier.NodeIdentifierWithPredicates(
-
-        QName.create(namespace, revision, localName), predicate);
+        return new NodeIdentifierWithPredicates(QName.create(namespace, revision, localName), predicate);
     }
 
-    public static YangInstanceIdentifier.NodeIdentifierWithPredicates getNodeIdentifierPredicate(final String localName,
+    public static NodeIdentifierWithPredicates getNodeIdentifierPredicate(final String localName,
             final String namespace, final String revision, final String... keysAndValues) throws ParseException {
-        if ((keysAndValues.length % 2) != 0) {
-            new IllegalArgumentException("number of keys argument have to be divisible by 2 (map)");
-        }
+        Preconditions.checkArgument(keysAndValues.length % 2 == 0,
+                "number of keys argument have to be divisible by 2 (map)");
         final Map<QName, Object> predicate = new HashMap<>();
 
-        int i = 0;
-        while (i < keysAndValues.length) {
-            predicate.put(QName.create(namespace, revision, keysAndValues[i++]), keysAndValues[i++]);
+        int index = 0;
+        while (index < keysAndValues.length) {
+            predicate.put(QName.create(namespace, revision, keysAndValues[index++]), keysAndValues[index++]);
         }
 
-        return new YangInstanceIdentifier.NodeIdentifierWithPredicates(QName.create(namespace, revision, localName),
-                predicate);
+        return new NodeIdentifierWithPredicates(QName.create(namespace, revision, localName), predicate);
     }
 
     static NormalizedNode<?,?> prepareNormalizedNodeWithIetfInterfacesInterfacesData() throws ParseException {
         final String ietfInterfacesDate = "2013-07-04";
         final String namespace = "urn:ietf:params:xml:ns:yang:ietf-interfaces";
-        final DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> mapEntryNode = ImmutableMapEntryNodeBuilder.create();
+        final DataContainerNodeAttrBuilder<NodeIdentifierWithPredicates, MapEntryNode> mapEntryNode =
+                ImmutableMapEntryNodeBuilder.create();
 
         final Map<String, Object> predicates = new HashMap<>();
         predicates.put("name", "eth0");