D-Net

package eu.dnetlib.msro.openaireplus.api.objects;

import java.io.InputStreamReader;

import java.io.StringReader;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Properties;

import com.google.gson.Gson;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

import org.apache.velocity.app.VelocityEngine;

import org.dom4j.Document;

import org.dom4j.io.OutputFormat;

import org.dom4j.io.SAXReader;

import org.dom4j.io.XMLWriter;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.invocation.InvocationOnMock;

import org.mockito.runners.MockitoJUnitRunner;

import org.mockito.stubbing.Answer;

import static org.mockito.Matchers.anyString;

import static org.mockito.Mockito.when;

/**

 * Created by michele on 14/12/15.

 */

@RunWith(MockitoJUnitRunner.class)

public class ResultEntryTest {

	private VelocityEngine ve;

	@Mock

	private ISLookUpService lookUpService;

	@Before

	public void setUp() throws Exception {

		final Properties props = new Properties();

		props.setProperty("resource.loader", "class");

		props.setProperty("class.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");

		props.setProperty("runtime.log.logsystem.class", "org.apache.velocity.runtime.log.Log4JLogChute");

		ve = new VelocityEngine();

		ve.init(props);

		when(lookUpService.quickSearchProfile(anyString())).thenAnswer(new Answer<Object>() {

			@Override

			public Object answer(final InvocationOnMock invocation) throws Throwable {

				final String query = invocation.getArguments()[0].toString();

				if (query.contains("dnet:result_typologies")) {

					return Arrays.asList("publication @@@ publication", "dataset @@@ dataset");

				} else if (query.contains("dnet:access_modes")) {

					return Arrays.asList("OPEN @@@ Open Access");

				} else if (query.contains("dnet:publication_resource")) {

					return Arrays.asList("0001 @@@ Journal Article");

				} else if (query.contains("dnet:pid_types")) {

					return Arrays.asList("oai @@@ Open Archive Initiative", "doi @@@ Digital object identifier");

				} else if (query.contains("dnet:languages")) {

					return Arrays.asList("ita @@@ Italian");

				} else if (query.contains("ContextDSResourceType")) {

					return Arrays.asList(

							"egi::classification::natsc::math::stats @@@ Statistics and Probability",

							"egi::classification::natsc::math::pure @@@ Pure Mathematics",

							"egi::classification::natsc::math @@@ Mathematics",

							"egi::classification::natsc @@@ Natural Sciences",

							"egi::classification @@@ EGI classification scheme",

							"egi @@@ EGI");

				} else {

					return new ArrayList<String>();

				}

			}

		});

		when(lookUpService.getResourceProfileByQuery(anyString())).thenAnswer(new Answer<Object>() {

			@Override

			public Object answer(final InvocationOnMock invocation) throws Throwable {

				return "REPO NAME @@@ repo________";

			}

		});

	}

	@Test

	public void testAsIndexRecord() throws Exception {

		final ResultEntry pub = new ResultEntry();

		pub.setOriginalId("ORIG_ID_1234");

		pub.setTitle("TEST TITLE <test>");

		pub.getAuthors().add("Michele Artini");

		pub.getAuthors().add("Claudio Atzori");

		pub.getAuthors().add("Alessia Bardi");

		pub.setPublisher("Test publisher");

		pub.setDescription("DESCRIPTION & DESCRIPTION & DESCRIPTION & DESCRIPTION & DESCRIPTION & DESCRIPTION & DESCRIPTION ");

		pub.setLanguage("ita");

		pub.setLicenseCode("OPEN");

		pub.setResourceType("0001");

		pub.setUrl("http://test.it/a=1&b=2");

		pub.getPids().add(new PidEntry("doi", "10.000/xyz-123"));

		pub.getPids().add(new PidEntry("oai", "oai:1234"));

		pub.setCollectedFromId("test________::zenodo");

		pub.setHostedById("test________::UNKNOWN");

		pub.getLinksToProjects().add("info:eu-repo/grantAgreement/EC/FP7/283595/EU//OpenAIREplus");

		pub.getLinksToProjects().add("info:eu-repo/grantAgreement/EC/FP7/244909/EU/Making Capabilities Work/WorkAble");

		pub.getContexts().add("egi::classification::natsc::math::pure");

		pub.getContexts().add("egi::classification::natsc::math::stats");

		final String xml = pub.asOafRecord(ve, lookUpService, "http://oaf/oaf.xsd");

		final Document doc = (new SAXReader()).read(new StringReader(xml));

		final OutputFormat format = OutputFormat.createPrettyPrint();

		final XMLWriter writer = new XMLWriter(System.out, format);

		writer.write(doc);

		/* writer.close(); */

	}

	@Test

	public void testAsIndexRecord_json() throws Exception {

		final ResultEntry pub = (new Gson()).fromJson(new InputStreamReader(getClass().getResourceAsStream("test_record.json")), ResultEntry.class);

		final String xml = pub.asOafRecord(ve, lookUpService, "http://oaf/oaf.xsd");

		final Document doc = (new SAXReader()).read(new StringReader(xml));

		final OutputFormat format = OutputFormat.createPrettyPrint();

		final XMLWriter writer = new XMLWriter(System.out, format);

		writer.write(doc);

		/* writer.close(); */

	}

	@Test

	public void testAsIndexRecord_json_with_greek_chars() throws Exception {

		final ResultEntry pub =

				(new Gson()).fromJson(new InputStreamReader(getClass().getResourceAsStream("test_record_with_greek_chars.json")), ResultEntry.class);

		final String xml = pub.asOafRecord(ve, lookUpService, "http://oaf/oaf.xsd");

		final Document doc = (new SAXReader()).read(new StringReader(xml));

		final OutputFormat format = OutputFormat.createPrettyPrint();

		final XMLWriter writer = new XMLWriter(System.out, format);

		writer.write(doc);

		/* writer.close(); */

	}

}