com.ats.script.actions.performance.ActionPerformanceStart Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ats-automated-testing Show documentation
Code generator library to create and execute GUI automated tests
The newest version!
/*
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements.  See the NOTICE file
distributed with this work for additional information
regarding copyright ownership.  The ASF 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 com.ats.script.actions.performance;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.function.Predicate;
import java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.text.StringEscapeUtils;

import com.ats.executor.ActionTestScript;
import com.ats.script.Script;
import com.ats.tools.Utils;

public class ActionPerformanceStart extends ActionPerformance {

	public static final String SCRIPT_LABEL = ActionPerformance.SCRIPT_PERFORMANCE_LABEL + "-start";
	public static final Predicate PREDICATE = g -> SCRIPT_LABEL.equals(g);

	public static final String SEND_LABEL = "upload";
	public static final String RECEIVE_LABEL = "download";
	public static final String IDLE_LABEL = "idle";
	public static final String LATENCY_LABEL = "latency";

	private static final String[] LABEL_REPLACE = new String[]{SEND_LABEL, RECEIVE_LABEL, IDLE_LABEL, LATENCY_LABEL, "=", " "};
	private static final String[] LABEL_REPLACEMENT = new String[]{"", "", "", "", "", ""};

	private long sendBandWidth = 0L;
	private long receiveBandWidth = 0L;
	private int latency = 0;
	private int trafficIdle = 0;

	private List whiteList = null;
	private String filters = "";

	public ActionPerformanceStart() {}

	public ActionPerformanceStart(Script script, ArrayList options, ArrayList dataArray) {
		super(script);
		setWhiteList(dataArray.stream().map(String :: trim).collect(Collectors.toList()));
		loadBandwidthValues(options);
	}

	public ActionPerformanceStart(Script script, int iddle, int ltcy, long sbw, long rbw, String[] whiteList) {
		super(script);
		setTrafficIdle(iddle);
		setSendBandWidth(sbw);
		setReceiveBandWidth(rbw);
		setWhiteList(Arrays.asList(whiteList));
	}

	private void loadBandwidthValues(ArrayList options) {
		for (String opt : options) {
			if(opt.contains(SEND_LABEL)) {
				sendBandWidth = Utils.string2Long(StringUtils.replaceEach(opt, LABEL_REPLACE, LABEL_REPLACEMENT));
			}else if(opt.contains(RECEIVE_LABEL)) {
				receiveBandWidth = Utils.string2Long(StringUtils.replaceEach(opt, LABEL_REPLACE, LABEL_REPLACEMENT));
			}else if(opt.contains(LATENCY_LABEL)) {
				latency = Utils.string2Int(StringUtils.replaceEach(opt, LABEL_REPLACE, LABEL_REPLACEMENT));
			}else if(opt.contains(IDLE_LABEL)) {
				trafficIdle = Utils.string2Int(StringUtils.replaceEach(opt, LABEL_REPLACE, LABEL_REPLACEMENT));
			}
		}
	}

	//---------------------------------------------------------------------------------------------------------------------------------
	// Code Generator
	//---------------------------------------------------------------------------------------------------------------------------------

	@Override
	public StringBuilder getJavaCode() {
		final ArrayList list = new ArrayList<>();
		for(String s : whiteList) {
			list.add("\"" + StringEscapeUtils.escapeJava(s) + "\"");
		}
		return super.getJavaCode().append(trafficIdle).append(", ").append(latency).append(", ").append(sendBandWidth).append("L, ").append(receiveBandWidth).append("L, new String[]{").append(String.join(", ", list)).append("})");
	}

	//---------------------------------------------------------------------------------------------------------------------------------
	// Execution
	//---------------------------------------------------------------------------------------------------------------------------------

	@Override
	public String execute(ActionTestScript ts, String testName, int testLine, int tryNum) {
		super.execute(ts, testName, testLine, tryNum);
		getCurrentChannel().startHarServer(status, whiteList, trafficIdle, latency, sendBandWidth, receiveBandWidth);
		status.endDuration();
		return null;
	}

	//--------------------------------------------------------
	// getters and setters for serialization
	//--------------------------------------------------------

	public int getLatency() {
		return latency;
	}

	public void setLatency(int latency) {
		this.latency = latency;
	}

	public int getTrafficIdle() {
		return trafficIdle;
	}

	public void setTrafficIdle(int value) {
		this.trafficIdle = value;
	}

	public long getSendBandWidth() {
		return sendBandWidth;
	}

	public void setSendBandWidth(long value) {
		this.sendBandWidth = value;
	}

	public long getReceiveBandWidth() {
		return receiveBandWidth;
	}

	public void setReceiveBandWidth(long value) {
		this.receiveBandWidth = value;
	}

	public List getWhiteList() {
		return whiteList;
	}

	public void setWhiteList(List data) {
		this.whiteList = data;
	}

	public String getFilters() {
		return filters;
	}

	public void setFilters(String filters) {
		this.filters = filters;
	}
}