Example usage for org.springframework.mock.web MockHttpServletRequest setAttribute

List of usage examples for org.springframework.mock.web MockHttpServletRequest setAttribute

  1. HOME
  2. Java
  3. org.springframework
  4. org.springframework.mock.web.*
  5. MockHttpServletRequest
  6. setAttribute

Introduction

In this page you can find the example usage for org.springframework.mock.web MockHttpServletRequest setAttribute.

Prototype

@Override
    public void setAttribute(String name, @Nullable Object value)

Source Link

Usage

From source file:org.openmrs.web.controller.PortletControllerTest.java

/**
 * Convenience method to get the "model" from the controller's handleRequest method
 * // w w w.ja  va 2s . co  m
 * @param patientId the patient id to fetch
 * @return the Map from string to object of everything in the generated "model"
 * @throws Exception
 */
private Map<String, Object> getModelFromController(Integer patientId) throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest("GET", "");
    HttpServletResponse response = new MockHttpServletResponse();

    request.setAttribute(WebConstants.INIT_REQ_UNIQUE_ID, "1");
    request.getSession().setAttribute(WebConstants.OPENMRS_PORTLET_LAST_REQ_ID, "0");
    request.setAttribute("javax.servlet.include.servlet_path", "testPortlet");
    request.setAttribute("org.openmrs.portlet.parameters", new HashMap());
    request.setAttribute("org.openmrs.portlet.patientId", patientId);

    ModelAndView modelAndView = new PortletController().handleRequest(request, response);

    return (Map<String, Object>) modelAndView.getModel().get("model");
}

From source file:com.mtt.myapp.home.controller.HomeControllerTest.java

@Test
public void testHome() {
    MockHttpServletResponse res = new MockHttpServletResponse();
    MockHttpServletRequest req = new MockHttpServletRequest();
    CookieLocaleResolver localeResolver = new CookieLocaleResolver();
    req.setAttribute(DispatcherServlet.LOCALE_RESOLVER_ATTRIBUTE, localeResolver);
    User testUser2 = getTestUser();/*from   w  w w  .ja v a 2  s .  co m*/
    testUser2.setUserLanguage("EN");
    ModelMap model = new ModelMap();
    String viewName = homeController.home(testUser2, null, null, model, res, req);
    assertThat(viewName).isEqualTo("index");

    User testUserError = Mockito.spy(testUser2);
    Mockito.when(testUserError.getRole()).thenReturn(Role.SYSTEM_USER);
    viewName = homeController.home(testUserError, "Test Error message!", null, model, res, req);
    assertThat(viewName).isEqualTo("login");

}

From source file:org.ngrinder.home.controller.HomeControllerTest.java

@Test
public void testHome() {
    MockHttpServletResponse res = new MockHttpServletResponse();
    MockHttpServletRequest req = new MockHttpServletRequest();
    CookieLocaleResolver localeResolver = new CookieLocaleResolver();
    req.setAttribute(DispatcherServlet.LOCALE_RESOLVER_ATTRIBUTE, localeResolver);
    User testUser2 = getTestUser();//from   w  ww. ja  v  a  2s.c  o  m
    testUser2.setUserLanguage("EN");
    ModelMap model = new ModelMap();
    String viewName = homeController.home(testUser2, null, null, model, res, req);
    assertThat(viewName, is("index"));

    User testUserError = Mockito.spy(testUser2);
    Mockito.when(testUserError.getRole()).thenReturn(Role.SYSTEM_USER);
    viewName = homeController.home(testUserError, "Test Error message!", null, model, res, req);
    assertThat(viewName, is("login"));

}

From source file:org.sventon.web.tags.UrlTagTest.java

protected MockPageContext createPageContext() {
    final MockServletContext sc = new MockServletContext();
    final MockHttpServletRequest request = new MockHttpServletRequest(sc);
    final MockHttpServletResponse response = new MockHttpServletResponse();
    request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE,
            Mockito.mock(WebApplicationContext.class));
    return new MockPageContext(sc, request, response);
}

From source file:com.domingosuarez.boot.autoconfigure.jade4j.Jade4JAutoConfigurationTests.java

@Test
public void createLayoutFromConfigClass() throws Exception {
    AnnotationConfigWebApplicationContext context = new AnnotationConfigWebApplicationContext();
    context.register(Jade4JAutoConfiguration.class, PropertyPlaceholderAutoConfiguration.class);
    MockServletContext servletContext = new MockServletContext();
    context.setServletContext(servletContext);
    context.refresh();//from   w  w w  .ja  v  a2  s  .c  o  m

    JadeView view = (JadeView) context.getBean(JadeViewResolver.class).resolveViewName("demo", Locale.UK);
    MockHttpServletResponse response = new MockHttpServletResponse();
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setAttribute(RequestContext.WEB_APPLICATION_CONTEXT_ATTRIBUTE, context);
    view.render(params(), request, response);
    String result = response.getContentAsString();
    assertTrue("Wrong result: " + result, result.contains("<title>Jade</title>"));
    assertTrue("Wrong result: " + result, result.contains("<h2>With user</h2>"));
    context.close();
}

From source file:ch.ralscha.extdirectspring.bean.ExtDirectResponseBuilderTest.java

