/*
* Copyright 2004-2014 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.support.dnd.handlers;
import java.util.HashSet;
import java.util.Set;
import com.eviware.soapui.impl.wsdl.WsdlProject;
import com.eviware.soapui.impl.wsdl.WsdlTestSuite;
import com.eviware.soapui.impl.wsdl.teststeps.WsdlTestStep;
import com.eviware.soapui.model.iface.Interface;
import com.eviware.soapui.model.testsuite.TestCase;
import com.eviware.soapui.support.UISupport;
public class TestSuiteToTestSuiteDropHandler extends
AbstractBeforeAfterModelItemDropHandler<WsdlTestSuite, WsdlTestSuite> {
public TestSuiteToTestSuiteDropHandler() {
super(WsdlTestSuite.class, WsdlTestSuite.class);
}
@Override
boolean canCopyAfter(WsdlTestSuite source, WsdlTestSuite target) {
return true;
}
@Override
boolean canMoveAfter(WsdlTestSuite source, WsdlTestSuite target) {
return source != target;
}
@Override
boolean copyAfter(WsdlTestSuite source, WsdlTestSuite target) {
WsdlTestSuite testCase = copyTestSuite(source, target.getProject(),
target.getProject().getIndexOfTestSuite(target) + 1);
if (testCase != null) {
UISupport.select(testCase);
}
return testCase != null;
}
@Override
boolean moveAfter(WsdlTestSuite source, WsdlTestSuite target) {
WsdlTestSuite testCase = moveTestSuite(source, target.getProject(),
target.getProject().getIndexOfTestSuite(target) + 1);
if (testCase != null) {
UISupport.select(testCase);
}
return testCase != null;
}
@Override
String getCopyAfterInfo(WsdlTestSuite source, WsdlTestSuite target) {
return "Copy TestSuite [" + source.getName() + "] to Project [" + target.getProject().getName() + "]";
}
@Override
String getMoveAfterInfo(WsdlTestSuite source, WsdlTestSuite target) {
return source == target ? "Move TestCase [" + source.getName() + "] within TestSuite" : "Move TestCase ["
+ source.getName() + "] to TestSuite in Project [" + target.getName() + "]";
}
@Override
boolean canCopyBefore(WsdlTestSuite source, WsdlTestSuite target) {
return true;
}
@Override
boolean canMoveBefore(WsdlTestSuite source, WsdlTestSuite target) {
return source != target;
}
@Override
boolean copyBefore(WsdlTestSuite source, WsdlTestSuite target) {
WsdlTestSuite testCase = copyTestSuite(source, source.getProject(),
target.getProject().getIndexOfTestSuite(target));
if (testCase != null) {
UISupport.select(testCase);
}
return testCase != null;
}
@Override
String getCopyBeforeInfo(WsdlTestSuite source, WsdlTestSuite target) {
return getCopyAfterInfo(source, target);
}
@Override
String getMoveBeforeInfo(WsdlTestSuite source, WsdlTestSuite target) {
return getMoveAfterInfo(source, target);
}
@Override
boolean moveBefore(WsdlTestSuite source, WsdlTestSuite target) {
WsdlTestSuite testCase = moveTestSuite(source, target.getProject(),
target.getProject().getIndexOfTestSuite(target));
if (testCase != null) {
UISupport.select(testCase);
}
return testCase != null;
}
public static WsdlTestSuite moveTestSuite(WsdlTestSuite testSuite, WsdlProject target, int position) {
if (testSuite.getProject() == target) {
int ix = target.getIndexOfTestSuite(testSuite);
if (position == -1) {
target.moveTestSuite(ix, target.getTestSuiteCount() - ix);
} else if (ix >= 0 && position != ix) {
int offset = position - ix;
if (offset > 0) {
offset--;
}
target.moveTestSuite(ix, offset);
}
} else if (UISupport.confirm("Move TestSuite [" + testSuite.getName() + "] to Project [" + target.getName() + "]",
"Move TestSuite")) {
Set<Interface> requiredInterfaces = new HashSet<Interface>();
// get required interfaces
for (TestCase testCase : testSuite.getTestCaseList()) {
for (int y = 0; y < testCase.getTestStepCount(); y++) {
WsdlTestStep testStep = (WsdlTestStep) testCase.getTestStepAt(y);
requiredInterfaces.addAll(testStep.getRequiredInterfaces());
}
}
if (DragAndDropSupport.importRequiredInterfaces(target, requiredInterfaces, "Move TestSuite")) {
WsdlTestSuite importedTestSuite = target.importTestSuite(testSuite, testSuite.getName(), position, true,
null);
if (importedTestSuite != null) {
testSuite.getProject().removeTestSuite(testSuite);
return importedTestSuite;
}
}
}
return null;
}
public static WsdlTestSuite copyTestSuite(WsdlTestSuite testSuite, WsdlProject target, int position) {
String name = UISupport.prompt("Specify name of copied TestCase", "Copy TestCase",
"Copy of " + testSuite.getName());
if (name == null) {
return null;
}
if (testSuite.getProject() == target) {
return target.importTestSuite(testSuite, name, position, true, null);
} else {
Set<Interface> requiredInterfaces = new HashSet<Interface>();
// get required interfaces
for (TestCase testCase : testSuite.getTestCaseList()) {
for (int y = 0; y < testCase.getTestStepCount(); y++) {
WsdlTestStep testStep = (WsdlTestStep) testCase.getTestStepAt(y);
requiredInterfaces.addAll(testStep.getRequiredInterfaces());
}
}
if (DragAndDropSupport.importRequiredInterfaces(target, requiredInterfaces, "Move TestSuite")) {
return target.importTestSuite(testSuite, testSuite.getName(), position, true, null);
}
}
return null;
}
}