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

Introduction

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

Prototype

@Override
    public Object getAttribute(String name)

Source Link

Usage

From source file:fi.okm.mpass.idp.authn.impl.SocialUserAuthServletTest.java

@Test
public void testAuthnStart() throws Exception {
    MockHttpServletRequest httpRequest = initHttpRequest();
    MockHttpServletResponse httpResponse = new MockHttpServletResponse();
    httpRequest.setRequestURI(nullAuthenticator);
    servlet.service(httpRequest, httpResponse);
    Assert.assertNull(httpRequest.getAttribute(ExternalAuthentication.AUTHENTICATION_ERROR_KEY));
    Assert.assertEquals(httpResponse.getRedirectedUrl(), authnRedirectUrl);
}

From source file:fi.okm.mpass.idp.authn.impl.SocialUserAuthServletTest.java

@Test
public void testAuthnSuccess() throws Exception {
    MockHttpServletRequest httpRequest = initHttpRequest();
    MockHttpServletResponse httpResponse = new MockHttpServletResponse();
    httpRequest.setRequestURI(subjectAuthenticator);
    servlet.service(httpRequest, httpResponse);
    Assert.assertNull(httpRequest.getAttribute(ExternalAuthentication.AUTHENTICATION_ERROR_KEY));
    Subject subject = (Subject) httpRequest.getAttribute(ExternalAuthentication.SUBJECT_KEY);
    Assert.assertEquals(subject.getPrincipals().iterator().next().getName(), username);
}

From source file:fi.okm.mpass.idp.authn.impl.SocialUserAuthServletTest.java

@Test
public void testAuthnThrows() throws Exception {
    MockHttpServletRequest httpRequest = initHttpRequest();
    MockHttpServletResponse httpResponse = new MockHttpServletResponse();
    httpRequest.setRequestURI(throwingAuthenticator);
    servlet.service(httpRequest, httpResponse);
    Assert.assertEquals(httpRequest.getAttribute(ExternalAuthentication.AUTHENTICATION_ERROR_KEY),
            throwingEvent);/*from w w w .  j a v  a2s . c o  m*/
}

From source file:io.jmnarloch.spring.request.correlation.filter.RequestCorrelationFilterTest.java

@Test
public void shouldInitiateCorrelationId() throws IOException, ServletException {

    // given/* w  w w. j a v  a  2 s  .co m*/
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockFilterChain chain = new MockFilterChain();

    // when
    instance.doFilter(request, response, chain);

    // then
    assertNotNull(request.getAttribute(RequestCorrelationConsts.ATTRIBUTE_NAME));
    assertNotNull(((HttpServletRequest) chain.getRequest()).getHeader(RequestCorrelationConsts.HEADER_NAME));
}

From source file:org.terasoluna.gfw.web.logging.mdc.XTrackMDCPutFilterTest.java

@Test
public void testGetMDCValue_default_attributeName() throws ServletException {
    xTrackMDCPutFilter.init(mockFilterConfig);
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    String xTrack = xTrackMDCPutFilter.getMDCValue(request, response);
    assertThat(xTrack, is(notNullValue()));
    assertThat(xTrack.matches("^[a-f0-9]{32}$"), is(true));
    assertThat(response.getHeader("X-Track"), is(xTrack));
    assertThat((String) request.getAttribute("X-Track"), is(xTrack));
}

From source file:org.terasoluna.gfw.web.logging.mdc.XTrackMDCPutFilterTest.java

@Test
public void testGetMDCValue_changed_by_initParam() throws ServletException {
    mockFilterConfig.addInitParameter("attributeName", "X-Hoge");
    xTrackMDCPutFilter.init(mockFilterConfig);
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    String xTrack = xTrackMDCPutFilter.getMDCValue(request, response);
    assertThat(xTrack, is(notNullValue()));
    assertThat(xTrack.matches("^[a-f0-9]{32}$"), is(true));
    assertThat(response.getHeader("X-Hoge"), is(xTrack));
    assertThat((String) request.getAttribute("X-Hoge"), is(xTrack));
}