private MockHttpServletRequest createRequest() {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);

    request.setParameter("extAction", "action");
    request.setParameter("extMethod", "method");
    request.setParameter("extType", "type");
    request.setParameter("extTID", "1");
    return request;
}

From source file:org.hdiv.web.servlet.tags.AbstractTagTests.java

protected MockPageContext createPageContext() {
    MockServletContext sc = new MockServletContext();
    SimpleWebApplicationContext wac = new SimpleWebApplicationContext();
    wac.setServletContext(sc);/* w w  w.  ja  v a2s.  c o m*/
    wac.setNamespace("test");
    wac.refresh();

    MockHttpServletRequest request = new MockHttpServletRequest(sc);
    MockHttpServletResponse response = new MockHttpServletResponse();
    if (inDispatcherServlet()) {
        request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
        LocaleResolver lr = new AcceptHeaderLocaleResolver();
        request.setAttribute(DispatcherServlet.LOCALE_RESOLVER_ATTRIBUTE, lr);
        ThemeResolver tr = new FixedThemeResolver();
        request.setAttribute(DispatcherServlet.THEME_RESOLVER_ATTRIBUTE, tr);
        request.setAttribute(DispatcherServlet.THEME_SOURCE_ATTRIBUTE, wac);
    } else {
        sc.setAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
    }

    return new MockPageContext(sc, request, response);
}

From source file:org.ambraproject.admin.filter.RedirectToAmbraFilterTest.java

@Test(dataProvider = "journalContext")
public void testFilter(VirtualJournalContext journalContext, String expectedRedirectUrl)
        throws ServletException, IOException {
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    MockFilterChain filterChain = new MockFilterChain();
    request.setAttribute(VirtualJournalContext.PUB_VIRTUALJOURNAL_CONTEXT, journalContext);
    request.setRequestURI("/article");
    request.setQueryString("id=foo:bar");

    filter.doFilter(request, response, filterChain);
    assertEquals(response.getRedirectedUrl(), expectedRedirectUrl + "/article?id=foo:bar",
            "response didn't get redirected correctly");
}

From source file:org.hdiv.web.servlet.view.freemarker.FreeMarkerViewTests.java

@Test
public void testValidTemplateName() throws Exception {
    FreeMarkerView fv = new FreeMarkerView();

    MockControl wmc = MockControl.createNiceControl(WebApplicationContext.class);
    WebApplicationContext wac = (WebApplicationContext) wmc.getMock();
    MockServletContext sc = new MockServletContext();

    wac.getBeansOfType(FreeMarkerConfig.class, true, false);
    Map configs = new HashMap();
    FreeMarkerConfigurer configurer = new FreeMarkerConfigurer();
    configurer.setConfiguration(new FreeMarkerTestConfiguration());
    configs.put("configurer", configurer);
    wmc.setReturnValue(configs);/*from w w w . j  a va 2s.  c o  m*/
    wac.getParentBeanFactory();
    wmc.setReturnValue(null);
    wac.getServletContext();
    wmc.setReturnValue(sc, 5);
    wmc.replay();

    fv.setUrl("templateName");
    fv.setApplicationContext(wac);

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addPreferredLocale(Locale.US);
    request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
    request.setAttribute(DispatcherServlet.LOCALE_RESOLVER_ATTRIBUTE, new AcceptHeaderLocaleResolver());
    HttpServletResponse response = new MockHttpServletResponse();

    Map model = new HashMap();
    model.put("myattr", "myvalue");
    fv.render(model, request, response);

    wmc.verify();
    assertEquals(AbstractView.DEFAULT_CONTENT_TYPE, response.getContentType());
}

From source file:org.hdiv.web.servlet.view.freemarker.FreeMarkerViewTests.java

@Test
public void testKeepExistingContentType() throws Exception {
    FreeMarkerView fv = new FreeMarkerView();

    MockControl wmc = MockControl.createNiceControl(WebApplicationContext.class);
    WebApplicationContext wac = (WebApplicationContext) wmc.getMock();
    MockServletContext sc = new MockServletContext();

    wac.getBeansOfType(FreeMarkerConfig.class, true, false);
    Map configs = new HashMap();
    FreeMarkerConfigurer configurer = new FreeMarkerConfigurer();
    configurer.setConfiguration(new FreeMarkerTestConfiguration());
    configs.put("configurer", configurer);
    wmc.setReturnValue(configs);/*  w ww .  j  av  a2  s .  c  o m*/
    wac.getParentBeanFactory();
    wmc.setReturnValue(null);
    wac.getServletContext();
    wmc.setReturnValue(sc, 5);
    wmc.replay();

    fv.setUrl("templateName");
    fv.setApplicationContext(wac);

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addPreferredLocale(Locale.US);
    request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
    request.setAttribute(DispatcherServlet.LOCALE_RESOLVER_ATTRIBUTE, new AcceptHeaderLocaleResolver());
    HttpServletResponse response = new MockHttpServletResponse();
    response.setContentType("myContentType");

    Map model = new HashMap();
    model.put("myattr", "myvalue");
    fv.render(model, request, response);

    wmc.verify();
    assertEquals("myContentType", response.getContentType());
}