D-Net

package eu.dnetlib.resolver;

import java.io.IOException;

import java.io.InputStream;

import java.util.Arrays;

import eu.dnetlib.dli.resolver.NCBINParser;

import eu.dnetlib.dli.resolver.NCBINResolver;

import eu.dnetlib.dli.resolver.OpenAireParser;

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

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

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

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

import eu.dnetlib.resolver.parser.DMFResolverParser;

import eu.dnetlib.resolver.parser.PMFResolverParser;

import eu.dnetlib.resolver.parser.ScholixResolverParser;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.junit.Assert;

import org.junit.Test;

/**

 * Created by sandro on 9/12/16.

 */

public class DLIParserTest {

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

	@Test

	public void testDMFParser() throws IOException {

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

		String str = IOUtils.toString(resourceAsStream);

		DMFResolverParser parser = new DMFResolverParser();

		ResolvedObject object = parser.parseObject(str);

		Assert.assertNotNull(object);

		Assert.assertNotNull(object.getTitles());

		Assert.assertNotNull(object.getType());

		object.setTitles(Arrays.asList("TITOLO 1", "TITOLO 2"));

		ResolvedObject nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputRecordScholix.xml")));

		Assert.assertNull(nullObject);

		nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputPMFRecord.xml")));

		Assert.assertNull(nullObject);

	}

	@Test

	public void testScholixParser() throws IOException {

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

		String str = IOUtils.toString(resourceAsStream);

		ScholixResolverParser parser = new ScholixResolverParser();

		ResolvedObject object = parser.parseObject(str);

		log.debug("object = " + object);

		Assert.assertNotNull(object);

		log.debug("object.getCompletionStatus() = " + object.getCompletionStatus());

		Assert.assertTrue(object.getCompletionStatus().equals(CompletionStatus.incomplete.toString()));

		Assert.assertNotNull(object.getRelations());

		Assert.assertTrue(object.getRelations().size() == 1);

		ResolvedObject nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputRecordDMF.xml")));

		Assert.assertNull(nullObject);

		nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputPMFRecord.xml")));

		Assert.assertNull(nullObject);

	}

	@Test

	public void testPMFParser() throws Exception {

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

		String xml = IOUtils.toString(resource);

		PMFResolverParser parser = new PMFResolverParser();

		final ResolvedObject resolvedObject = parser.parseObject(xml);

		Assert.assertNotNull(resolvedObject);

		log.debug("resolvedObject = " + resolvedObject);

		Assert.assertNotNull(resolvedObject.getRelations());

		resolvedObject.getRelations().forEach(it -> Assert.assertNotNull(it));

		ResolvedObject nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputRecordScholix.xml")));

		Assert.assertNull(nullObject);

		nullObject = parser.parseObject(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dli/parser/InputRecordDMF.xml")));

		Assert.assertNull(nullObject);

	}

	@Test

	public void testOpenAIREParser() throws Exception {

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

		final String record = IOUtils.toString(resource);

		final OpenAireParser parser = new OpenAireParser();

		final ResolvedObject object = parser.parseObject(record);

		object.setPid("od_______908::ff0270cae25acd9367ef26a53ee5e4a3");

		object.setPidType("openaire");

		ObjectProvenance provenance = new ObjectProvenance();

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

		provenance.setDatasourceId("dli::r3d100010778");

		provenance.setDatasource("NCBI Nucleotide");

		provenance.setProvisionMode(ObjectProvisionMode.resolved.toString());

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

		log.debug(object);

	}

	@Test

	public void testProteinParser() throws IOException {

		final NCBINResolver resolver = new NCBINResolver();

		resolver.setNCBINParser(new NCBINParser());

		final ResolvedObject p02768 = resolver.resolve("U36380", "protein");

		log.debug(p02768.getDescription());

		Assert.assertNotNull(p02768);

		Assert.assertNotNull(p02768.getDescription());

		log.debug("p02768 = " + p02768);

	}

}