Example usage for org.springframework.mock.web MockHttpServletRequest 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.cloudfoundry.identity.uaa.scim.endpoints.ScimUserEndpointsTests.java

public void update_when_internal_user_management_is_disabled(String origin) throws Exception {
    ScimUser user = new ScimUser(null, "dave", "David", "Syer");
    user.setPassword("password");
    user.addEmail("dave@vmware.com");
    user.setOrigin(origin);//from   ww w  .  j  a  v  a  2 s .  c  o  m

    user = endpoints.createUser(user, new MockHttpServletRequest(), new MockHttpServletResponse());

    MockHttpServletResponse httpServletResponse = new MockHttpServletResponse();
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setAttribute(DisableInternalUserManagementFilter.DISABLE_INTERNAL_USER_MANAGEMENT, true);
    endpoints.updateUser(user, user.getId(), "*", request, httpServletResponse);
}

From source file:org.jtalks.jcommune.web.controller.UserProfileControllerTest.java

@Test
public void testSaveUserLanguage() throws ServletException {
    LocaleResolver localeResolver = mock(LocaleResolver.class);
    JCUser user = getUser();/*from  w  w  w .  ja  v  a  2  s . co  m*/
    MockHttpServletRequest reuqest = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();

    reuqest.setAttribute(LOCALE_RESOLVER_ATTRIBUTE, localeResolver);
    when(userService.getCurrentUser()).thenReturn(user);

    profileController.saveUserLanguage("ru", response, reuqest);

    verify(userService).changeLanguage(user, Language.RUSSIAN);
    verify(localeResolver).setLocale(reuqest, response, Language.RUSSIAN.getLocale());
}

From source file:org.springframework.security.web.authentication.www.BasicAuthenticationFilterTests.java

@Test
public void skippedOnErrorDispatch() throws Exception {

    String token = "bad:credentials";
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("Authorization", "Basic " + new String(Base64.encodeBase64(token.getBytes())));
    request.setServletPath("/some_file.html");
    request.setAttribute(WebUtils.ERROR_REQUEST_URI_ATTRIBUTE, "/error");
    MockHttpServletResponse response = new MockHttpServletResponse();

    FilterChain chain = mock(FilterChain.class);

    filter.doFilter(request, response, chain);

    assertThat(response.getStatus()).isEqualTo(200);
}

From source file:org.springframework.test.context.web.ServletTestExecutionListener.java

private void setUpRequestContextIfNecessary(TestContext testContext) {
    if (!isActivated(testContext) || alreadyPopulatedRequestContextHolder(testContext)) {
        return;/*from w  w w  . ja va  2  s.  c  o  m*/
    }

    ApplicationContext context = testContext.getApplicationContext();

    if (context instanceof WebApplicationContext) {
        WebApplicationContext wac = (WebApplicationContext) context;
        ServletContext servletContext = wac.getServletContext();
        Assert.state(servletContext instanceof MockServletContext, () -> String.format(
                "The WebApplicationContext for test context %s must be configured with a MockServletContext.",
                testContext));

        if (logger.isDebugEnabled()) {
            logger.debug(String.format(
                    "Setting up MockHttpServletRequest, MockHttpServletResponse, ServletWebRequest, and RequestContextHolder for test context %s.",
                    testContext));
        }

        MockServletContext mockServletContext = (MockServletContext) servletContext;
        MockHttpServletRequest request = new MockHttpServletRequest(mockServletContext);
        request.setAttribute(CREATED_BY_THE_TESTCONTEXT_FRAMEWORK, Boolean.TRUE);
        MockHttpServletResponse response = new MockHttpServletResponse();
        ServletWebRequest servletWebRequest = new ServletWebRequest(request, response);

        RequestContextHolder.setRequestAttributes(servletWebRequest);
        testContext.setAttribute(POPULATED_REQUEST_CONTEXT_HOLDER_ATTRIBUTE, Boolean.TRUE);
        testContext.setAttribute(RESET_REQUEST_CONTEXT_HOLDER_ATTRIBUTE, Boolean.TRUE);

        if (wac instanceof ConfigurableApplicationContext) {
            @SuppressWarnings("resource")
            ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) wac;
            ConfigurableListableBeanFactory bf = configurableApplicationContext.getBeanFactory();
            bf.registerResolvableDependency(MockHttpServletResponse.class, response);
            bf.registerResolvableDependency(ServletWebRequest.class, servletWebRequest);
        }
    }
}