D-Net

package eu.dnetlib.resolver;

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

import eu.dnetlib.dli.resolver.model.serializer.ResolverSerializer;

import eu.dnetlib.resolver.parser.DMFResolverParser;

import eu.dnetlib.resolver.parser.PMFResolverParser;

import eu.dnetlib.resolver.store.ConfigurationResolverStoreTestConfig;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

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;

/**

 * Created by sandro on 10/4/16.

 */

@RunWith(SpringJUnit4ClassRunner.class)

@ContextConfiguration(classes = { ConfigurationTestConfig.class, ConfigurationResolverStoreTestConfig.class })

public class DLIRSerializerTest {

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

	@Autowired

	private ResolverSerializer serializer;

	@Test

	public void testSerializeScholix() {

		log.debug(serializer.serializeToXML(createScholixMock()));

	}

	@Test

	public void testSerializePMF() throws IOException {

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

		final ResolvedObject object = createPMFMock();

		object.setCompletionStatus(CompletionStatus.complete.toString());

		final ObjectProvenance provencance = new ObjectProvenance();

		provencance.setDatasource("Provencance1");

		provencance.setDatasourceId("pid1");

		provencance.setProvisionMode("resolved");

		provencance.setCompletionStatus("complete");

		object.getDatasourceProvenance().add(provencance);

		object.setPid("MOCKPID");

		object.setPidType("MOCKPIDType");

		final String replacingXML = serializer.serializeReplacingXML(IOUtils.toString(recordStream), object);

		log.debug(replacingXML);

	}

	@Test

	public void testSerializeDMF() throws Exception {

		final ResolvedObject object = createDMFMock();

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

		final String replacingXML = serializer.serializeReplacingXML(IOUtils.toString(recordStream), object);

		log.debug(replacingXML);

	}

	private ResolvedObject createPMFMock() throws IOException {

		final InputStream resource = getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputPMF.xml");

		String xml = IOUtils.toString(resource);

		PMFResolverParser parser = new PMFResolverParser();

		return parser.parseObject(xml);

	}

	private ResolvedObject createDMFMock() throws Exception {

		final InputStream xsltAsStream = this.getClass().getResourceAsStream("/eu/dnetlib/dli/transform/oai_datacite_transform.xsl");

		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();

		return parser.parseObject(record);

	}

	private ResolvedObject createScholixMock() {

		final ResolvedObject mock = new ResolvedObject();

		mock.setPid("mock");

		mock.setPidType("mockType");

		final ObjectProvenance provenance = new ObjectProvenance();

		provenance.setDatasource("Mock Datasource");

		mock.setDatasourceProvenance(Arrays.asList(provenance));

		final ObjectRelation relation = new ObjectRelation();

		relation.setTargetPID(new PID("mockTarget", "mockType"));

		relation.setTargetType(ObjectType.unknown);

		relation.setRelationProvenance(Arrays.asList(provenance));

		mock.setRelations(Arrays.asList(relation));

		return mock;

	}

}