ObsController1_9Test.java

/**
 * This Source Code Form is subject to the terms of the Mozilla Public License,
 * v. 2.0. If a copy of the MPL was not distributed with this file, You can
 * obtain one at http://mozilla.org/MPL/2.0/. OpenMRS is also distributed under
 * the terms of the Healthcare Disclaimer located at http://openmrs.org/license.
 *
 * Copyright (C) OpenMRS Inc. OpenMRS is a registered trademark and the OpenMRS
 * graphic logo is a trademark of OpenMRS Inc.
 */
package org.openmrs.module.webservices.rest.web.v1_0.controller.openmrs1_9;

import org.apache.commons.beanutils.PropertyUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.output.ByteArrayOutputStream;
import org.junit.Assert;
import org.junit.Test;
import org.openmrs.ConceptComplex;
import org.openmrs.ConceptName;
import org.openmrs.GlobalProperty;
import org.openmrs.api.AdministrationService;
import org.openmrs.api.ConceptService;
import org.openmrs.api.context.Context;
import org.openmrs.module.webservices.rest.SimpleObject;
import org.openmrs.module.webservices.rest.web.RestTestConstants1_8;
import org.openmrs.module.webservices.rest.web.response.ResourceDoesNotSupportOperationException;
import org.openmrs.module.webservices.rest.web.v1_0.controller.MainResourceControllerTest;
import org.openmrs.util.OpenmrsConstants;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.mock.web.MockMultipartHttpServletRequest;

import javax.xml.bind.DatatypeConverter;
import java.io.InputStream;
import java.util.Locale;
import static org.junit.Assert.*;
import static org.hamcrest.Matchers.*;

public class ObsController1_9Test extends MainResourceControllerTest {
	
	@Autowired
	ConceptService conceptService;
	
	@Autowired
	AdministrationService adminService;
	
	@Test
	public void getObs_shouldGetObsConceptByConceptMappings() throws Exception {
		String json = "{ \"value\":\"" + 10.0 + "\", \"person\":\"" + RestTestConstants1_8.PERSON_UUID
		        + "\", \"concept\":\"SNOMED CT:2332523\", \"obsDatetime\":\"2013-12-09T00:00:00.000+0100\"}";
		
		Object newObs = deserialize(handle(newPostRequest(getURI(), json)));
		Assert.assertNotNull(PropertyUtils.getProperty(newObs, "concept"));
	}
	
	/**
	 * @see org.openmrs.module.webservices.rest.web.v1_0.controller.MainResourceControllerTest#getURI()
	 */
	@Override
	public String getURI() {
		return "obs";
	}
	
	@Override
	public long getAllCount() {
		return Context.getObsService().getObservationCount(null, true);
	}
	
	/**
	 * @see org.openmrs.module.webservices.rest.web.v1_0.controller.MainResourceControllerTest#getUuid()
	 */
	@Override
	public String getUuid() {
		return RestTestConstants1_8.OBS_UUID;
	}
	
	/**
	 * @see org.openmrs.module.webservices.rest.web.v1_0.controller.MainResourceControllerTest#shouldGetAll()
	 */
	@Override
	@Test(expected = ResourceDoesNotSupportOperationException.class)
	public void shouldGetAll() throws Exception {
		super.shouldGetAll();
	}
	
