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

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

  1. HOME
  2. Java
  3. org.springframework
  4. org.springframework.mock.web.*
  5. MockHttpServletRequest
  6. 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.wrml.server.WrmlServletTest.java

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

    MockHttpServletRequest request = new MockHttpServletRequest();
    initMockHttpRequest(request, CAPRICA_SIX_ENDPOINT);
    request.setMethod(Method.Get.getProtocolGivenName());
    //request.addHeader(HttpHeaders.ACCEPT, JSON_MEDIA_TYPE);
    request.addHeader(WrmlServlet.WRML_HOST_HEADER_NAME, BAD_HOST_1);

    MockHttpServletResponse response = new MockHttpServletResponse();

    initMockWrmlRequest(request, Method.Get, CAPRICA_SIX_ENDPOINT, CAPRICA_SCHEMA_URI);

    _Servlet.service(request, response);

    // Verify Model Write
    Assert.assertTrue(response.getContentType().contains("text/plain"));
    Assert.assertEquals(HttpServletResponse.SC_BAD_REQUEST, response.getStatus());
    Assert.assertTrue(response.getContentLength() > 0);
    Assert.assertTrue(response.getContentAsString().contains("moose::/squirrel:"));
}

From source file:com.doitnext.http.router.RestRouterServletTest.java

private void setUpRequest(Object[] testCase, MockHttpServletRequest request) {
    String httpMethod = (String) testCase[0];
    String pathPrefix = (String) testCase[1];
    String pathInfo = (String) testCase[2];
    String queryString = (String) testCase[3];
    String parts[] = queryString.split("&");
    String acceptHeader = (String) testCase[4];
    String contentTypeHeader = (String) testCase[5];

    request.setServletPath("");
    request.setContextPath(pathPrefix);/* w w w  .  j av  a2  s .c o m*/
    request.setPathInfo(pathInfo);
    request.setMethod(httpMethod);
    request.setQueryString(queryString);
    for (String part : parts) {
        String pieces[] = part.split("=");
        if (pieces.length > 1)
            request.addParameter(pieces[0], pieces[1]);
    }
    if (acceptHeader != null)
        request.addHeader("Accept", acceptHeader);
    if (contentTypeHeader != null)
        request.setContentType(contentTypeHeader);
    HttpMethod mthd = HttpMethod.valueOf(httpMethod);
    if (mthd == HttpMethod.POST || mthd == HttpMethod.PUT) {

    }
}

From source file:com.photon.phresco.framework.rest.api.ConfigurationServiceTest.java

@Test
public void uploadFileOnTarget() {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("appDirName", appDirName);
    request.addHeader("actionType", "editconfig");
    request.addHeader("envName", "Production");
    request.addHeader("configType", "Theme");
    request.addHeader("configName", "serverconfig");
    request.addHeader("oldName", "test");
    request.addHeader("propName", "port");
    request.addHeader("X-File-Name", "test.zip");
    HttpServletRequest httpServletRequest = (HttpServletRequest) request;
    Response uploadFile = configurationService.uploadFile(httpServletRequest);
    Assert.assertEquals(200, uploadFile.getStatus());
}

From source file:com.photon.phresco.framework.rest.api.ConfigurationServiceTest.java

@Test
public void uploadFile() {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("appDirName", appDirName);
    request.addHeader("actionType", "editconfig");
    request.addHeader("envName", "Production");
    request.addHeader("configType", "Server");
    request.addHeader("configName", "serverconfig");
    request.addHeader("oldName", "test");
    request.addHeader("propName", "port");
    request.addHeader("X-File-Name", "test.zip");
    HttpServletRequest httpServletRequest = (HttpServletRequest) request;
    Response uploadFile = configurationService.uploadFile(httpServletRequest);
    Assert.assertEquals(200, uploadFile.getStatus());

    MockHttpServletRequest request1 = new MockHttpServletRequest();
    request1.addHeader("appDirName", appDirName);
    request1.addHeader("actionType", "editconfig");
    request1.addHeader("envName", "Production");
    request1.addHeader("configType", "Server");
    request1.addHeader("configName", "serverconfig");
    request1.addHeader("oldName", "");
    request1.addHeader("propName", "port");
    request1.addHeader("X-File-Name", "test.zip");
    HttpServletRequest httpServletRequest1 = (HttpServletRequest) request1;
    Response uploadFile1 = configurationService.uploadFile(httpServletRequest1);
    Assert.assertEquals(200, uploadFile1.getStatus());

    request1.addHeader("oldName", "serverconfig");
    HttpServletRequest httpServletRequest2 = (HttpServletRequest) request1;
    Response uploadFile2 = configurationService.uploadFile(httpServletRequest2);
    Assert.assertEquals(200, uploadFile2.getStatus());

}

