org.openmrs.web.controller.program.ProgramFormControllerTest.java Source code

Introduction

Here is the source code for org.openmrs.web.controller.program.ProgramFormControllerTest.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.program;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.junit.Assert;
import org.junit.Test;
import org.openmrs.api.context.Context;
import org.openmrs.test.Verifies;
import org.openmrs.web.test.BaseModuleWebContextSensitiveTest;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.BindException;

/**
 * Tests the {@link ProgramFormController} class.
 */
public class ProgramFormControllerTest extends BaseModuleWebContextSensitiveTest {

    /**
     * @see ProgramFormController#onSubmit(HttpServletRequest,HttpServletResponse,Object,BindException)
     */
    @Test
    @Transactional(readOnly = true)
    @Verifies(value = "should save workflows with program", method = "onSubmit(HttpServletRequest,HttpServletResponse,Object,BindException)")
    public void onSubmit_shouldSaveWorkflowsWithProgram() throws Exception {

        // sanity check to make sure that program #3 doesn't have any workflows already:
        Assert.assertEquals(0, Context.getProgramWorkflowService().getProgram(3).getAllWorkflows().size());

        MockHttpServletRequest request = new MockHttpServletRequest("POST", "");
        request.setParameter("programId", "3");
        request.setParameter("allWorkflows", ":3"); // set one workflow on this program

        ProgramFormController controller = (ProgramFormController) applicationContext.getBean("programForm");
        controller.handleRequest(request, new MockHttpServletResponse());

        Assert.assertNotSame(0, Context.getProgramWorkflowService().getProgram(3).getAllWorkflows().size());
        Assert.assertEquals(1, Context.getProgramWorkflowService().getProgram(3).getAllWorkflows().size());
    }

    /**
     * @see ProgramFormController#onSubmit(HttpServletRequest,HttpServletResponse,Object,BindException)
     * @verifies edit existing workflows within programs
     */
    @Test
    @Transactional(readOnly = true)
    public void onSubmit_shouldEditExistingWorkflowsWithinPrograms() throws Exception {
        MockHttpServletRequest request = new MockHttpServletRequest("POST", "");
        request.setParameter("programId", "3");
        request.setParameter("allWorkflows", ":3 4"); // set two workflows on this program

        ProgramFormController controller = (ProgramFormController) applicationContext.getBean("programForm");
        controller.handleRequest(request, new MockHttpServletResponse());

        Assert.assertEquals(2, Context.getProgramWorkflowService().getProgram(3).getWorkflows().size());

        request = new MockHttpServletRequest("POST", "");
        request.setParameter("programId", "3");
        request.setParameter("allWorkflows", ":5"); // set one workflow on this program

        controller.handleRequest(request, new MockHttpServletResponse());

        Assert.assertEquals(1, Context.getProgramWorkflowService().getProgram(3).getWorkflows().size());
        Assert.assertEquals(5, Context.getProgramWorkflowService().getProgram(3).getWorkflows().iterator().next()
                .getConcept().getConceptId().intValue());
    }
}