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

Introduction

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

Prototype

public void addHeader(String name, Object value)

Source Link

Document

Add an HTTP header entry for the given name.

Usage

From source file:org.ngrinder.agent.controller.AgentManagerControllerTest.java

@Before
public void setMockRequest() {
    MockHttpServletRequest req = new MockHttpServletRequest();
    req.addHeader("User-Agent", "Win");
    SecurityContextHolderAwareRequestWrapper reqWrapper = new SecurityContextHolderAwareRequestWrapper(req,
            "U");
    RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(reqWrapper));
}

From source file:com.linecorp.bot.servlet.LineBotCallbackRequestParserTest.java

@Test
public void testCallRequest() throws Exception {
    InputStream resource = getClass().getClassLoader().getResourceAsStream("callback-request.json");
    byte[] requestBody = ByteStreams.toByteArray(resource);

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("X-Line-Signature", "SSSSIGNATURE");
    request.setContent(requestBody);/*from  w w w.j  a  va  2 s  .c  o m*/

    doReturn(true).when(lineSignatureValidator).validateSignature(requestBody, "SSSSIGNATURE");

    CallbackRequest callbackRequest = lineBotCallbackRequestParser.handle(request);

    assertThat(callbackRequest).isNotNull();

    final List<Event> result = callbackRequest.getEvents();

    final MessageEvent messageEvent = (MessageEvent) result.get(0);
    final TextMessageContent text = (TextMessageContent) messageEvent.getMessage();
    assertThat(text.getText()).isEqualTo("Hello, world");

    final String followedUserId = messageEvent.getSource().getUserId();
    assertThat(followedUserId).isEqualTo("u206d25c2ea6bd87c17655609a1c37cb8");
    assertThat(messageEvent.getTimestamp()).isEqualTo(Instant.parse("2016-05-07T13:57:59.859Z"));
}

From source file:org.jasig.cas.services.web.ServiceThemeResolverTests.java

public void testGetDefaultService() {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    request.setParameter("service", "myServiceId");
    request.addHeader("User-Agent", "Mozilla");
    assertEquals("test", this.serviceThemeResolver.resolveThemeName(request));
}

From source file:org.jasig.cas.services.web.ServiceThemeResolverTests.java

public void testGetDefaultServiceWithNoServicesManager() {
    this.serviceThemeResolver.setServicesManager(null);
    final MockHttpServletRequest request = new MockHttpServletRequest();
    request.setParameter("service", "myServiceId");
    request.addHeader("User-Agent", "Mozilla");
    assertEquals("test", this.serviceThemeResolver.resolveThemeName(request));
}

From source file:com.google.api.server.spi.IoUtilTest.java

@Test
public void testGetRequestInputStream_emptyStream() throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContent(new byte[0]);
    request.addHeader("Content-Encoding", "gzip");
    InputStream stream = IoUtil.getRequestInputStream(request);
    assertThat(stream).isNotInstanceOf(GZIPInputStream.class);
    assertThat(IoUtil.readStream(stream)).isEqualTo("");
}

From source file:com.thoughtworks.go.server.security.BasicAuthenticationFilterTest.java

@Test
public void shouldConvey_itsBasicProcessingFilter() throws IOException, ServletException {
    BasicAuthenticationFilter filter = new BasicAuthenticationFilter();
    final Boolean[] hadBasicMarkOnInsideAuthenticationManager = new Boolean[] { false };

    filter.setAuthenticationManager(new AuthenticationManager() {
        public Authentication authenticate(Authentication authentication) throws AuthenticationException {
            hadBasicMarkOnInsideAuthenticationManager[0] = BasicAuthenticationFilter.isProcessingBasicAuth();
            return new UsernamePasswordAuthenticationToken("school-principal",
                    "u can be principal if you know this!");
        }/*from   w ww.ja  v  a 2 s . c  o  m*/
    });
    assertThat(BasicAuthenticationFilter.isProcessingBasicAuth(), is(false));
    MockHttpServletRequest httpRequest = new MockHttpServletRequest();
    httpRequest.addHeader("Authorization", "Basic " + Base64.encodeBase64String("loser:boozer".getBytes()));
    filter.doFilterHttp(httpRequest, new MockHttpServletResponse(), new FilterChain() {
        public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse)
                throws IOException, ServletException {

        }
    });
    assertThat(BasicAuthenticationFilter.isProcessingBasicAuth(), is(false));

    assertThat(hadBasicMarkOnInsideAuthenticationManager[0], is(true));
}

