/*
* SoapUI, Copyright (C) 2004-2016 SmartBear Software
*
* Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent
* versions of the EUPL (the "Licence");
* You may not use this work except in compliance with the Licence.
* You may obtain a copy of the Licence at:
*
* http://ec.europa.eu/idabc/eupl
*
* Unless required by applicable law or agreed to in writing, software distributed under the Licence is
* distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the Licence for the specific language governing permissions and limitations
* under the Licence.
*/
package com.eviware.soapui.impl.wsdl.loadtest.data;
import com.eviware.soapui.model.support.LoadTestRunListenerAdapter;
import com.eviware.soapui.model.support.TestSuiteListenerAdapter;
import com.eviware.soapui.model.testsuite.LoadTest;
import com.eviware.soapui.model.testsuite.LoadTestRunContext;
import com.eviware.soapui.model.testsuite.LoadTestRunner;
import com.eviware.soapui.model.testsuite.TestCaseRunContext;
import com.eviware.soapui.model.testsuite.TestCaseRunner;
import com.eviware.soapui.model.testsuite.TestStep;
import com.eviware.soapui.model.testsuite.TestStepResult;
import org.apache.log4j.Logger;
import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
/**
* Collector of samples from a loadtest, exposed as TableModel
*
* @author Ole.Matzura
*/
public class LoadTestSamples extends AbstractTableModel {
private final LoadTest loadTest;
private List<List<LoadTestStepSample[]>> samples = new ArrayList<List<LoadTestStepSample[]>>();
private List<Long> timestamps = new ArrayList<Long>();
private InternalLoadTestRunListener loadTestRunListener = new InternalLoadTestRunListener();
private InternalTestSuiteListener testSuiteListener = new InternalTestSuiteListener();
private final static Logger log = Logger.getLogger(LoadTestSamples.class);
public LoadTestSamples(LoadTest loadTest) {
this.loadTest = loadTest;
loadTest.addLoadTestRunListener(loadTestRunListener);
loadTest.getTestCase().getTestSuite().addTestSuiteListener(testSuiteListener);
}
public int getRowCount() {
return samples.size();
}
public void release() {
loadTest.removeLoadTestRunListener(loadTestRunListener);
loadTest.getTestCase().getTestSuite().removeTestSuiteListener(testSuiteListener);
}
public int getColumnCount() {
return loadTest.getTestCase().getTestStepCount() + 1;
}
public Object getValueAt(int rowIndex, int columnIndex) {
return columnIndex == 0 ? new Date(timestamps.get(rowIndex)) : samples.get(rowIndex).get(columnIndex - 1);
}
public Class<?> getColumnClass(int columnIndex) {
return columnIndex == 0 ? Date.class : LoadTestStepSample[].class;
}
public String getColumnName(int columnIndex) {
return columnIndex == 0 ? "Timestamp" : loadTest.getTestCase().getTestStepAt(columnIndex - 1).getName();
}
public synchronized void clear() {
int size = samples.size();
samples.clear();
timestamps.clear();
fireTableRowsDeleted(0, size - 1);
}
private final class InternalTestSuiteListener extends TestSuiteListenerAdapter {
public void loadTestRemoved(LoadTest loadTest) {
if (loadTest.equals(LoadTestSamples.this.loadTest)) {
loadTest.removeLoadTestRunListener(loadTestRunListener);
}
}
public void testStepAdded(TestStep testStep, int index) {
if (testStep.getTestCase() == loadTest.getTestCase()) {
for (List<LoadTestStepSample[]> values : samples) {
values.add(index, new LoadTestStepSample[0]);
}
}
}
public void testStepMoved(TestStep testStep, int fromIndex, int offset) {
if (testStep.getTestCase() == loadTest.getTestCase()) {
for (List<LoadTestStepSample[]> values : samples) {
LoadTestStepSample[] s = values.remove(fromIndex);
values.add(offset, s);
}
}
}
public void testStepRemoved(TestStep testStep, int index) {
if (testStep.getTestCase() == loadTest.getTestCase()) {
for (List<LoadTestStepSample[]> values : samples) {
values.remove(index);
}
}
}
}
private class InternalLoadTestRunListener extends LoadTestRunListenerAdapter {
public void afterTestCase(LoadTestRunner loadTestRunner, LoadTestRunContext context, TestCaseRunner testRunner,
TestCaseRunContext runContext) {
long timestamp = System.currentTimeMillis();
List<LoadTestStepSample[]> s = new ArrayList<LoadTestStepSample[]>();
List<TestStepResult> testResults = testRunner.getResults();
for (int c = 0; c < loadTest.getTestCase().getTestStepCount(); c++) {
TestStep testStep = loadTest.getTestCase().getTestStepAt(c);
List<LoadTestStepSample> results = new ArrayList<LoadTestStepSample>();
for (int i = 0; i < testResults.size(); i++) {
TestStepResult stepResult = testResults.get(i);
if (stepResult == null) {
log.warn("Result [" + c + "] is null in TestCase [" + testRunner.getTestCase().getName() + "]");
continue;
}
if (stepResult.getTestStep().equals(testStep)) {
results.add(new LoadTestStepSample(stepResult));
}
}
s.add(results.toArray(new LoadTestStepSample[results.size()]));
}
synchronized (this) {
samples.add(s);
timestamps.add(timestamp);
fireTableRowsInserted(samples.size() - 1, samples.size() - 1);
}
}
}
}