1
2
3
4
5
6
7
8
9
10
11
12
13 package com.eviware.soapui.support.dnd.handlers;
14
15 import java.util.HashSet;
16
17 import com.eviware.soapui.SoapUI;
18 import com.eviware.soapui.impl.wsdl.WsdlProject;
19 import com.eviware.soapui.impl.wsdl.WsdlRequest;
20 import com.eviware.soapui.impl.wsdl.actions.request.AddRequestToTestCaseAction;
21 import com.eviware.soapui.model.iface.Interface;
22 import com.eviware.soapui.model.tree.nodes.support.WsdlTestStepsModelItem;
23 import com.eviware.soapui.support.UISupport;
24 import com.eviware.soapui.support.action.SoapUIAction;
25
26 public class RequestToTestStepsDropHandler extends
27 AbstractAfterModelItemDropHandler<WsdlRequest, WsdlTestStepsModelItem>
28 {
29 public RequestToTestStepsDropHandler()
30 {
31 super( WsdlRequest.class, WsdlTestStepsModelItem.class );
32 }
33
34 @Override
35 boolean canCopyAfter( WsdlRequest source, WsdlTestStepsModelItem target )
36 {
37 return true;
38 }
39
40 @Override
41 boolean canMoveAfter( WsdlRequest source, WsdlTestStepsModelItem target )
42 {
43 return true;
44 }
45
46 @Override
47 boolean copyAfter( WsdlRequest source, WsdlTestStepsModelItem target )
48 {
49 return addRequestToTestCase( source, target );
50 }
51
52 private boolean addRequestToTestCase( WsdlRequest source, WsdlTestStepsModelItem target )
53 {
54 if( !UISupport.confirm( "Add Request [" + source.getName() + "] to TestCase [" + target.getTestCase().getName()
55 + "]", "Add Request to TestCase" ) )
56 return false;
57
58 WsdlProject targetProject = target.getTestCase().getTestSuite().getProject();
59 if( targetProject != source.getOperation().getInterface().getProject() )
60 {
61 HashSet<Interface> requiredInterfaces = new HashSet<Interface>();
62 requiredInterfaces.add( source.getOperation().getInterface() );
63
64 if( !DragAndDropSupport
65 .importRequiredInterfaces( targetProject, requiredInterfaces, "Add Request to TestCase" ) )
66 {
67 return false;
68 }
69 }
70
71 SoapUIAction<WsdlRequest> action = SoapUI.getActionRegistry().getAction(
72 AddRequestToTestCaseAction.SOAPUI_ACTION_ID );
73 return ( ( AddRequestToTestCaseAction )action ).addRequest( target.getTestCase(), source, 0 ) != null;
74 }
75
76 @Override
77 boolean moveAfter( WsdlRequest source, WsdlTestStepsModelItem target )
78 {
79 return addRequestToTestCase( source, target );
80 }
81
82 @Override
83 String getCopyAfterInfo( WsdlRequest source, WsdlTestStepsModelItem target )
84 {
85 return "Add Request [" + source.getName() + "] to TestCase [" + target.getTestCase().getName() + "]";
86 }
87
88 @Override
89 String getMoveAfterInfo( WsdlRequest source, WsdlTestStepsModelItem target )
90 {
91 return getCopyAfterInfo( source, target );
92 }
93 }