D-Net

package eu.dnetlib.parthenos.registry;

import java.io.IOException;

import java.io.InputStream;

import com.google.common.collect.Iterables;

import eu.dnetlib.parthenos.CRMpe;

import org.apache.commons.io.IOUtils;

import org.apache.jena.rdf.model.*;

import org.apache.jena.vocabulary.RDF;

import org.junit.Assert;

import org.junit.Test;

import org.springframework.core.io.ClassPathResource;

/**

 * Created by Alessia Bardi on 28/09/2017.

 *

 * @author Alessia Bardi

 */

public class GCubeResourceGeneratorTest {

	private static String recordPath = "eu/dnetlib/parthenos/registry/sample1.rdf";

	private static String baseURI = "http://parthenos.d4science.org/CRMext/CRMpe.rdfs/";

	private static GCubeResourceRegistrator generator = new GCubeResourceRegistrator();

	private static InfModel baseModel;

	static{

		generator.setFacetWriter(new FacetWriter());

		baseModel = generator.loadBaseModel();

		try {

			baseModel.read(getStream(recordPath), baseURI);

			baseModel.read(getStream("eu/dnetlib/parthenos/registry/test.rdf"), baseURI);

		} catch (IOException e) {

			e.printStackTrace();

		}

	}

	@Test

	public void testLoadBaseModel() {

		GCubeResourceRegistrator generator1 = new GCubeResourceRegistrator();

		InfModel model = generator1.loadBaseModel();

		//let's snapshot to a normal model to write all statements

		Model snapshot = ModelFactory.createDefaultModel();

		snapshot.add(model);

	}

	@Test

	public void testLoadRDFFileWithModel() throws IOException {

		//let's snapshot to a normal model to write all statements

		Model snapshot = ModelFactory.createDefaultModel();

		snapshot.add(baseModel);

		snapshot.write(System.out);

	}

	@Test

	public void testRules() throws IOException {

		StmtIterator it = baseModel.listStatements(new SimpleSelector(ResourceFactory.createResource("uuid:AAAH"), RDF.type, (Resource) null));

		while(it.hasNext()){

			System.out.println(it.nextStatement().toString());

		}

		Resource classHS = ResourceFactory.createResource(CRMpe.NS+"PE1_Service");

		StmtIterator it2 = baseModel.listStatements(classHS, RDF.type, (Resource) null);

		while(it2.hasNext()){

			System.out.println(it2.nextStatement().toString());

		}

	}

	@Test

	public void testAmbiguousSpecificType() throws IOException {

		Resource res = baseModel.getResource("uuid:AAAH");

		Resource specificType = generator.findSpecificType(res, CRMpe.PE1_Service);

		System.out.println(specificType.getURI());

		Assert.assertEquals(CRMpe.PE1_Service.getURI(), specificType.getURI());

	}

	@Test

	public void testJsonGeneration1() throws IOException {

		testJsonGeneration("eu/dnetlib/parthenos/registry/test.rdf", 0);

	}

	@Test

	public void testJsonGeneration2() throws IOException {

		testJsonGeneration(recordPath, 0);

	}

	private void testJsonGeneration(final String recordClasspath, final int minCount) throws IOException {

		Iterable<String> it = generator.getGCubeER(getString(recordClasspath));

		int count = Iterables.size(it);

		System.out.println(it);

		System.out.println(count);

		Assert.assertTrue(count > minCount);

	}

	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 static InputStream getStream(final String classpath) throws IOException {

		return new ClassPathResource(classpath).getInputStream();

	}

}