DocumentProjectionTest.java

import static play.test.Helpers.GET;
import static play.test.Helpers.POST;
import static play.test.Helpers.DELETE;
import static play.test.Helpers.fakeApplication;
import static play.test.Helpers.routeAndCall;
import static play.test.Helpers.running;

import java.io.IOException;

import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.json.JSONObject;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import play.mvc.Result;
import play.mvc.Http.Status;
import play.test.FakeRequest;
import core.AbstractDocumentTest;
import core.TestConfig;


public class DocumentProjectionTest extends AbstractDocumentTest {
	private Object json = null;
	private JsonNode document1;
	private JsonNode document2;
	private JsonNode document3;
	private JsonNode document4;
	
	private String sFakeCollection;

	@Override
	public String getRouteAddress() {
		return SERVICE_ROUTE + TestConfig.TEST_COLLECTION_NAME;
	}

	@Override
	public String getMethod() {
		return GET;
	}

	@Override
	protected void assertContent(String s) {
		json = toJSON(s);
		assertJSON(json, "@version");
	}
	
	@Before
	public void createCollection() throws JsonProcessingException, IOException{
		document1 = new ObjectMapper().readTree("{\"total\":2,\"city\":\"rome\"}");
		document2 = new ObjectMapper().readTree("{\"total\":3,\"city\":\"rome\"}");
		document3 = new ObjectMapper().readTree("{\"total\":4,\"city\":\"milan\"}");
		document4 = new ObjectMapper().readTree("{\"total\":\"5\",\"city\":\"milan\"}");
	}
	
	@Test
	public void testOnlyFields(){
		running
		(
			getFakeApplication(), 
			new Runnable() 		{
				public void run() {
			 		continueOnFail(false);
					sFakeCollection = new AdminCollectionFunctionalTest().routeCreateCollection();
					//create three different documents
					FakeRequest request = new FakeRequest(POST, getRouteAddress(sFakeCollection));
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					request = request.withJsonBody(document1);
					Result result = routeAndCall(request); 
					assertRoute(result, "testOnlyFields CREATE 1", Status.OK, null, true);
					String id1=getUuid();
					
					request = new FakeRequest(POST, getRouteAddress(sFakeCollection));
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					request = request.withJsonBody(document2);
					result = routeAndCall(request); 
					assertRoute(result, "testOnlyFields CREATE 2", Status.OK, null, true);
					String id2=getUuid();
					
					request = new FakeRequest(POST, getRouteAddress(sFakeCollection));
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					request = request.withJsonBody(document3);
					result = routeAndCall(request); 
					assertRoute(result, "testOnlyFields CREATE 3", Status.OK, null, true);
					String id3=getUuid();

					request = new FakeRequest(POST, getRouteAddress(sFakeCollection));
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					request = request.withJsonBody(document4);
					result = routeAndCall(request); 
					assertRoute(result, "testOnlyFields CREATE 4", Status.OK, null, true);
					String id4=getUuid();
					
					//read one document
					request = new FakeRequest(GET, "/document/" + sFakeCollection + "/" + id1);
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					result = routeAndCall(request);
					assertRoute(result, "testOnlyFields Reload document 1", Status.OK, "\"total\":2,\"city\":\"rome\"", true);
					
					//retrieve fields with aggregate functions amd group by clause
					request = new FakeRequest(GET, "/document/" + sFakeCollection + "?fields=sum(total) as tot,city&groupBy=city&orderBy=city");
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					result = routeAndCall(request);
					assertRoute(result, "testOnlyFields fields 1", Status.OK, "\"tot\":5,\"city\":\"rome\"", true);
					assertRoute(result, "testOnlyFields fields 2", Status.OK, "\"tot\":4,\"city\":\"milan\"", true);					

					
					request = new FakeRequest(GET, "/document/" + sFakeCollection + "?fields=sum(eval(\"(total - 10) * 2\")) as tot,city&groupBy=city&orderBy=city");
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					result = routeAndCall(request);
					assertRoute(result, "testOnlyFields fields 3", Status.OK, "\"tot\":-30,\"city\":\"rome\"", true);
					
					//one total field is a string, to include its value we must use the eval() function
					request = new FakeRequest(GET, "/document/" + sFakeCollection + "?fields=sum(eval(\"total * 1\")) as tot,city&groupBy=city&orderBy=city");
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					result = routeAndCall(request);
					assertRoute(result, "testOnlyFields fields 4", Status.OK, "\"tot\":5,\"city\":\"rome\"", true);
					assertRoute(result, "testOnlyFields fields 5", Status.OK, "\"tot\":9,\"city\":\"milan\"", true);					

					
					//delete collection
					request = new FakeRequest(DELETE, "/admin/collection/" + sFakeCollection );
					request = request.withHeader(TestConfig.KEY_APPCODE, TestConfig.VALUE_APPCODE);
					request = request.withHeader(TestConfig.KEY_AUTH, TestConfig.AUTH_ADMIN_ENC);
					result = routeAndCall(request);
					assertRoute(result, "testOnlyFields delete", Status.OK, null, false);					
					
				}
			}
		);			
	}
	
	private String getUuid()	{
		String sUuid = null;

		try	{
			JSONObject jo = (JSONObject)json;
			sUuid = jo.getJSONObject("data").getString("id");
		}
		catch (Exception ex)	{
			Assert.fail("Cannot get UUID (id) value: " + ex.getMessage() + "\n The json object is: \n" + json);
		}
		return sUuid;
	}
}