src/test/java/CollectionQueryTreeTest.java

import static org.junit.Assert.*;

import java.io.IOException;
import java.util.ArrayList;

import de.ids_mannheim.korap.query.serialize.QuerySerializer;
import de.ids_mannheim.korap.util.QueryException;

import org.junit.Test;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

public class CollectionQueryTreeTest {

	String query = "foo";
	String ql = "poliqarpplus";
	String collection;
	ArrayList<JsonNode> operands;

	QuerySerializer qs = new QuerySerializer();
	ObjectMapper mapper = new ObjectMapper();
	JsonNode res;
	
	@Test
	public void testContext() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=politik";
		String contextString = "http://ids-mannheim.de/ns/KorAP/json-ld/v0.2/context.jsonld";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals(contextString, res.get("@context").asText());
	}
	
	@Test
	public void testSimple() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=politik";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("textClass", 		res.at("/collection/key").asText());
		assertEquals("politik", 		res.at("/collection/value").asText());
		assertEquals("match:eq", 		res.at("/collection/match").asText());
		
		collection = "textClass!=politik";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("textClass", 		res.at("/collection/key").asText());
		assertEquals("politik", 		res.at("/collection/value").asText());
		assertEquals("match:ne", 		res.at("/collection/match").asText());
	}
	
	@Test
	public void testContains() throws QueryException, JsonProcessingException, IOException {
		collection = "title~Mannheim";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("title", 			res.at("/collection/key").asText());
		assertEquals("Mannheim", 		res.at("/collection/value").asText());
		assertEquals("match:contains",  res.at("/collection/match").asText());
		
		collection = "title~\"IDS Mannheim\"";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("title",	 		res.at("/collection/key").asText());
		assertEquals("IDS Mannheim",	res.at("/collection/value").asText());
		assertEquals("match:contains",	res.at("/collection/match").asText());
	}
	
	@Test
	public void testTwoConjuncts() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=Sport & pubDate in 2014";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/match").asText());
		
		collection = "textClass=Sport & pubDate=2014";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/value").asText());
		assertEquals(true,				res.at("/collection/operands/1/type").isMissingNode());
		assertEquals("match:eq",		res.at("/collection/operands/1/match").asText());
		assertEquals("",				res.at("/warnings/0/"));	
	}

	@Test
	public void testThreeConjuncts() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=Sport & pubDate in 2014 & corpusId=WPD";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:docGroup", 	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/operands/0/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/operands/0/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("corpusId", 		res.at("/collection/operands/1/operands/1/key").asText());
		assertEquals("WPD",				res.at("/collection/operands/1/operands/1/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/1/match").asText());
	}
	@Test
	public void testTwoDisjuncts() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=Sport | pubDate in 2014";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/match").asText());
	}
	
	@Test
	public void testThreeDisjuncts() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=Sport | pubDate in 2014 | corpusId=WPD";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:docGroup", 	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/operands/0/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/operands/0/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("corpusId", 		res.at("/collection/operands/1/operands/1/key").asText());
		assertEquals("WPD",				res.at("/collection/operands/1/operands/1/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/1/match").asText());
	}
	
	@Test
	public void testMixed() throws QueryException, JsonProcessingException, IOException {
		collection = "textClass=Sport | (pubDate in 2014 & corpusId=WPD)";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:docGroup", 	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/operands/0/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/operands/0/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("corpusId", 		res.at("/collection/operands/1/operands/1/key").asText());
		assertEquals("WPD",				res.at("/collection/operands/1/operands/1/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/1/match").asText());
		
		collection = "textClass=Sport | pubDate in 2014 & corpusId=WPD";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/@type").asText());
		assertEquals("textClass", 		res.at("/collection/operands/0/key").asText());
		assertEquals("Sport",			res.at("/collection/operands/0/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/0/match").asText());
		assertEquals("korap:docGroup", 	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("pubDate",	 		res.at("/collection/operands/1/operands/0/key").asText());
		assertEquals("2014",			res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("type:date",		res.at("/collection/operands/1/operands/0/type").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/0/match").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("corpusId", 		res.at("/collection/operands/1/operands/1/key").asText());
		assertEquals("WPD",				res.at("/collection/operands/1/operands/1/value").asText());
		assertEquals("match:eq",		res.at("/collection/operands/1/operands/1/match").asText());
		
		collection = "(textClass=Sport | pubDate in 2014) & corpusId=WPD";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/0/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operands/0/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/0/@type").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/1/@type").asText());
		assertEquals("korap:doc",	 	res.at("/collection/operands/1/@type").asText());
		
		collection = "(textClass=Sport & pubDate in 2014) & corpusId=WPD";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/0/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/0/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/0/@type").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/1/@type").asText());
		assertEquals("korap:doc",	 	res.at("/collection/operands/1/@type").asText());
		
		collection = "(textClass=Sport & textClass=ausland) | (corpusID=WPD & author=White)";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/0/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/0/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/0/@type").asText());
		assertEquals("Sport",	 		res.at("/collection/operands/0/operands/0/value").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/1/@type").asText());
		assertEquals("ausland",	 		res.at("/collection/operands/0/operands/1/value").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("WPD",		 		res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("White",	 		res.at("/collection/operands/1/operands/1/value").asText());
		
		collection = "(textClass=Sport & textClass=ausland) | (corpusID=WPD & author=White & pubDate in 2000)";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:docGroup", 	res.at("/collection/@type").asText());
		assertEquals("operation:or",	res.at("/collection/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/0/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/0/operation").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operation").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/0/@type").asText());
		assertEquals("Sport",	 		res.at("/collection/operands/0/operands/0/value").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/0/operands/1/@type").asText());
		assertEquals("ausland",	 		res.at("/collection/operands/0/operands/1/value").asText());
		assertEquals("korap:doc", 		res.at("/collection/operands/1/operands/0/@type").asText());
		assertEquals("WPD",		 		res.at("/collection/operands/1/operands/0/value").asText());
		assertEquals("korap:docGroup",	res.at("/collection/operands/1/operands/1/@type").asText());
		assertEquals("operation:and",	res.at("/collection/operands/1/operands/1/operation").asText());
		assertEquals("White",	 		res.at("/collection/operands/1/operands/1/operands/0/value").asText());
		assertEquals("2000",	 		res.at("/collection/operands/1/operands/1/operands/1/value").asText());
	}

	@Test
	public void testDateYear() throws QueryException, JsonProcessingException, IOException {
		collection = "pubDate in 2000";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000",	 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:eq", 		res.at("/collection/match").asText());
		
		collection = "pubDate = 2000";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000",	 		res.at("/collection/value").asText());
		assertEquals(true, 				res.at("/collection/type").isMissingNode());
		assertEquals("match:eq", 		res.at("/collection/match").asText());
		
		collection = "pubDate since 2000";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000",	 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:geq", 		res.at("/collection/match").asText());
		
		collection = "pubDate until 2000";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000",	 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:leq", 		res.at("/collection/match").asText());
	}
	
	@Test
	public void testDateMonthDay() throws QueryException, JsonProcessingException, IOException {
		collection = "pubDate in 2000-02";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000-02",	 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:eq", 		res.at("/collection/match").asText());
		
		collection = "pubDate = 2000-12";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000-12",	 		res.at("/collection/value").asText());
		assertEquals(true, 				res.at("/collection/type").isMissingNode());
		assertEquals("match:eq", 		res.at("/collection/match").asText());
		
		collection = "pubDate since 2000-02-01";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000-02-01", 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:geq", 		res.at("/collection/match").asText());
		
		collection = "pubDate until 2000-01-01";
		qs.setQuery(query,ql);
		qs.setCollection(collection);
		res = mapper.readTree(qs.toJSON());
		assertEquals("korap:doc", 		res.at("/collection/@type").asText());
		assertEquals("pubDate", 		res.at("/collection/key").asText());
		assertEquals("2000-01-01", 		res.at("/collection/value").asText());
		assertEquals("type:date", 		res.at("/collection/type").asText());
		assertEquals("match:leq", 		res.at("/collection/match").asText());
	}
}