arjuna.lib.httpclient.BasicRestClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arjuna-java Show documentation

Arjuna-Java is the client implementation in Java for development of test automation using Arjuna. It uses TestNG as the test engine. With minor tweaks, it can be used with any other test engine or custom test automation implementations. Arjuna is a Python based test automation framework developed by Rahul Verma (www.rahulverma.net)

The newest version!

package arjuna.lib.httpclient;

import java.nio.charset.Charset;

import org.apache.http.HttpEntity;
import org.apache.http.HttpHost;
import org.apache.http.HttpResponse;
import org.apache.http.client.ResponseHandler;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicHeader;
import org.apache.http.protocol.HTTP;
import org.apache.http.util.EntityUtils;

import arjuna.tpi.Arjuna;

public class BasicRestClient {
	private String baseUrl = null;
	private HttpHost proxy = null;
	private CloseableHttpClient httpclient = null;
	private ResponseHandler responseHandler;
	
	public BasicRestClient(String baseUrl) {
		httpclient = HttpClients.createDefault();
		this.baseUrl = baseUrl;	
	}
	
	private String handleResponse(HttpResponse response) throws Exception{
        int status = response.getStatusLine().getStatusCode();
        HttpEntity entity = response.getEntity();
        if (status >= 200 && status < 300) {
            return entity != null ? EntityUtils.toString(entity) : null;		
        } else {
        	if (entity == null) {
        		throw new SetuHttpException("Setu returned null response object.", status, EntityUtils.toString(entity));
        	} else {
        		throw new SetuHttpException("Setu returned an error response.", status, EntityUtils.toString(entity));
        	}
        }
	}
	
	public String get(String uri) throws Exception {
        HttpGet httpget = new HttpGet(this.baseUrl +  uri);
        Arjuna.getLogger().debug("Executing GET request " + httpget.getRequestLine());
        String responseBody = handleResponse(httpclient.execute(httpget));
        Arjuna.getLogger().debug("----------------------------------------");
        return responseBody;
	}
	
	public String post(String uri, String content) throws Exception{
		HttpPost httppost = new HttpPost(this.baseUrl +  uri);
		StringEntity se = new StringEntity(content, Charset.forName("UTF-8"));
		se.setContentType("application/json");
		se.setContentEncoding(new BasicHeader(HTTP.CONTENT_TYPE, "UTF-8"));
		httppost.setEntity(se);
		if (Arjuna.getLogger() != null) {
			Arjuna.getLogger().debug("Executing POST request " + httppost.getRequestLine());
			Arjuna.getLogger().debug(content);
		} else {
			System.out.println("Executing POST request " + httppost.getRequestLine());
			System.out.println(content);			
		}
		String response = handleResponse(httpclient.execute(httppost));
		if (Arjuna.getLogger() != null) {
			Arjuna.getLogger().debug("----------Setu Response ----------------");
			Arjuna.getLogger().debug(response);
			Arjuna.getLogger().debug("----------------------------------------");
		} else {
			System.out.println("----------Setu Response ----------------");
			System.out.println(response);
			System.out.println("----------------------------------------");
		}
        return response;
	}
}