From source file:io.jmnarloch.spring.request.correlation.filter.RequestCorrelationFilterTest.java

@Test
public void shouldUseExistingCorrelationId() throws IOException, ServletException {

    // given//ww  w.ja v  a2  s  . co m
    final String requestId = UUID.randomUUID().toString();
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockFilterChain chain = new MockFilterChain();

    request.addHeader(RequestCorrelationConsts.HEADER_NAME, requestId);

    // when
    instance.doFilter(request, response, chain);

    // then
    final Object requestCorrelation = request.getAttribute(RequestCorrelationConsts.ATTRIBUTE_NAME);
    assertNotNull(requestCorrelation);
    assertEquals(requestId, ((RequestCorrelation) requestCorrelation).getRequestId());

    final String header = ((HttpServletRequest) chain.getRequest())
            .getHeader(RequestCorrelationConsts.HEADER_NAME);
    assertNotNull(header);
    assertEquals(requestId, header);
}

From source file:org.terasoluna.gfw.web.logging.mdc.XTrackMDCPutFilterTest.java

@Test
public void testGetMDCValue_default_attributeName_set_in_http_request() throws ServletException {
    xTrackMDCPutFilter.init(mockFilterConfig);
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();

    request.addHeader("X-Track", "hoge");
    String xTrack = xTrackMDCPutFilter.getMDCValue(request, response);
    assertThat(xTrack, is(notNullValue()));
    assertThat(xTrack, is("hoge"));
    assertThat(response.getHeader("X-Track"), is("hoge"));
    assertThat((String) request.getAttribute("X-Track"), is("hoge"));
}

From source file:io.jmnarloch.spring.request.correlation.filter.RequestCorrelationFilterTest.java

@Test
public void shouldUseCustomHeader() throws IOException, ServletException {

    // given// w  w w .j  a  va2s .  com
    final String headerName = "X-TraceId";
    final String requestId = UUID.randomUUID().toString();
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockFilterChain chain = new MockFilterChain();

    request.addHeader(headerName, requestId);
    properties.setHeaderName(headerName);

    // when
    instance.doFilter(request, response, chain);

    // then
    final Object requestCorrelation = request.getAttribute(RequestCorrelationConsts.ATTRIBUTE_NAME);
    assertNotNull(requestCorrelation);
    assertEquals(requestId, ((RequestCorrelation) requestCorrelation).getRequestId());

    final String header = ((HttpServletRequest) chain.getRequest()).getHeader(headerName);
    assertNotNull(header);
    assertEquals(requestId, header);
}

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

/**
 * Check to see if the admin's secret question comes back
 *
 * @throws Exception/*from  w w  w .j av  a 2s.  co m*/
 */

@Test
public void shouldSetARandomSecretQuestionWhenTheUsernameIsInvalid() throws Exception {

    ForgotPasswordFormController controller = new ForgotPasswordFormController();

    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();

    request.setParameter("uname", "invaliduser");
    request.setMethod("POST");

    controller.handleRequest(request, response);

    Assert.assertEquals("invaliduser", request.getAttribute("uname"));

    List<String> questions = new ArrayList<String>();

    questions.add(Context.getMessageSourceService().getMessage("What is your best friend's name?"));
    questions.add(Context.getMessageSourceService().getMessage("What is your grandfather's home town?"));
    questions.add(Context.getMessageSourceService().getMessage("What is your mother's maiden name?"));
    questions.add(Context.getMessageSourceService().getMessage("What is your favorite band?"));
    questions.add(Context.getMessageSourceService().getMessage("What is your first pet's name?"));
    questions.add(Context.getMessageSourceService().getMessage("What is your brother's middle name?"));
    questions.add(Context.getMessageSourceService().getMessage("Which city were you born in?"));

    //Check that one of the fake questions is assigned to the invalid username
    Assert.assertTrue(questions.contains(request.getAttribute("secretQuestion")));
}