Java Code Examples for org.springframework.mock.web.MockHttpServletResponse#getHeaderNames()

@Test
public void getUser() throws Exception {

    MvcResult result= mockMvc.perform(get("/users/update/{id}/", 1))
            .andExpect(status().isOk())
            .andExpect(view().name("userForm"))
            .andExpect(MockMvcResultMatchers.model().attributeExists("action"))
            .andExpect(model().attribute("user", hasProperty("id", is(1L))))
            .andExpect(model().attribute("user", hasProperty("name", is("Sergey"))))
            .andExpect(model().attribute("user", hasProperty("age", is(24))))
            .andExpect(model().attribute("user", hasProperty("birthday", is("1994-01-01"))))
            .andReturn();


    MockHttpServletResponse mockResponse=result.getResponse();
    assertThat(mockResponse.getContentType()).isEqualTo("text/html;charset=UTF-8");

    Collection<String> responseHeaders = mockResponse.getHeaderNames();
    assertNotNull(responseHeaders);
    assertEquals(2, responseHeaders.size());
    assertEquals("Check for Content-Type header", "Accept-Language", responseHeaders.iterator().next());
    String responseAsString=mockResponse.getContentAsString();
    assertTrue(responseAsString.contains("用户管理"));
}
 

protected final HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		headers.put(name, response.getHeaders(name));
	}
	return headers;
}
 

private HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		List<String> values = response.getHeaders(name);
		for (String value : values) {
			headers.add(name, value);
		}
	}
	return headers;
}
 

protected final HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		headers.put(name, response.getHeaders(name));
	}
	return headers;
}
 

private HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		List<String> values = response.getHeaders(name);
		for (String value : values) {
			headers.add(name, value);
		}
	}
	return headers;
}
 

private static void logResponseHeaders(StringBuilder sb, int id, MockHttpServletResponse response) {
    Set<String> keys = new TreeSet(response.getHeaderNames());
    for (String key : keys) {
        List<String> entries = response.getHeaders(key);
        sb.append(id).append(' ').append('<').append(' ')
                .append(key).append(": ").append(entries.size() == 1 ? entries.get(0) : entries).append('\n');
    }
}
 

protected final HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		headers.put(name, response.getHeaders(name));
	}
	return headers;
}
 

private HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		List<String> values = response.getHeaders(name);
		for (String value : values) {
			headers.add(name, value);
		}
	}
	return headers;
}
 

@Override
  public void messageReceived(ChannelHandlerContext ctx, HttpRequest request) throws Exception {

      if (!request.getDecoderResult().isSuccess()) {
          sendError(ctx, BAD_REQUEST);
          return;
      }

MockHttpServletRequest servletRequest = createServletRequest(request);
      MockHttpServletResponse servletResponse = new MockHttpServletResponse();

this.servlet.service(servletRequest, servletResponse);

      HttpResponseStatus status = HttpResponseStatus.valueOf(servletResponse.getStatus());
HttpResponse response = new DefaultHttpResponse(HTTP_1_1, status);

for (String name : servletResponse.getHeaderNames()) {
	for (Object value : servletResponse.getHeaderValues(name)) {
		response.addHeader(name, value);
	}
}

      // Write the initial line and the header.
      ctx.write(response);

      InputStream contentStream = new ByteArrayInputStream(servletResponse.getContentAsByteArray());

// Write the content.
      ChannelFuture writeFuture = ctx.write(new ChunkedStream(contentStream));
      writeFuture.addListener(ChannelFutureListener.CLOSE);
  }
 

@Override
protected void channelRead0(ChannelHandlerContext channelHandlerContext, FullHttpRequest fullHttpRequest) throws Exception {
	if (!fullHttpRequest.getDecoderResult().isSuccess()) {
		sendError(channelHandlerContext, BAD_REQUEST);
		return;
	}

	MockHttpServletRequest servletRequest = createHttpServletRequest(fullHttpRequest);
	MockHttpServletResponse servletResponse = new MockHttpServletResponse();

	this.servlet.service(servletRequest, servletResponse);

	HttpResponseStatus status = HttpResponseStatus.valueOf(servletResponse.getStatus());
	HttpResponse response = new DefaultHttpResponse(HTTP_1_1, status);
	
	for (String name : servletResponse.getHeaderNames()) {
		for (String value : servletResponse.getHeaders(name)) {
			response.headers().add(name, value);
		}
	}

	// Write the initial line and the header.
	channelHandlerContext.write(response);
	InputStream contentStream = new ByteArrayInputStream(servletResponse.getContentAsByteArray());

	// Write the content and flush it.
	ChannelFuture writeFuture = channelHandlerContext.writeAndFlush(new ChunkedStream(contentStream));
	writeFuture.addListener(ChannelFutureListener.CLOSE);
}
 

