/*
* soapUI, copyright (C) 2004-2011 eviware.com
*
* soapUI is free software; you can redistribute it and/or modify it under the
* terms of version 2.1 of the GNU Lesser General Public License as published by
* the Free Software Foundation.
*
* soapUI is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
* even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Lesser General Public License for more details at gnu.org.
*/
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;
}
}