D-Net

package eu.dnetlib.dli.transform;

import eu.dnetlib.dli.resolver.model.DLIResolvedObject;

import eu.dnetlib.resolver.DLIAbstractTest;

import eu.dnetlib.resolver.parser.DMFResolverParser;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.io.SAXReader;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.runners.MockitoJUnitRunner;

import javax.xml.transform.OutputKeys;

import javax.xml.transform.Transformer;

import javax.xml.transform.TransformerFactory;

import javax.xml.transform.stream.StreamResult;

import javax.xml.transform.stream.StreamSource;

import java.io.IOException;

import java.io.InputStream;

import java.io.StringWriter;

/**

 * Created by sandro on 9/16/16.

 */

@RunWith(MockitoJUnitRunner.class)

public class DLITransformTest extends DLIAbstractTest {

    private static final Log log = LogFactory.getLog(DLITransformTest.class);

    private static final String BASE_URL = "http://%s:9200/%s?pretty";

    private static final String BASE_CFG_URL = "http://%s:9200/%s?pretty";

    private InputStream loadFromTransformationProfile(final String profilePath) throws IOException {

        InputStream profile = getClass().getResourceAsStream(profilePath);

        SAXReader saxReader = new SAXReader();

        Document doc = null;

        try {

            doc = saxReader.read(profile);

        } catch (DocumentException e) {

            e.printStackTrace();

            throw new RuntimeException(e);

        }

        String xslt = doc.selectSingleNode("//SCRIPT/CODE/text()").asXML();

        return IOUtils.toInputStream(xslt.replace("<![CDATA[", "").replace("]]>", ""));

    }

    @Test

    public void testRCSBTransform() throws Exception {

        final InputStream xsltAsStream =

                loadFromTransformationProfile("/eu/dnetlib/bootstrap/profiles/TransformationRuleDSResources/rcsb_transform.xml");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/transform/rcsb_input.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void testANDSTransform() throws Exception {

        final InputStream xsltAsStream =

                getClass().getResourceAsStream("/eu/dnetlib/dli/transform/ands.xslt");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/inputANDS.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void test3TUTransform() throws Exception {

        final InputStream xsltAsStream =

                getClass().getResourceAsStream("/eu/dnetlib/dli/transform/3tu.xslt");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/3tu.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void testIEEETransform() throws Exception {

        final InputStream xsltAsStream =

                getClass().getResourceAsStream("/eu/dnetlib/dli/transform/ieee.xslt");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/ieee.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void testIEDATransform() throws Exception {

        final InputStream xsltAsStream =

                getClass().getResourceAsStream("/eu/dnetlib/dli/transform/ieda.xslt");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/ieda.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void testNPGTransform() throws Exception {

        final InputStream xsltAsStream =

                getClass().getResourceAsStream("/eu/dnetlib/dli/transform/dli.xslt");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/npg_incomplete.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        System.out.println(output.toString());

    }

    @Test

    public void testOAI2DMFTransform() throws Exception {

        final InputStream xsltAsStream =

                loadFromTransformationProfile("/eu/dnetlib/bootstrap/profiles/TransformationRuleDSResources/oai_datacite_transform.xml");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/inputDatacite.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        final String record = output.toString();

        final DMFResolverParser parser = new DMFResolverParser();

        log.debug(record);

        final DLIResolvedObject object = parser.parseObject(record);

        //Assert.assertNotNull(object.getSubjects());

//        Assert.assertTrue(object.getSubjects().size() > 0);

//

//

//

//        Assert.assertTrue(object.getRelations().size() > 0);

//

//        object.getRelations().forEach(it -> log.debug(it.getTargetPID()));

//

//        log.debug(object);

//

//        StringTemplate st = new StringTemplate();

//

//        final String template = IOUtils.toString(this.getClass().getResourceAsStream("/eu/dnetlib/dli/templates/DMFXML.st"));

//

//        st.setTemplate(template);

//

//        st.setAttribute("object", object);

//

//        System.out.println(st.toString());

    }

    @Test

    public void testICPSRTransform() throws Exception {

        final InputStream xsltAsStream =

                loadFromTransformationProfile("/eu/dnetlib/bootstrap/profiles/TransformationRuleDSResources/oai_icpsr_transform.xml");

        final TransformerFactory factory = TransformerFactory.newInstance();

        factory.newTransformer();

        final StreamSource xsltSource = new StreamSource(xsltAsStream);

        final InputStream recordStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputICPSR.xml");

        final Transformer transformer = factory.newTransformer(xsltSource);

        transformer.setOutputProperty(OutputKeys.INDENT, "yes");

        final StringWriter output = new StringWriter();

        transformer.transform(new StreamSource(recordStream), new StreamResult(output));

        final String record = output.toString();

        System.out.println(record);

    }

}