From source file:org.fenixedu.bennu.oauth.OAuthServletTest.java

@Test
public void getServiceAccessTokenHeaderEmptyTest() {
    MockHttpServletRequest req = new MockHttpServletRequest();
    MockHttpServletResponse res = new MockHttpServletResponse();
    Authenticate.unmock();//  w w  w  .  j  a  va2 s. com
    String clientSecret = "";

    req.addHeader(HttpHeaders.AUTHORIZATION,
            "Basic " + Base64.getEncoder().encodeToString(clientSecret.getBytes(StandardCharsets.UTF_8)));
    req.addParameter(GRANT_TYPE, GRANT_TYPE_CLIENT_CREDENTIALS);
    req.setMethod("POST");
    req.setPathInfo("/access_token");

    try {
        oauthServlet.service(req, res);
        Assert.assertEquals("must return BAD_REQUEST", 400, res.getStatus());

    } catch (ServletException | IOException e) {
        Assert.fail(e.getMessage());
    }
}

From source file:org.fao.geonet.api.records.formatters.FormatterApiIntegrationTest.java

@Test
public void testLastModified() throws Exception {
    String stage = systemInfo.getStagingProfile();
    systemInfo.setStagingProfile(SystemInfo.STAGE_PRODUCTION);
    try {//w  w w .ja v  a2  s.com
        metadataRepository.update(id, new Updater<Metadata>() {
            @Override
            public void apply(@Nonnull Metadata entity) {
                entity.getDataInfo().setChangeDate(new ISODate("2012-01-18T15:04:43"));
            }
        });
        dataManager.indexMetadata(Lists.newArrayList("" + this.id));

        final String formatterName = "full_view";

        MockHttpServletRequest request = new MockHttpServletRequest();
        request.getSession();
        request.addParameter("h2IdentInfo", "true");

        MockHttpServletResponse response = new MockHttpServletResponse();
        formatService.exec("eng", "html", "" + id, null, formatterName, "true", false, _100,
                new ServletWebRequest(request, response));
        final String lastModified = response.getHeader("Last-Modified");
        assertEquals("no-cache", response.getHeader("Cache-Control"));
        final String viewString = response.getContentAsString();
        assertNotNull(viewString);

        request = new MockHttpServletRequest();
        request.getSession();
        request.setMethod("GET");
        response = new MockHttpServletResponse();

        request.addHeader("If-Modified-Since", lastModified);
        formatService.exec("eng", "html", "" + id, null, formatterName, "true", false, _100,
                new ServletWebRequest(request, response));
        assertEquals(HttpStatus.SC_NOT_MODIFIED, response.getStatus());
        final ISODate newChangeDate = new ISODate();
        metadataRepository.update(id, new Updater<Metadata>() {
            @Override
            public void apply(@Nonnull Metadata entity) {
                entity.getDataInfo().setChangeDate(newChangeDate);
            }
        });

        dataManager.indexMetadata(Lists.newArrayList("" + this.id));

        request = new MockHttpServletRequest();
        request.getSession();
        request.setMethod("GET");
        response = new MockHttpServletResponse();

        request.addHeader("If-Modified-Since", lastModified);
        formatService.exec("eng", "html", "" + id, null, formatterName, "true", false, _100,
                new ServletWebRequest(request, response));
        assertEquals(HttpStatus.SC_OK, response.getStatus());
    } finally {
        systemInfo.setStagingProfile(stage);
    }
}

From source file:fragment.web.AbstractBaseControllerTest.java

@Test
public void testHandleGenericException() {
    Exception ex = new Exception("generic");
    MockHttpServletRequest request = new MockHttpServletRequest();
    ModelAndView mav = controller.handleGenericException(ex, request);
    Assert.assertNotNull(mav.getModelMap());
    ModelAndViewAssert.assertViewName(mav, "errors/error");
    ModelAndViewAssert.assertModelAttributeAvailable(mav, "errorMessage");
    ModelAndViewAssert.assertModelAttributeValue(mav, "errorMessage",
            messageSource.getMessage("ui.error.page.server.error.message", null, null));

    ex = new NullPointerException("NPE generic");
    mav = controller.handleGenericException(ex, request);
    Assert.assertNotNull(mav.getModelMap());
    ModelAndViewAssert.assertViewName(mav, "errors/error");
    ModelAndViewAssert.assertModelAttributeAvailable(mav, "errorMessage");
    ModelAndViewAssert.assertModelAttributeValue(mav, "errorMessage",
            messageSource.getMessage("ui.error.page.server.error.message", null, null));

    request.addHeader("X-Requested-With", "XMLHttpRequest");
    mav = controller.handleGenericException(ex, request);
    Assert.assertNotNull(mav.getModelMap());
    ModelAndViewAssert.assertViewName(mav, "errors.messagepage");
    ModelAndViewAssert.assertModelAttributeAvailable(mav, "errorMessage");
    ModelAndViewAssert.assertModelAttributeValue(mav, "errorMessage",
            messageSource.getMessage("ui.error.page.server.error.message", null, null));

    ex = new NullPointerException("NPE generic");
    mav = controller.handleGenericException(ex, request);
    Assert.assertNotNull(mav.getModelMap());
    ModelAndViewAssert.assertViewName(mav, "errors.messagepage");
    ModelAndViewAssert.assertModelAttributeAvailable(mav, "errorMessage");
    ModelAndViewAssert.assertModelAttributeValue(mav, "errorMessage",
            messageSource.getMessage("ui.error.page.server.error.message", null, null));

}