	@Test
	public void shouldSubmitProperValueCodedWhenBooleanConceptUuidIsPassedAsValue() throws Exception {
		final String yesConceptUuid = "b055abd8-a420-4a11-8b98-02ee170a7b54";
		final String yesConceptId = "7";
		final String noConceptUuid = "934d8ef1-ea43-4f98-906e-dd03d5faaeb4";
		final String noConceptId = "8";
		
		AdministrationService as = Context.getAdministrationService();
		
		as.saveGlobalProperty(new GlobalProperty(OpenmrsConstants.GLOBAL_PROPERTY_TRUE_CONCEPT, yesConceptId));
		as.saveGlobalProperty(new GlobalProperty(OpenmrsConstants.GLOBAL_PROPERTY_FALSE_CONCEPT, noConceptId));
		
		long before = getAllCount();
		
		String yesPayload = "{\"concept\":\"89ca642a-dab6-4f20-b712-e12ca4fc6d36\"," + "\"value\":\"" + yesConceptUuid
		        + "\",\"person\":\"5946f880-b197-400b-9caa-a3c661d23041\","
		        + "\"obsDatetime\":\"2015-09-07T00:00:00.000+0530\"}";
		
		String noPayload = "{\"concept\":\"89ca642a-dab6-4f20-b712-e12ca4fc6d36\"," + "\"value\":\"" + noConceptUuid
		        + "\",\"person\":\"5946f880-b197-400b-9caa-a3c661d23041\","
		        + "\"obsDatetime\":\"2015-09-07T00:00:00.000+0530\"}";
		
		Object yesCreated = deserialize(handle(newPostRequest(getURI(), yesPayload)));
		Object yesValue = PropertyUtils.getProperty(yesCreated, "value");
		
		Object noCreated = deserialize(handle(newPostRequest(getURI(), noPayload)));
		Object noValue = PropertyUtils.getProperty(noCreated, "value");
		
		Assert.assertEquals(before + 2, getAllCount());
		Assert.assertEquals(yesConceptUuid, PropertyUtils.getProperty(yesValue, "uuid"));
		Assert.assertEquals(noConceptUuid, PropertyUtils.getProperty(noValue, "uuid"));
	}
	
	@Test
	public void shouldPostValueInJsonAndFetchComplexObs() throws Exception {
		ConceptComplex conceptComplex = newConceptComplex();
		
		InputStream in = getClass().getClassLoader().getResourceAsStream("customTestDataset.xml");
		
		ByteArrayOutputStream out = new ByteArrayOutputStream();
		IOUtils.copy(in, out);
		
		String value = DatatypeConverter.printBase64Binary(out.toByteArray());
		
		String json = "{\"concept\":\"" + conceptComplex.getUuid() + "\"," + "\"value\":\"" + value
		        + "\",\"person\":\"5946f880-b197-400b-9caa-a3c661d23041\","
		        + "\"obsDatetime\":\"2015-09-07T00:00:00.000+0530\"}";
		
		SimpleObject response = deserialize(handle(newPostRequest(getURI(), json)));
		
		MockHttpServletResponse rawResponse = handle(newGetRequest(getURI() + "/" + response.get("uuid") + "/value"));
		
		assertThat(out.toByteArray(), is(equalTo(rawResponse.getContentAsByteArray())));
	}
	
	@Test
	public void shouldUploadFileAndFetchComplexObs() throws Exception {
		ConceptComplex conceptComplex = newConceptComplex();
		
		InputStream in = getClass().getClassLoader().getResourceAsStream("customTestDataset.xml");
		
		ByteArrayOutputStream out = new ByteArrayOutputStream();
		IOUtils.copy(in, out);
		
		String json = "{\"concept\":\"" + conceptComplex.getUuid()
		        + "\", \"person\":\"5946f880-b197-400b-9caa-a3c661d23041\","
		        + "\"obsDatetime\":\"2015-09-07T00:00:00.000+0530\"}";
		
		MockMultipartHttpServletRequest request = newUploadRequest(getURI());
		request.addFile(new MockMultipartFile("file", "customTestDataset.xml", null, out.toByteArray()));
		request.addParameter("json", json);
		
		SimpleObject response = deserialize(handle(request));
		
		MockHttpServletResponse rawResponse = handle(newGetRequest(getURI() + "/" + response.get("uuid") + "/value"));
		
		assertThat(out.toByteArray(), is(equalTo(rawResponse.getContentAsByteArray())));
	}
	
	private ConceptComplex newConceptComplex() {
		setupBinaryDataHandler();
		
		ConceptComplex conceptComplex = new ConceptComplex();
		conceptComplex.setHandler("BinaryDataHandler");
		conceptComplex.addName(new ConceptName("Xml Test Data", Locale.ENGLISH));
		conceptComplex.setDatatype(conceptService.getConceptDatatypeByName("Complex"));
		conceptComplex.setConceptClass(conceptService.getConceptClassByName("Misc"));
		conceptService.saveConcept(conceptComplex);
		return conceptComplex;
	}
	
	private void setupBinaryDataHandler() {
		adminService.saveGlobalProperty(new GlobalProperty("obs.complex_obs_dir", "complexObsDir"));
	}
}