PlainSparqlQueryExecutorTest.java

package com.google.refine.test.org.deri.reconcile.executors;

import static org.testng.Assert.assertTrue;
import static org.testng.Assert.fail;

import java.io.InputStream;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

import com.google.common.collect.ImmutableList;
import com.google.refine.org.deri.reconcile.model.ReconciliationRequest;
import com.google.refine.org.deri.reconcile.model.ReconciliationRequestContext;
import com.google.refine.org.deri.reconcile.model.ReconciliationRequestContext.IdentifiedValueContext;
import com.google.refine.org.deri.reconcile.model.ReconciliationRequestContext.PropertyContext;
import com.google.refine.org.deri.reconcile.model.ReconciliationRequestContext.TextualValueContext;
import com.google.refine.org.deri.reconcile.rdf.executors.DumpQueryExecutor;
import com.google.refine.org.deri.reconcile.rdf.executors.QueryExecutor;
import com.google.refine.org.deri.reconcile.rdf.factories.PlainSparqlQueryFactory;
import com.hp.hpl.jena.query.QuerySolution;
import com.hp.hpl.jena.query.ResultSet;
import com.hp.hpl.jena.rdf.model.Model;
import com.hp.hpl.jena.rdf.model.ModelFactory;
import com.hp.hpl.jena.rdf.model.Resource;

public class PlainSparqlQueryExecutorTest {

	QueryExecutor executor;
	PlainSparqlQueryFactory factory;
	
	//query
	int limit =8;
	String queryString = "godfather";
	String exactMatchQueryString = "Godfather";
	ImmutableList<String> searchPropertyUris = ImmutableList.of("http://www.w3.org/2000/01/rdf-schema#label",
														"http://www.w3.org/2004/02/skos/core#prefLabel");
	
	@BeforeClass
	public void setUp(){
		Model m = ModelFactory.createDefaultModel();
		InputStream in = this.getClass().getResourceAsStream("../files/films.ttl");
		m.read(in,null,"TTL");
		
		executor = new DumpQueryExecutor(m);
		factory = new PlainSparqlQueryFactory();
	}
	
	@Test
	public void executeSimpleReconciliationQuery(){
		ReconciliationRequest request = new ReconciliationRequest(queryString, limit);
		String sparql = factory.getReconciliationSparqlQuery(request, searchPropertyUris);
		ResultSet resultset = executor.sparql(sparql);
		assertResult(null,resultset,"http://data.linkedmdb.org/resource/film/930","http://data.linkedmdb.org/resource/film/329",
		"http://data.linkedmdb.org/resource/film_series/261");
	}
	
	@Test
	public void executeOneLabelOneTypeReconciliationQuery(){
		ReconciliationRequest request = new ReconciliationRequest(queryString, limit);
		request.setTypes(new String[] {"http://data.linkedmdb.org/resource/movie/film"});
		String sparql = factory.getReconciliationSparqlQuery(request, ImmutableList.of("http://www.w3.org/2000/01/rdf-schema#label"));
		ResultSet resultset = executor.sparql(sparql);
		assertResult(null,resultset,"http://data.linkedmdb.org/resource/film/930","http://data.linkedmdb.org/resource/film/329");
	}
	
	@Test
	public void executeReconciliationQueryWithType(){
		ReconciliationRequest request = new ReconciliationRequest(queryString, limit);
		request.setTypes(new String[] {"http://data.linkedmdb.org/resource/movie/film"});
		String sparql = factory.getReconciliationSparqlQuery(request, searchPropertyUris);
		ResultSet resultset = executor.sparql(sparql);
		assertResult("http://data.linkedmdb.org/resource/film_series/261", resultset,"http://data.linkedmdb.org/resource/film/930","http://data.linkedmdb.org/resource/film/329");
	}
	
	@Test
	public void executeSimpleExactMatchReconciliationQuery(){
		ReconciliationRequest request = new ReconciliationRequest(exactMatchQueryString, limit);
		String sparql = factory.getExactMatchReconciliationSparqlQuery(request, searchPropertyUris);
		ResultSet resultset = executor.sparql(sparql);
		assertResult("http://data.linkedmdb.org/resource/film/329",resultset,"http://data.linkedmdb.org/resource/film/930");
	}
	
