com.gargoylesoftware.htmlunit.WebResponseData Java Examples

/**
 * @throws Exception if the test fails
 */
@Test
public void wrapper() throws Exception {
    final List<NameValuePair> emptyList = Collections.emptyList();
    final WebResponseData data = new WebResponseData(new byte[]{}, HttpStatus.SC_OK, "", emptyList);
    final WebResponse response = new WebResponse(data, URL_FIRST, HttpMethod.GET, 0);
    final WebRequest wrs = new WebRequest(URL_FIRST);

    final WebConnection realConnection = new WebConnection() {
        @Override
        public WebResponse getResponse(final WebRequest request) {
            assertSame(wrs, request);
            return response;
        }
        @Override
        public void close() {
            // nothing
        }
    };

    try (WebConnectionWrapper wrapper = new WebConnectionWrapper(realConnection)) {
        assertSame(response, wrapper.getResponse(wrs));
    }
}
 

private WebResponseData webResponseData() throws IOException {
	List<NameValuePair> responseHeaders = responseHeaders();
	int statusCode = (this.response.getRedirectedUrl() != null ?
			HttpStatus.MOVED_PERMANENTLY.value() : this.response.getStatus());
	String statusMessage = statusMessage(statusCode);
	return new WebResponseData(this.response.getContentAsByteArray(), statusCode, statusMessage, responseHeaders);
}
 

@Before
public void setup() throws Exception {
	request = new WebRequest(new URL("http://localhost/"));
	WebResponseData data = new WebResponseData("".getBytes("UTF-8"), 200, "", Collections.emptyList());
	expectedResponse = new WebResponse(data, request, 100L);
	webConnection = new DelegatingWebConnection(defaultConnection,
			new DelegateWebConnection(matcher1, connection1), new DelegateWebConnection(matcher2, connection2));
}
 

private WebResponseData webResponseData() throws IOException {
	List<NameValuePair> responseHeaders = responseHeaders();
	int statusCode = (this.response.getRedirectedUrl() != null ?
			HttpStatus.MOVED_PERMANENTLY.value() : this.response.getStatus());
	String statusMessage = statusMessage(statusCode);
	return new WebResponseData(this.response.getContentAsByteArray(), statusCode, statusMessage, responseHeaders);
}
 

@Before
public void setup() throws Exception {
	request = new WebRequest(new URL("http://localhost/"));
	WebResponseData data = new WebResponseData("".getBytes("UTF-8"), 200, "", Collections.emptyList());
	expectedResponse = new WebResponse(data, request, 100L);
	webConnection = new DelegatingWebConnection(defaultConnection,
			new DelegateWebConnection(matcher1, connection1), new DelegateWebConnection(matcher2, connection2));
}
 

/**
 * Builds a WebResponse with new content, preserving all other information.
 * @param wr the web response to adapt
 * @param newContent the new content to place in the response
 * @return a web response with the new content
 * @throws IOException if an encoding problem occurred
 */
protected WebResponse replaceContent(final WebResponse wr, final String newContent) throws IOException {
    final byte[] body = newContent.getBytes(wr.getContentCharset());
    final WebResponseData wrd = new WebResponseData(body, wr.getStatusCode(), wr.getStatusMessage(),
        wr.getResponseHeaders());
    return new WebResponse(wrd, wr.getWebRequest().getUrl(), wr.getWebRequest().getHttpMethod(),
            wr.getLoadTime());
}
 

/**
 * Builds a WebResponse with new content, preserving all other information.
 * @param wr the web response to adapt
 * @param newContent the new content to place in the response
 * @return a web response with the new content
 * @throws IOException if an encoding problem occurred
 */
protected WebResponse replaceContent(final WebResponse wr, final String newContent) throws IOException {
    final byte[] body = newContent.getBytes(wr.getContentCharset());
    final WebResponseData wrd = new WebResponseData(body, wr.getStatusCode(), wr.getStatusMessage(),
        wr.getResponseHeaders());
    return new WebResponse(wrd, wr.getWebRequest().getUrl(), wr.getWebRequest().getHttpMethod(),
            wr.getLoadTime());
}
 

private WebResponseData webResponseData() throws IOException {
	List<NameValuePair> responseHeaders = responseHeaders();
	int statusCode = (this.response.getRedirectedUrl() != null ?
			HttpStatus.MOVED_PERMANENTLY.value() : this.response.getStatus());
	String statusMessage = statusMessage(statusCode);
	return new WebResponseData(this.response.getContentAsByteArray(), statusCode, statusMessage, responseHeaders);
}
 

@Before
public void setUp() throws Exception {
	request = new WebRequest(new URL("http://localhost/"));
	WebResponseData data = new WebResponseData("".getBytes("UTF-8"), 200, "", Collections.<NameValuePair> emptyList());
	expectedResponse = new WebResponse(data, request, 100L);
	webConnection = new DelegatingWebConnection(defaultConnection,
		new DelegateWebConnection(matcher1, connection1), new DelegateWebConnection(matcher2, connection2));
}
 

public WebResponse build() throws IOException {
	WebResponseData webResponseData = webResponseData();
	long endTime = System.currentTimeMillis();
	return new WebResponse(webResponseData, this.webRequest, endTime - this.startTime);
}
 

public WebResponse build() throws IOException {
	WebResponseData webResponseData = webResponseData();
	long endTime = System.currentTimeMillis();
	return new WebResponse(webResponseData, this.webRequest, endTime - this.startTime);
}
 

public WebResponse build() throws IOException {
	WebResponseData webResponseData = webResponseData();
	long endTime = System.currentTimeMillis();
	return new WebResponse(webResponseData, webRequest, endTime - startTime);
}
 

/**
 * Creates a faked WebResponse for the request with the provided content.
 * @param wr the web request for which a response should be created
 * @param content the content to place in the response
 * @param contentType the content type of the response
 * @param responseCode the HTTP code for the response
 * @param responseMessage the HTTP message for the response
 * @return a web response with the provided content
 * @throws IOException if an encoding problem occurred
 */
protected WebResponse createWebResponse(final WebRequest wr, final String content,
        final String contentType, final int responseCode, final String responseMessage) throws IOException {
    final List<NameValuePair> headers = new ArrayList<>();
    headers.add(new NameValuePair(HttpHeader.CONTENT_TYPE_LC, contentType + "; charset=" + UTF_8));
    final byte[] body = content.getBytes(UTF_8);
    final WebResponseData wrd = new WebResponseData(body, responseCode, responseMessage, headers);
    return new WebResponse(wrd, wr.getUrl(), wr.getHttpMethod(), 0);
}
 

/**
 * Creates a faked WebResponse for the request with the provided content.
 * @param wr the web request for which a response should be created
 * @param content the content to place in the response
 * @param contentType the content type of the response
 * @param responseCode the HTTP code for the response
 * @param responseMessage the HTTP message for the response
 * @return a web response with the provided content
 * @throws IOException if an encoding problem occurred
 */
protected WebResponse createWebResponse(final WebRequest wr, final String content,
        final String contentType, final int responseCode, final String responseMessage) throws IOException {
    final List<NameValuePair> headers = new ArrayList<>();
    headers.add(new NameValuePair(HttpHeader.CONTENT_TYPE_LC, contentType + "; charset=" + UTF_8));
    final byte[] body = content.getBytes(UTF_8);
    final WebResponseData wrd = new WebResponseData(body, responseCode, responseMessage, headers);
    return new WebResponse(wrd, wr.getUrl(), wr.getHttpMethod(), 0);
}