arjuna.client.core.connector.BaseSetuObject 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!
/*******************************************************************************
 * Copyright 2015-19 Test Mile Software Testing Pvt Ltd
 * 
 * Website: www.TestMile.com
 * Email: support [at] testmile.com
 * 
 * Licensed 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 arjuna.client.core.connector;

import java.util.HashMap;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

import arjuna.client.core.action.SetuActionType;
import arjuna.lib.httpclient.BasicRestClient;
import arjuna.lib.httpclient.SetuHttpException;
import arjuna.lib.value.AnyRefValue;
import arjuna.tpi.Arjuna;
import arjuna.tpi.value.Value;

public class BaseSetuObject implements SetuManagedObject {
	private String setuId;
	private Map coreArgs = new HashMap();

	@Override
	public String getSetuId() {
		return this.setuId;
	}
	
	protected void setSetuId(String idValue) {
		this.setuId = idValue;
	}
	
	protected void setTestSessionSetuIdArg(String id) {
		SetuConnectUtils.setTestSessionSetuIdArg(coreArgs, id);
	}
	
	protected void setAutomatorSetuIdArg(String id) {
		SetuConnectUtils.setAutomatorSetuIdArg(coreArgs, id);
	}
	
	protected void setGuiSetuIdArg(String id) {
		SetuConnectUtils.setGuiSetuIdArg(coreArgs, id);
	}
	
	protected void setSelfSetuIdArg(String idName) {
		this.coreArgs.put(idName, this.setuId);
	}
	
	protected void addArgs(SetuArg... args) {
		SetuConnectUtils.addArgs(coreArgs, args);		
	}
	
	public ,E2 extends Enum> SetuResponse sendRequest(E1 component, E2 actionType) throws Exception {
		return SetuConnectUtils.sendRequest(component, actionType, coreArgs);
	}
	
	public ,E2 extends Enum> SetuResponse sendRequest(E1 component, E2 actionType, SetuArg... args) throws Exception {
		return SetuConnectUtils.sendRequest(component, actionType, coreArgs, args);
	}

}

interface SetuRequester {

	SetuResponse post(SetuRequest action) throws Exception;
	void postExit(SetuRequest action) throws Exception;

}

enum ResponseCode {
	SUCCESS,
	ERROR
}

class DefaultSetuRequester implements SetuRequester {
	private String setuUrl = "http://localhost:9000";
	private String baseUri = "/setu";
	protected BasicRestClient restClient;

	public DefaultSetuRequester() {
		this.restClient = new BasicRestClient(setuUrl);
	}
	
	@Override
	public void postExit(SetuRequest action) throws Exception {
		try {
			this.restClient.post(baseUri, action.asJsonString());
		} catch (Exception e) {
			
		}
	}

	@Override
	public SetuResponse post(SetuRequest action) throws Exception {
		SetuResponse setuResponse = null;
		String response = null;
		try {
			response = this.restClient.post(baseUri, action.asJsonString());
			setuResponse = DefaultSetuResponse.fromJsonStr(response);
		} catch (SetuHttpException e) {
			if (Arjuna.getLogger() != null) {
				Arjuna.getLogger().error("---------- Setu Error Response ----------------------");
				Arjuna.getLogger().error("Got Http Error: " + e.getStatusCode());
			} else {
				System.err.println("---------- Setu Error Response ----------------------");
				System.err.println("Got Http Error: " + e.getStatusCode());				
			}
			setuResponse = DefaultSetuResponse.fromJsonStr(e.getResponse());
			if (Arjuna.getLogger() != null) {
				Arjuna.getLogger().error("Error: " +  setuResponse.getMessage());
				Arjuna.getLogger().error("---------- Setu Error Trace -------------------------");
				Arjuna.getLogger().error(setuResponse.getTrace());
				Arjuna.getLogger().error("-----------------------------------------------------");
			} else {
				System.err.println("Error: " +  setuResponse.getMessage());
				System.err.println("---------- Setu Error Trace -------------------------");
				System.err.println(setuResponse.getTrace());
				System.err.println("-----------------------------------------------------");				
			}
			throw e;
		}
		return setuResponse;
	}
}

interface SetuRequest {
	void addArg(String name, Object value);
	String asJsonString();
}

class DefaultSetuRequest implements SetuRequest {
	
	private static Gson gson = new GsonBuilder().setPrettyPrinting().create();
	private RequestBody actionRequest;
	
	public DefaultSetuRequest(String component, String actionType) {
		this.actionRequest = new RequestBody();
		this.actionRequest.setArjunaComponent(component);
		this.actionRequest.setAction(actionType);
	}
	
	public DefaultSetuRequest(String testSessionSetuId) {
		this.actionRequest = new RequestBody();
		this.addArg("testSessionSetuId", testSessionSetuId);
	}

	public void addAllArgs(Map args) {
		for (String argName: args.keySet()) {
			this.addArg(argName, args.get(argName));
		}
	}
	
	public void addArg(String name, Object value) {
		this.getActionRequest().addArg(name, value);
	}
	
	@Override
	public String asJsonString() {
		return gson.toJson(this.getActionRequest(), RequestBody.class);
	}

	protected RequestBody getActionRequest() {
		return actionRequest;
	}
	
	private class RequestBody {
		private String arjunaComponent;
		private String action;
//		private String setuId;
		private Map args = null;
		
		public void addArg(String name, Object value) {
			if (this.args == null) {
				this.args = new HashMap();
			}
			this.args.put(name, value);
		}
		
		public void setArjunaComponent(String component) {
			this.arjunaComponent = component;
		}

		public void setAction(String action) {
			this.action = action;
		}
	}	
}

class DefaultSetuResponse implements SetuResponse {
	private static final Gson gson = new Gson();
	private ResponseCode result;
	private String emessage;
	private String etrace;
	private Map responseData = null;
	
	@Override
	public ResponseCode getResult() {
		return result;
	}
	
	public void setResult(ResponseCode code) {
		this.result = code;
	}

	@Override
	public String getMessage() {
		return emessage;
	}
	public void setMessage(String message) {
		this.emessage = message;
	}

	@Override
	public String getTrace() {
		return etrace;
	}
	
	public void setTrace(String trace) {
		this.etrace = trace;
	}
	
	@Override
	public Map getData(){
		return this.responseData;
	}
	
	public Value getValueForKey(String keyName) {
		return new AnyRefValue(this.getData().get(keyName));
	}
	
	public Value getValue() {
		return getValueForKey("value");
	}
	
	public String getValueForValueAttr() throws Exception {
		return getValueForKey("value").asString();
	}
	
	public String getValueForText() throws Exception {
		return getValueForKey("text").asString();
	}
	
	public boolean getValueForCheckResult() throws Exception {
		return getValueForKey("checkResult").asBoolean();
	}
	
	@Override
	public String getValueForElementSetuId() throws Exception {
		return getValueForKey("elementSetuId").asString();
	}
	
	@Override
	public String getValueForTestSessionSetuId() throws Exception {
		return getValueForKey("testSessionSetuId").asString();
	}
	
	@Override
	public String getValueForConfigSetuId() throws Exception {
		return getValueForKey("configSetuId").asString();
	}
	
	@Override
	public String getValueForGuiAutomatorSetuId() throws Exception {
		return getValueForKey("automatorSetuId").asString();
	}
	
	@Override
	public String getGuiSetuId() throws Exception {
		return getValueForKey("guiSetuId").asString();
	}
	
	@Override
	public String getDataSourceSetuId() throws Exception {
		return getValueForKey("dataSourceSetuId").asString();
	}
	
	public void addDataItem(String name, Object value) {
		if (this.responseData == null) {
			this.responseData = new HashMap();
		}
		this.responseData.put(name, value);
	}
	
	public static SetuResponse fromJsonStr(String json) {
		return gson.fromJson(json, DefaultSetuResponse.class);
	}
}