From source file:org.fenixedu.bennu.oauth.OAuthServletTest.java

@Test
public void getServiceAccessTokenHeaderTest() {
    MockHttpServletRequest req = new MockHttpServletRequest();
    MockHttpServletResponse res = new MockHttpServletResponse();
    Authenticate.unmock();// w ww  .  j  a va 2s  .co m
    String clientSecret = serviceApplication.getExternalId() + ":" + serviceApplication.getSecret();
    req.addHeader(HttpHeaders.AUTHORIZATION,
            "Basic " + Base64.getEncoder().encodeToString(clientSecret.getBytes(StandardCharsets.UTF_8)));
    req.addParameter(GRANT_TYPE, GRANT_TYPE_CLIENT_CREDENTIALS);
    req.setMethod("POST");
    req.setPathInfo("/access_token");

    try {
        oauthServlet.service(req, res);

        Assert.assertEquals("must return status OK", 200, res.getStatus());

        String tokenJson = res.getContentAsString();

        final JsonObject token = new JsonParser().parse(tokenJson).getAsJsonObject();

        Assert.assertTrue("response must be a valid json and have access_token field",
                token.has(ACCESS_TOKEN) && token.get(ACCESS_TOKEN).getAsString().length() > 0);

    } catch (ServletException | IOException e) {
        Assert.fail(e.getMessage());
    }
}

From source file:org.openmrs.module.atomfeed.web.AtomFeedDownloadServletTest.java

/**
 * @see AtomFeedDownloadServlet#doHead(javax.servlet.http.HttpServletRequest,
 *      javax.servlet.http.HttpServletResponse)
 * @verifies send not modified error if atom feed has not changed
 *//*from w w w  .  j  a va 2 s.c o m*/
@Test
public void doHead_shouldSendNotModifiedErrorIfAtomFeedHasNotChanged() throws Exception {
    // create servlet and corresponding request and response object to be sent
    AtomFeedDownloadServlet atomFeedDownloadServlet = new AtomFeedDownloadServlet();
    MockHttpServletRequest request = new MockHttpServletRequest("HEAD", "/atomfeed");
    request.setContextPath("/somecontextpath");
    MockHttpServletResponse response = new MockHttpServletResponse();

    // intentionally change atom feed in order to not depend from other tests
    AtomFeedUtil.objectCreated(new Encounter());

    // read atom feed header specific information from the header file
    String headerFileContent = AtomFeedUtil.readFeedHeaderFile();
    int contentLength = 0;
    String etagToken = "";
    Date lastModified = null;
    if (StringUtils.isNotBlank(headerFileContent)) {
        contentLength = headerFileContent.length() + Integer
                .valueOf(StringUtils.substringBetween(headerFileContent, "<entriesSize>", "</entriesSize>"));
        etagToken = StringUtils.substringBetween(headerFileContent, "<versionId>", "</versionId>");
        try {
            lastModified = new SimpleDateFormat(AtomFeedUtil.RFC_3339_DATE_FORMAT)
                    .parse(StringUtils.substringBetween(headerFileContent, "<updated>", "</updated>"));
        } catch (ParseException e) {
            // ignore it here
        }
    }
    // set request headers 
    request.addHeader("If-None-Match", '"' + etagToken + '"');
    request.addHeader("If-Modified-Since", lastModified);

    atomFeedDownloadServlet.service(request, response);
    // check response headers
    Assert.assertEquals(contentLength, response.getContentLength());
    Assert.assertEquals("application/atom+xml", response.getContentType());
    Assert.assertEquals(HttpServletResponse.SC_NOT_MODIFIED, response.getStatus());
    Assert.assertNotNull(response.getHeader("Last-Modified"));
}