View Javadoc

1   /*
2    *  soapUI, copyright (C) 2004-2007 eviware.com 
3    *
4    *  soapUI is free software; you can redistribute it and/or modify it under the 
5    *  terms of version 2.1 of the GNU Lesser General Public License as published by 
6    *  the Free Software Foundation.
7    *
8    *  soapUI is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without 
9    *  even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
10   *  See the GNU Lesser General Public License for more details at gnu.org.
11   */
12  
13  package com.eviware.soapui.impl.wsdl.teststeps;
14  
15  import com.eviware.soapui.impl.wsdl.WsdlOperation;
16  import com.eviware.soapui.impl.wsdl.WsdlRequest;
17  import com.eviware.soapui.impl.wsdl.submit.WsdlMessageExchange;
18  import com.eviware.soapui.impl.wsdl.submit.transports.http.WsdlResponse;
19  import com.eviware.soapui.model.iface.Attachment;
20  import com.eviware.soapui.support.types.StringToStringMap;
21  
22  public class WsdlResponseMessageExchange extends WsdlMessageExchange
23  {
24  	private final WsdlRequest request;
25  	private WsdlResponse response;
26  	private String requestContent;
27  
28  	public WsdlResponseMessageExchange( WsdlRequest request )
29  	{
30  		this.request = request;
31  		response = request.getResponse();
32  	}
33  	
34  	public void setResponse( WsdlResponse response )
35  	{
36  		this.response = response;
37  	}
38  
39  	public String getRequestContent()
40  	{
41  		if( requestContent != null )
42  			return requestContent;
43  		
44  		if( response == null )
45  			response = request.getResponse();
46  		
47  		return response == null ? request.getRequestContent() : response.getRequestContent(); 
48  	}
49  
50  	public StringToStringMap getRequestHeaders()
51  	{
52  		if( response == null )
53  			response = request.getResponse();
54  		
55  		return response == null ? request.getRequestHeaders() : response.getRequestHeaders(); 
56  	}
57  	
58  	public Attachment[] getRequestAttachments()
59  	{
60  		return request.getAttachments();
61  	}
62  
63  	public Attachment[] getResponseAttachments()
64  	{
65  		if( response == null )
66  			response = request.getResponse();
67  		
68  		return response == null ? null : response.getAttachments();
69  	}
70  
71  	public String getResponseContent()
72  	{
73  		if( response == null )
74  			response = request.getResponse();
75  		
76  		return response == null ? null : response.getContentAsString();
77  	}
78  
79  	public StringToStringMap getResponseHeaders()
80  	{
81  		if( response == null )
82  			response = request.getResponse();
83  		
84  		return response == null ? null : response.getResponseHeaders();
85  	}
86  
87  	public WsdlOperation getOperation()
88  	{
89  		return request.getOperation();
90  	}
91  
92  	public long getTimeTaken()
93  	{
94  		if( response == null )
95  			response = request.getResponse();
96  		
97  		return response == null ? 0 : response.getTimeTaken();
98  	}
99  
100 	public long getTimestamp()
101 	{
102 		if( response == null )
103 			response = request.getResponse();
104 		
105 		return response == null ? 0 : response.getTimestamp();
106 	}
107 
108 	public void setRequestContent( String requestContent )
109 	{
110 		this.requestContent = requestContent;
111 	}
112 }