D-Net

package eu.dnetlib.dli.transform;

import eu.dnetlib.data.transform.OntologyLoader;

import eu.dnetlib.dli.DLIUtils;

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

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.resolver.parser.DMFResolverParser;

import eu.dnetlib.rmi.enabling.ISLookUpException;

import eu.dnetlib.rmi.enabling.ISLookUpService;

import org.antlr.stringtemplate.StringTemplate;

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.Assert;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.Mockito;

import org.mockito.runners.MockitoJUnitRunner;

import org.springframework.web.client.RestTemplate;

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;

import java.util.Arrays;

import java.util.List;

import static org.mockito.Mockito.when;

/**

 * Created by sandro on 9/16/16.

 */

@RunWith(MockitoJUnitRunner.class)

public class DLITransformTest {

    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";

    @Mock

    private DLIUtils dliUtilsInstance;

    @Mock

    private UniqueServiceLocator uniqueServiceLocator;

    @Mock

    private ISLookUpService isLookUpServiceMock;

    @Before

    public void initialize() throws ISLookUpException, IOException {

        final String query = "for $x in collection('/db/DRIVER/RepositoryServiceResources/RepositoryServiceResourceType') "

                + "return concat($x//FIELD[./key='NamespacePrefix']/value/text(),'@--@',$x//FIELD[./key='DataSourceId']/value/text(),'@--@',$x//ENGLISH_NAME )";

        dliUtilsInstance = Mockito.mock(DLIUtils.class);

        uniqueServiceLocator = Mockito.mock(UniqueServiceLocator.class);

        isLookUpServiceMock = Mockito.mock(ISLookUpService.class);

        dliUtilsInstance.setServiceLocator(uniqueServiceLocator);

        DLIUtils.setInstance(dliUtilsInstance);

        when(dliUtilsInstance.getServiceLocator()).thenReturn(uniqueServiceLocator);

        when(uniqueServiceLocator.getService(ISLookUpService.class)).thenReturn(isLookUpServiceMock);

        List<String> resutlDS = Arrays.asList(

                "r3d100010255@--@dli_________::r3d100010255@--@icpsr",

                "r3d100010134@--@dli_________::r3d100010134@--@pangaea"

        );

        when(isLookUpServiceMock.quickSearchProfile(query)).thenReturn(resutlDS);

        DLIUtils.ontologies = OntologyLoader.loadOntologiesFromCp();

    }

    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 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/InputRecordDMF.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 ResolvedObject object = parser.parseObject(record);

        Assert.assertNotNull(object.getSubjects());

        Assert.assertTrue(object.getSubjects().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);

    }

}