AriadnePlus/dnet-ariadneplus/src/test/java/eu/dnetlib/x3m/ApplyX3MappingTest.java

package eu.dnetlib.x3m;

import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;

import eu.dnetlib.msro.workflows.nodes.transform.ApplyX3Mapping;
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.springframework.core.io.ClassPathResource;

public class ApplyX3MappingTest {

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

	private static String basePathProfiles = "/eu/dnetlib/bootstrap/profiles/TransformationRuleDSResources/TransformationRuleDSResourceType/";

	private String mappingUrl = "https://mapping.d4science.org/3MEditor/Services?id=575&output=text/xml&method=export";

	//Ariadne policy
	final String ariadnePolicyPath = "ARIADNE_Policy.xml";

	//ADS record
	final String adsRecordPath = "/eu/dnetlib/x3m/ads-record.xml";

	@Test
	public void testAriadne() {
		doBasicTest(ariadnePolicyPath, adsRecordPath);
	}

	public void doBasicTest(String policyPath, String recordPath){
		String g = loadFromTransformationProfile(policyPath);
		String r = getString(recordPath);
		
		URL mappingURL = null;
		try {
			mappingURL = new URL(mappingUrl);
		} catch (MalformedURLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		ApplyX3Mapping x3m = new ApplyX3Mapping(mappingURL, g, true);
		String res = x3m.apply(r);
		System.out.println("res is \n"+res);
	}


	private String getString(final String classpath) {
		try {
			final ClassPathResource resource = new ClassPathResource(classpath);
			return IOUtils.toString(resource.getInputStream(), "UTF-8");
		}catch(IOException e){
			return null;
		}
	}


	private String loadFromTransformationProfile(String profilePath){
		log.info("Loading xslt from: " + basePathProfiles + profilePath);
		InputStream profile = getClass().getResourceAsStream(basePathProfiles + profilePath);
		SAXReader saxReader = new SAXReader();
		Document doc = null;
		try {
			doc = saxReader.read(profile);
		} catch (DocumentException e) {
			e.printStackTrace();
			throw new RuntimeException(e);
		}
		String s = doc.selectSingleNode("//SCRIPT/CODE").getText();
		//log.debug(profilePath+": "+s);
		return s;
	}
}