From source file:org.jasig.cas.services.web.ServiceThemeResolverTests.java

public void testGetServiceTheme() {
    final RegisteredServiceImpl r = new RegisteredServiceImpl();
    r.setTheme("myTheme");
    r.setId(1000);/* ww  w  . ja  va 2s  . com*/
    r.setServiceId("myServiceId");

    this.servicesManager.save(r);

    final MockHttpServletRequest request = new MockHttpServletRequest();
    request.setParameter("service", "myServiceId");
    request.addHeader("User-Agent", "Mozilla");
    System.out.println("1");
    assertEquals("myTheme", this.serviceThemeResolver.resolveThemeName(request));
}

From source file:test.com.tsc9526.monalisa.service.actions.AbstractActionTest.java

protected Response getRespone(MockHttpServletRequest req) throws Exception {
    MockHttpServletResponse resp = new MockHttpServletResponse();
    req.addHeader("DEV_TEST", "true");

    DbQueryHttpServlet ms = new DbQueryHttpServlet();
    ms.service(req, resp);//from w w  w  .j  a  va2 s.  co  m

    String type = resp.getContentType();
    Assert.assertTrue(type.indexOf("json") > 0 && type.indexOf("utf-8") > 0);

    String body = resp.getContentAsString();

    Response r = Response.fromJson(body);

    if (req.getParameter("page") != null && r.getStatus() == 200) {
        int total = Integer.parseInt(resp.getHeader("X-Total-Page"));
        Assert.assertTrue(total >= 0);
    }

    return r;
}

From source file:se.vgregion.portal.auditlog.AuditLogInfoContainerFactoryImplTest.java

@Test
public final void testRequestHeaderInfo() {
    MockPortletRequest portletRequest = new MockPortletRequest();

    Map<String, String> uInfoMap = new HashMap<String, String>();
    uInfoMap.put(PortletRequest.P3PUserInfos.USER_LOGIN_ID.toString(), "remoteUid");
    portletRequest.setAttribute(PortletRequest.USER_INFO, uInfoMap);

    MockHttpServletRequest httpRequest = new MockHttpServletRequest();
    httpRequest.addHeader(AuditLogInfoContainerFactoryImpl.HEADER_X_FORWARDED_FOR, "127.0.0.2");

    given(converter.getHttpServletRequest(portletRequest)).willReturn(httpRequest);

    LdapUser ldapUser = new SimpleLdapUser("dn");
    ldapUser.setAttributeValue("cn", "searcherId");
    given(ldapService.getLdapUserByUid("remoteUid")).willReturn(ldapUser);

    AuditLogInfoContainer container = factory.getAuditLogInfoContainer("patientId", portletRequest);

    assertEquals("127.0.0.2", container.getRemoteIpAddress());
}

From source file:org.surfnet.oaaas.auth.AuthorizationServerFilterTest.java

private MockFilterChain doCallFilter(Resource[] resource, MockHttpServletResponse response)
        throws IOException, ServletException {
    super.setResponseResource(resource);
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("Origin", "http://www.test.org");
    request.addHeader(HttpHeaders.AUTHORIZATION, "bearer dummy-access-token");
    MockFilterChain chain = new MockFilterChain();
    filter.doFilter(request, response, chain);
    return chain;
}