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
  32     static final Document loadDocument(final String resourcePath) {
  33         return requireNonNull(readXmlToDocument(Bug5446Test.class.getResourceAsStream(resourcePath)));
  34     }
  35
  36     static final Document readXmlToDocument(final InputStream xmlContent) {
  37         final Document doc;
  38         try {
  39             doc = UntrustedXML.newDocumentBuilder().parse(xmlContent);
  40         } catch (SAXException | IOException e) {
  41             throw new AssertionFailedError(null, e);
  42         }
  43         doc.getDocumentElement().normalize();
  44         return doc;
  45     }
  46
  47     static final String toString(final Node xml) {
  48         final Transformer transformer;
  49         try {
  50             transformer = TF.newTransformer();
  51         } catch (TransformerConfigurationException e) {
  52             throw new AssertionFailedError(null, e);
  53         }
  54         transformer.setOutputProperty(OutputKeys.INDENT, "yes");
  55         transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
  56
  57         final var result = new StreamResult(new StringWriter());
  58         try {
  59             transformer.transform(new DOMSource(xml), result);
  60         } catch (TransformerException e) {
  61             throw new AssertionFailedError(null, e);
  62         }
  63         return result.getWriter().toString();
  64     }
  65 }