	@Test
	public void executeOneLabelOneTypeExactMatchReconciliationQuery(){
		ReconciliationRequest request = new ReconciliationRequest(exactMatchQueryString, limit);
		request.setTypes(new String[] {"http://data.linkedmdb.org/resource/movie/film"});
		String sparql = factory.getExactMatchReconciliationSparqlQuery(request, ImmutableList.of("http://www.w3.org/2000/01/rdf-schema#label"));
		ResultSet resultset = executor.sparql(sparql);
		assertResult("http://data.linkedmdb.org/resource/film/329",resultset,"http://data.linkedmdb.org/resource/film/930");
	}
	
	@Test
	public void executeExactMatchReconciliationQueryWithType(){
		ReconciliationRequest request = new ReconciliationRequest(exactMatchQueryString, limit);
		request.setTypes(new String[] {"http://data.linkedmdb.org/resource/movie/film"});
		String sparql = factory.getExactMatchReconciliationSparqlQuery(request, searchPropertyUris);
		ResultSet resultset = executor.sparql(sparql);
		assertResult("http://data.linkedmdb.org/resource/film_series/261", resultset,"http://data.linkedmdb.org/resource/film/930");
	}

	@Test
	public void executeExactMatchWithContextReconciliationQueryWithType(){
		ReconciliationRequest request = new ReconciliationRequest(exactMatchQueryString, limit);
		request.setTypes(new String[] {"http://data.linkedmdb.org/resource/movie/film"});
		PropertyContext prop1 = new PropertyContext("http://example.org/ontology/worksFor", new IdentifiedValueContext("http://example.org/resource/DERI"));
		PropertyContext prop2 = new PropertyContext("http://xmlns.com/foaf/0.1/nick", new TextualValueContext("fadmaa"));
		request.setContext(new ReconciliationRequestContext(prop1, prop2));
		String sparql = factory.getExactMatchReconciliationSparqlQuery(request, searchPropertyUris);
		ResultSet resultset = executor.sparql(sparql);
		assertTrue(!resultset.hasNext());//empty result
	}
	
	private void assertResult(String notContainedUri, ResultSet resultset, String... containedUris){
		Set<String> urisSet = new HashSet<String>(Arrays.asList(containedUris));
		while(resultset.hasNext()){
			QuerySolution solution = resultset.nextSolution();
			Resource r = solution.getResource("entity");
			String uri = r.getURI();
			urisSet.remove(uri);
			if(uri.equals(notContainedUri)){
				fail(notContainedUri + " was found in the result") ;
			}
		}
		
		assertTrue(urisSet.isEmpty(), urisSet + " were not found in the result");
	}
	
	
	/*
	 * Suggest type 
	 */
	@Test
	public void sugestTypeTest(){
		String prefix = "fil";
		String sparql = factory.getTypeSuggestSparqlQuery(prefix, limit);
		ResultSet resultset = executor.sparql(sparql);
		assertResultsetExact("type",resultset,"http://data.linkedmdb.org/resource/movie/film");
	}
	
	private void assertResultsetExact(String propertyName, ResultSet resultset, String... containedUris){
		Set<String> urisSet = new HashSet<String>(Arrays.asList(containedUris));
		while(resultset.hasNext()){
			QuerySolution solution = resultset.nextSolution();
			Resource r = solution.getResource(propertyName);
			String uri = r.getURI();
			assertTrue(urisSet.remove(uri));
		}
		
		assertTrue(urisSet.isEmpty(), urisSet + " were not found in the result");
	}
	
	/*
	 * Suggest property 
	 */
	@Test
	public void sugestPropertyTest(){
		String prefix = "init";
		String sparql = factory.getPropertySuggestSparqlQuery(prefix, "http://data.linkedmdb.org/resource/movie/film", limit);
		ResultSet resultset = executor.sparql(sparql);
		assertResultsetExact("p", resultset,"http://data.linkedmdb.org/resource/movie/initial_release_date");
	}
	
	/*
	 * search entity
	 */
	@Test
	public void searchEntityTest(){
		String prefix = "godf";
		String sparql = factory.getEntitySearchSparqlQuery(prefix, searchPropertyUris, limit);
		ResultSet resultset = executor.sparql(sparql);
		assertResult(null,resultset,"http://data.linkedmdb.org/resource/film/930","http://data.linkedmdb.org/resource/film/329",
		"http://data.linkedmdb.org/resource/film_series/261");
	}
}