codec/yang-data-codec-xml/src/test/java/org/opendaylight/yangtools/yang/data/codec/xml/AbstractXmlTest.java

   1 /*
   2  * Copyright (c) 2023 PANTHEON.tech, s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.yang.data.codec.xml;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import java.io.IOException;
  13 import java.io.InputStream;
  14 import java.io.StringWriter;
  15 import javax.xml.transform.OutputKeys;
  16 import javax.xml.transform.Transformer;
  17 import javax.xml.transform.TransformerConfigurationException;
  18 import javax.xml.transform.TransformerException;
  19 import javax.xml.transform.TransformerFactory;
  20 import javax.xml.transform.dom.DOMSource;
  21 import javax.xml.transform.stream.StreamResult;
  22 import org.opendaylight.yangtools.util.xml.UntrustedXML;
  23 import org.opentest4j.AssertionFailedError;
  24 import org.w3c.dom.Document;
  25 import org.w3c.dom.Node;
  26 import org.xml.sax.SAXException;
  27
  28 abstract class AbstractXmlTest {
  29     private static final TransformerFactory TF = TransformerFactory.newInstance();
  30
  31     static final Document loadDocument(final String resourcePath) {
  32         return requireNonNull(readXmlToDocument(Bug5446Test.class.getResourceAsStream(resourcePath)));
  33     }
  34
  35     static final Document readXmlToDocument(final InputStream xmlContent) {
  36         final Document doc;
  37         try {
  38             doc = UntrustedXML.newDocumentBuilder().parse(xmlContent);
  39         } catch (SAXException | IOException e) {
  40             throw new AssertionFailedError(null, e);
  41         }
  42         doc.getDocumentElement().normalize();
  43         return doc;
  44     }
  45
  46     static final String toString(final Node xml) {
  47         final Transformer transformer;
  48         try {
  49             transformer = TF.newTransformer();
  50         } catch (TransformerConfigurationException e) {
  51             throw new AssertionFailedError(null, e);
  52         }
  53         transformer.setOutputProperty(OutputKeys.INDENT, "yes");
  54         transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
  55
  56         final var result = new StreamResult(new StringWriter());
  57         try {
  58             transformer.transform(new DOMSource(xml), result);
  59         } catch (TransformerException e) {
  60             throw new AssertionFailedError(null, e);
  61         }
  62         return result.getWriter().toString();
  63     }
  64 }