objective.taskboard.utils.XmlUtilsTest.java Source code

Introduction

Here is the source code for objective.taskboard.utils.XmlUtilsTest.java

Source

package objective.taskboard.utils;

import static org.hamcrest.Matchers.greaterThan;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertThat;

import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.nio.file.Files;

import javax.xml.transform.OutputKeys;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
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;

/*-
 * [LICENSE]
 * Taskboard
 * - - -
 * Copyright (C) 2015 - 2016 Objective Solutions
 * - - -
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * [/LICENSE]
 */

import org.apache.commons.io.FileUtils;
import org.junit.Test;
import org.w3c.dom.Document;
import org.w3c.dom.NodeList;

public class XmlUtilsTest {

    @Test
    public void xpathToAttributes() throws URISyntaxException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element/@attribute";

        NodeList nodeList = XmlUtils.xpath(xmlFile, locator);
        assertThat(nodeList.getLength(), is(2));
        assertThat(nodeList.item(0).getNodeValue(), is("value0"));
        assertThat(nodeList.item(1).getNodeValue(), is("value1"));
    }

    @Test
    public void xpathToNode() throws URISyntaxException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element[@attribute='value0']";

        NodeList nodeList = XmlUtils.xpath(xmlFile, locator);
        assertThat(nodeList.getLength(), greaterThan(0));
        assertThat(nodeList.item(0).getAttributes().getNamedItem("attribute").getNodeValue(), is("value0"));
    }

    @Test
    public void formatXml() throws URISyntaxException, IOException, TransformerException {
        File outputXmlFile = Files.createTempFile("formatted", "xml").toFile();
        try {
            File inputXmlFile = new File(XmlUtilsTest.class.getResource("unformattedFile.xml").toURI());
            File expectedResult = new File(XmlUtilsTest.class.getResource("file.xml").toURI());

            format(inputXmlFile, outputXmlFile);

            assertEquals("The files differ!", FileUtils.readFileToString(expectedResult, "utf-8"),
                    FileUtils.readFileToString(outputXmlFile, "utf-8"));
        } finally {
            FileUtils.deleteQuietly(outputXmlFile);
        }
    }

    @Test
    public void asStringAttribute() throws URISyntaxException, TransformerException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element[@attribute='value0']/@attribute";

        assertThat(XmlUtils.asString(XmlUtils.xpath(xmlFile, locator)), is("value0"));
    }

    @Test
    public void asStringAttributes() throws URISyntaxException, TransformerException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element/@attribute";

        assertThat(XmlUtils.asString(XmlUtils.xpath(xmlFile, locator)), is("value0\nvalue1"));
    }

    @Test
    public void asStringNode() throws URISyntaxException, TransformerException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element[@attribute='value0']";

        assertThat(XmlUtils.asString(XmlUtils.xpath(xmlFile, locator)), is("<element attribute=\"value0\"/>"));
    }

    @Test
    public void asStringNodes() throws URISyntaxException, TransformerException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element";

        assertThat(XmlUtils.asString(XmlUtils.xpath(xmlFile, locator)),
                is("<element attribute=\"value0\"/>\n<element attribute=\"value1\">text-content</element>"));
    }

    @Test
    public void asStringText() throws URISyntaxException, TransformerException {
        File xmlFile = new File(XmlUtilsTest.class.getResource("file.xml").toURI());
        String locator = "//element[@attribute='value1']/text()";

        assertThat(XmlUtils.asString(XmlUtils.xpath(xmlFile, locator)), is("text-content"));
    }

    private static void format(File inputXmlFile, File outputXmlFile) {
        try {
            Document document = XmlUtils.asDocument(inputXmlFile);
            TransformerFactory transFactory = TransformerFactory.newInstance();
            Transformer idTransform = transFactory.newTransformer();
            idTransform.setOutputProperty(OutputKeys.METHOD, "xml");
            idTransform.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
            idTransform.setOutputProperty(OutputKeys.INDENT, "yes");
            idTransform.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "2");
            Source input = new DOMSource(document);
            Result output = new StreamResult(outputXmlFile);
            idTransform.transform(input, output);
        } catch (TransformerException e) {
            throw new RuntimeException(e);
        }
    }
}