private HttpHeaders getResponseHeaders(MockHttpServletResponse response) {
	HttpHeaders headers = new HttpHeaders();
	for (String name : response.getHeaderNames()) {
		List<String> values = response.getHeaders(name);
		for (String value : values) {
			headers.add(name, value);
		}
	}
	return headers;
}
 

public HttpResponse execute(HttpRequestBase rq) throws IOException {
  URI uri = rq.getURI();
  DapController controller = getController(uri);
  StandaloneMockMvcBuilder mvcbuilder = MockMvcBuilders.standaloneSetup(controller);
  mvcbuilder.setValidator(new TestServlet.NullValidator());
  MockMvc mockMvc = mvcbuilder.build();
  MockHttpServletRequestBuilder mockrb = MockMvcRequestBuilders.get(uri);
  // We need to use only the path part
  mockrb.servletPath(uri.getPath());
  // Move any headers from rq to mockrb
  Header[] headers = rq.getAllHeaders();
  for (int i = 0; i < headers.length; i++) {
    Header h = headers[i];
    mockrb.header(h.getName(), h.getValue());
  }
  // Since the url has the query parameters,
  // they will automatically be parsed and added
  // to the rb parameters.

  // Finally set the resource dir
  mockrb.requestAttr("RESOURCEDIR", this.resourcepath);

  // Now invoke the servlet
  MvcResult result;
  try {
    result = mockMvc.perform(mockrb).andReturn();
  } catch (Exception e) {
    throw new IOException(e);
  }

  // Collect the output
  MockHttpServletResponse res = result.getResponse();
  byte[] byteresult = res.getContentAsByteArray();

  // Convert to HttpResponse
  HttpResponse response = new BasicHttpResponse(HttpVersion.HTTP_1_1, res.getStatus(), "");
  if (response == null)
    throw new IOException("HTTPMethod.executeMock: Response was null");
  Collection<String> keys = res.getHeaderNames();
  // Move headers to the response
  for (String key : keys) {
    List<String> values = res.getHeaders(key);
    for (String v : values) {
      response.addHeader(key, v);
    }
  }
  ByteArrayEntity entity = new ByteArrayEntity(byteresult);
  String sct = res.getContentType();
  entity.setContentType(sct);
  response.setEntity(entity);
  return response;
}
 

@Override
protected HttpResponse makeHttpRequest(HttpBody entity, ScenarioContext context) {
    logger.info("making mock http client request: {} - {}", request.getMethod(), getRequestUri());
    MockHttpServletRequest req = requestBuilder.buildRequest(getServletContext());
    byte[] bytes;
    if (entity != null) {
        bytes = entity.getBytes();
        req.setContentType(entity.getContentType());
        if (entity.isMultiPart()) {
            for (MultiPartItem item : entity.getParts()) {
                MockMultiPart part = new MockMultiPart(item);
                req.addPart(part);
                if (!part.isFile()) {
                    req.addParameter(part.getName(), part.getValue());
                }
            }
        } else if (entity.isUrlEncoded()) {
            req.addParameters(entity.getParameters());
        } else {
            req.setContent(bytes);
        }
    } else {
        bytes = null;
    }
    MockHttpServletResponse res = new MockHttpServletResponse();
    logRequest(req, bytes);
    long startTime = System.currentTimeMillis();
    try {
        getServlet(request).service(req, res);
    } catch (Exception e) {
        String message = e.getMessage();
        if (message == null && e.getCause() != null) {
            message = e.getCause().getMessage();
        }
        logger.error("mock servlet request failed: {}", message);
        throw new RuntimeException(e);
    }
    HttpResponse response = new HttpResponse(startTime, System.currentTimeMillis());
    bytes = res.getContentAsByteArray();
    logResponse(res, bytes);
    response.setUri(getRequestUri());
    response.setBody(bytes);
    response.setStatus(res.getStatus());
    for (Cookie c : res.getCookies()) {
        com.intuit.karate.http.Cookie cookie = new com.intuit.karate.http.Cookie(c.getName(), c.getValue());
        cookie.put(DOMAIN, c.getDomain());
        cookie.put(PATH, c.getPath());
        cookie.put(SECURE, c.getSecure() + "");
        cookie.put(MAX_AGE, c.getMaxAge() + "");
        cookie.put(VERSION, c.getVersion() + "");
        response.addCookie(cookie);
    }
    for (String headerName : res.getHeaderNames()) {
        response.putHeader(headerName, res.getHeaders(headerName));
    }
    return response;
}
 

private void addResponseHeaders(ResponseDefinitionBuilder definition,
		MockHttpServletResponse input) {
	for (String name : input.getHeaderNames()) {
		definition.withHeader(name, input.getHeader(name));
	}
}