Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.jspringbot.keyword.http.HTTPHelper Maven / Gradle / Ivy
/*
* Copyright (c) 2012. JSpringBot. All Rights Reserved.
*
* See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The JSpringBot licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jspringbot.keyword.http;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.http.*;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.AuthCache;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpDelete;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.protocol.ClientContext;
import org.apache.http.client.utils.URLEncodedUtils;
import org.apache.http.cookie.Cookie;
import org.apache.http.entity.StringEntity;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.impl.auth.BasicScheme;
import org.apache.http.impl.client.AbstractHttpClient;
import org.apache.http.impl.client.BasicAuthCache;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.BasicHttpContext;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
import org.jspringbot.keyword.expression.ELUtils;
import org.jspringbot.keyword.json.JSONHelper;
import org.jspringbot.keyword.json.JSONUtils;
import org.jspringbot.keyword.xml.XMLHelper;
import org.jspringbot.keyword.xml.XMLUtils;
import org.jspringbot.syntax.HighlightRobotLogger;
import org.springframework.expression.common.ExpressionUtils;
import org.xml.sax.SAXException;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.LinkedList;
import java.util.List;
public class HTTPHelper {
public static final HighlightRobotLogger LOG = HighlightRobotLogger.getLogger(HTTPHelper.class);
public static final String GET_METHOD = "GET";
public static final String POST_METHOD = "POST";
public static final String PUT_METHOD = "PUT";
public static final String DELETE_METHOD = "DELETE";
public static final String ENCODING_UTF_8 = "utf-8";
public static final String USER_AGENT = "User-Agent";
protected AbstractHttpClient client;
protected HttpContext context;
protected HttpRequest request;
protected HttpResponse response;
protected String responseString;
protected StatusLine status;
protected HttpEntity responseEntity;
protected HttpHost targetHost;
protected List params;
protected List headers;
protected URI uri;
protected JSONHelper jsonHelper;
protected XMLHelper xmlHelper;
protected BasicCookieStore cookieStore;
protected boolean useUserAgent;
protected String userAgent;
protected MultipartEntityBuilder builder;
public HTTPHelper(AbstractHttpClient client) {
this.client = client;
newSession();
}
public void setJsonHelper(JSONHelper jsonHelper) {
this.jsonHelper = jsonHelper;
}
public void setXmlHelper(XMLHelper xmlHelper) {
this.xmlHelper = xmlHelper;
}
public void setUserAgentString(String userAgent) {
this.userAgent = userAgent;
}
public void setUserAgentFlag(boolean userAgentFlag) {
this.useUserAgent = userAgentFlag;
}
public void consume() {
try {
if (responseEntity != null) {
EntityUtils.consume(responseEntity);
}
} catch (IOException e) {
e.printStackTrace();
}
}
public void reset() {
if(xmlHelper != null) {
xmlHelper.reset();
}
if(jsonHelper != null) {
jsonHelper.reset();
}
consume();
params = new LinkedList();
headers = new LinkedList();
status = null;
responseString = null;
responseEntity = null;
response = null;
}
public void newSession() {
LOG.info("Created New HTTP Session.");
this.context = new BasicHttpContext();
cookieStore = new BasicCookieStore();
context.setAttribute(ClientContext.COOKIE_STORE, cookieStore);
reset();
}
public void createRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, GET_METHOD);
}
public void createGetRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, GET_METHOD);
}
public void createPostRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, POST_METHOD);
}
public void createFileUploadRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, POST_METHOD);
builder = MultipartEntityBuilder.create();
}
public void createPutRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, PUT_METHOD);
}
public void createDeleteRequest(String url) throws MalformedURLException, URISyntaxException {
createRequest(url, DELETE_METHOD);
}
public void createRequest(String paramUrl, String method) throws MalformedURLException, URISyntaxException {
uri = new URL(paramUrl).toURI();
targetHost = new HttpHost(uri.getHost(), uri.getPort(), uri.getScheme());
String uriPath = uri.getPath();
if(uri.getQuery() != null) {
uriPath += "?" + uri.getQuery();
}
LOG.keywordAppender()
.appendFullArgument("URL", paramUrl)
.appendFullArgument("Path", uri.getPath())
.appendFullArgument("Query String", uri.getQuery())
.appendArgument("Method", method);
if (method.equalsIgnoreCase(POST_METHOD)) {
request = new HttpPost(uriPath);
} else if (method.equalsIgnoreCase(GET_METHOD)) {
request = new HttpGet(uriPath);
} else if (method.equalsIgnoreCase(PUT_METHOD)) {
request = new HttpPut(uriPath);
} else if (method.equalsIgnoreCase(DELETE_METHOD)) {
request = new HttpDelete(uriPath);
} else {
throw new IllegalArgumentException(String.format("Unknown http method '%s' for url '%s'.", method, paramUrl));
}
if (this.useUserAgent) {
request.setHeader("User-Agent", this.userAgent);
}
reset();
}
public void addRequestHeader(String name, String value) {
LOG.keywordAppender()
.appendArgument("Name", name)
.appendArgument("Value", value);
headers.add(new BasicNameValuePair(name, value));
}
public void addRequestParameter(String name, String value) {
LOG.keywordAppender()
.appendArgument("Name", name)
.appendArgument("Value", value);
params.add(new BasicNameValuePair(name, value));
}
public void addFileParameter(String name, File file) {
LOG.keywordAppender()
.appendArgument("Name", name)
.appendArgument("File", file.getAbsolutePath());
if(builder == null) {
throw new IllegalStateException("Not setup for file upload.");
}
builder.addPart(name, new FileBody(file));
}
public void setRequestBody(String stringBody) throws UnsupportedEncodingException {
if(!HttpEntityEnclosingRequest.class.isInstance(request)) {
throw new IllegalArgumentException("Does not support string entity.");
}
stringBody = StringUtils.trimToEmpty(stringBody);
/*if(JSONUtils.isJSONValid(stringBody)) {
LOG.keywordAppender().appendJSON(JSONHelper.prettyPrint(stringBody));
} else if(XMLUtils.isValidXML(stringBody)) {
LOG.keywordAppender().appendXML(JSONHelper.prettyPrint(stringBody));
} else {
LOG.keywordAppender().appendCode(stringBody);
} */
((HttpEntityEnclosingRequest) request).setEntity(new StringEntity(stringBody, ENCODING_UTF_8));
}
public void setBasicAuthentication(String username, String password) {
UsernamePasswordCredentials credentials = new UsernamePasswordCredentials(username, password);
client.getCredentialsProvider().setCredentials(new AuthScope(targetHost.getHostName(), targetHost.getPort()), credentials);
// Create AuthCache instance
AuthCache authCache = new BasicAuthCache();
// Generate BASIC scheme object and add it to the local auth cache
BasicScheme basicAuth = new BasicScheme();
authCache.put(targetHost, basicAuth);
LOG.keywordAppender()
.appendArgument("Username", username)
.appendArgument("Password", password);
context.setAttribute(ClientContext.AUTH_CACHE, authCache);
}
private void addRequestHeadersToHttpRequest() {
for(NameValuePair nv : headers) {
String name = nv.getName();
String value = nv.getValue();
if (name.equalsIgnoreCase(USER_AGENT)) {
request.setHeader(USER_AGENT, value);
return;
}
request.addHeader(name, value);
}
}
public void invokeRequest() throws IOException {
String actualUri = uri.getPath();
if (uri.getQuery() != null) {
actualUri += "?" + uri.getQuery();
}
if (CollectionUtils.isNotEmpty(params)) {
if (request instanceof HttpPost) {
if(builder == null) {
((HttpPost) request).setEntity(new UrlEncodedFormEntity(params, ENCODING_UTF_8));
} else {
for(NameValuePair nv : params) {
builder.addTextBody(nv.getName(), nv.getValue());
}
((HttpPost) request).setEntity(builder.build());
}
} else {
String queryString = URLEncodedUtils.format(params, ENCODING_UTF_8);
if(actualUri.contains("?")) {
actualUri += "&" + queryString;
} else {
actualUri += "?" + queryString;
}
if (request instanceof HttpGet) {
request = new HttpGet(actualUri);
} else if (request instanceof HttpPut) {
request = new HttpPut(actualUri);
} else if (request instanceof HttpDelete) {
request = new HttpDelete(actualUri);
}
}
}
addRequestHeadersToHttpRequest();
LOG.keywordAppender().appendArgument("Request Line", String.valueOf(request.getRequestLine()));
response = client.execute(targetHost, request, context);
status = response.getStatusLine();
responseEntity = response.getEntity();
// request headers
LOG.keywordAppender().createPath("Request Headers:");
for(Header header : request.getAllHeaders()) {
LOG.keywordAppender().appendProperty(header.getName(), header.getValue());
}
LOG.keywordAppender().endPath();
// request parameters for post
if (request instanceof HttpPost && CollectionUtils.isNotEmpty(params)) {
LOG.keywordAppender().createPath("Request Parameters:");
for(NameValuePair nv : params) {
LOG.keywordAppender().appendProperty(nv.getName(), nv.getValue());
}
LOG.keywordAppender().endPath();
}
LOG.keywordAppender().appendArgument("Status", String.valueOf(status));
// response headers
LOG.keywordAppender().createPath("Response Headers:");
for(Header header : response.getAllHeaders()) {
LOG.keywordAppender().appendProperty(header.getName(), header.getValue());
}
LOG.keywordAppender().endPath();
}
public String getResponseHeader(String name) {
String value = response.getFirstHeader(name).getValue();
LOG.keywordAppender()
.appendArgument("Name", name)
.appendArgument("Value", value);
return value;
}
public String getCookieValue(String name) {
List cookies = cookieStore.getCookies();
String cookieValue = null;
for(Cookie cookie : cookies) {
if(StringUtils.equals(cookie.getName(), name)) {
cookieValue = cookie.getValue();
break;
}
}
LOG.keywordAppender()
.appendArgument("name", name)
.appendArgument("value", cookieValue);
return cookieValue;
}
public int getResponseStatusCode() {
return response.getStatusLine().getStatusCode();
}
public void responseShouldBeXML () throws IOException {
try {
if(responseString == null) {
responseString = EntityUtils.toString(responseEntity);
if (xmlHelper == null) {
throw new IllegalStateException("Json checking is not supported");
}
xmlHelper.setXmlString(responseString);
consume();
}
} catch (SAXException e) {
throw new IOException(e);
}
}
public void responseShouldBeJson () throws IOException {
if(responseString == null) {
responseString = EntityUtils.toString(responseEntity);
if (jsonHelper == null) {
throw new IllegalStateException("Json checking is not supported");
}
jsonHelper.setJsonString(responseString);
consume();
}
}
public String getResponseString() throws IOException {
if(responseString == null) {
responseString = EntityUtils.toString(responseEntity);
consume();
}
return responseString;
}
public String captureResponseRegex(String regex, int... groups) throws IOException {
getResponseString();
return ELUtils.regexCapture(responseString, regex, groups);
}
public void responseStatusCodeShouldBeEqualTo(int statusCode) {
LOG.keywordAppender()
.appendArgument("Expected Status Code", statusCode)
.appendArgument("Actual Status Code", status.getStatusCode());
if(status.getStatusCode() != statusCode) {
throw new IllegalStateException(String.format("Expecting response status '%d' but was '%s'.", statusCode, status.toString()));
}
}
public void responseShouldContain(String value) throws IOException {
LOG.keywordAppender().appendArgument("Text", value);
if(responseString == null) {
responseString = EntityUtils.toString(responseEntity);
LOG.keywordAppender().appendCode(responseString);
consume();
}
if(!responseString.contains(value)) {
throw new IllegalStateException(String.format("Expecting response should contain value '%s'.", value));
}
}
}
