org.openmrs.web.controller.observation.ObsFormControllerTest.java Source code

Introduction

Here is the source code for org.openmrs.web.controller.observation.ObsFormControllerTest.java

Source

/**
 * 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.web.controller.observation;

import static org.junit.Assert.assertEquals;

import java.util.List;

import javax.servlet.http.HttpServletResponse;

import org.junit.Assert;
import org.junit.Test;
import org.openmrs.Obs;
import org.openmrs.Person;
import org.openmrs.api.ObsService;
import org.openmrs.api.context.Context;
import org.openmrs.web.test.BaseModuleWebContextSensitiveTest;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.web.servlet.ModelAndView;

/**
 * Test the methods on the {@link org.openmrs.web.controller.observation.ObsFormController}
 */
public class ObsFormControllerTest extends BaseModuleWebContextSensitiveTest {

    /**
     * Tests that an "encounterId" parameter sets the obs.encounter attribute on an empty obs
     * 
     * @throws Exception
     */
    @Test
    public void shouldGetObsFormWithEncounterFilledIn() throws Exception {
        MockHttpServletRequest request = new MockHttpServletRequest("GET", "");
        request.setParameter("encounterId", "3");

        HttpServletResponse response = new MockHttpServletResponse();

        ObsFormController controller = new ObsFormController();

        ModelAndView modelAndView = controller.handleRequest(request, response);

        // make sure there is an "encounterId" element on the obs
        Obs commandObs = (Obs) modelAndView.getModel().get("command");
        Assert.assertNotNull(commandObs.getEncounter());

    }

    /**
     * Test to make sure a new patient form can save a person relationship
     * 
     * @throws Exception
     */
    @Test
    public void shouldSaveObsFormNormally() throws Exception {
        ObsService os = Context.getObsService();

        // set up the controller
        ObsFormController controller = new ObsFormController();
        controller.setApplicationContext(applicationContext);
        controller.setSuccessView("encounter.form");
        controller.setFormView("obs.form");

        // set up the request and do an initial "get" as if the user loaded the
        // page for the first time
        MockHttpServletRequest request = new MockHttpServletRequest("GET", "/admin/observations/obs.form");
        request.setSession(new MockHttpSession(null));
        HttpServletResponse response = new MockHttpServletResponse();
        controller.handleRequest(request, response);

        // set this to be a page submission
        request.setMethod("POST");

        // add all of the parameters that are expected
        // all but the relationship "3a" should match the stored data
        request.addParameter("person", "2");
        request.addParameter("encounter", "3");
        request.addParameter("location", "1");
        request.addParameter("obsDatetime", "05/05/2005");
        request.addParameter("concept", "4"); // CIVIL_STATUS (conceptid=4) concept
        request.addParameter("valueCoded", "5"); // conceptNameId=2458 for SINGLE concept
        request.addParameter("saveObs", "Save Obs"); // so that the form is processed

        // send the parameters to the controller
        controller.handleRequest(request, response);

        // make sure an obs was created
        List<Obs> obsForPatient = os.getObservationsByPerson(new Person(2));
        assertEquals(1, obsForPatient.size());
        assertEquals(3, obsForPatient.get(0).getEncounter().getId().intValue());
        assertEquals(1, obsForPatient.get(0).getLocation().getId().intValue());
    }

}