io.magentys.jaxb.XmlUtils Maven / Gradle / Ivy

Go to download
package io.magentys.jaxb;

import io.magentys.junit.model.ObjectFactory;
import io.magentys.junit.model.Testsuite;
import org.apache.commons.io.FileUtils;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class XmlUtils {

    private final ObjectFactory factory = new ObjectFactory();

    public List unmarshal(String folderPath) throws Exception {

        File file = new File(folderPath);
        if (file.isFile()) {
            throw new Exception("Provided path is that of a file. Please provide path of the folder containing result xmls.");
        }

        if (file.isDirectory()) {
            List files = (List) FileUtils.listFiles(file, new String[]{"xml"}, true);
            List testsuites = new ArrayList<>();
            for (File resultFile : files) {
                testsuites.add(unmarshal(resultFile));
            }
            return testsuites;
        }

        throw new Exception("Provided path isn't a directory or the path isn't absolute.");
    }

    private Testsuite unmarshal(File file) throws JAXBException {
        JAXBContext context = JAXBContext.newInstance(factory.getClass());
        Unmarshaller unmarshaller = context.createUnmarshaller();

        return (Testsuite) unmarshaller.unmarshal(file);
    }
}