View Javadoc

1   /*
2    *  soapUI, copyright (C) 2006 eviware.com 
3    *
4    *  soapUI is free software; you can redistribute it and/or modify it under the 
5    *  terms of the GNU Lesser General Public License as published by the Free Software Foundation; 
6    *  either version 2.1 of the License, or (at your option) any later version.
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.submit.transports.http;
14  
15  import java.io.ByteArrayInputStream;
16  import java.io.IOException;
17  import java.io.InputStream;
18  import java.io.OutputStream;
19  
20  import javax.activation.DataSource;
21  
22  import com.eviware.soapui.support.Tools;
23  
24  public class PostResponseDataSource implements DataSource
25  {
26  	private final TimeablePostMethod postMethod;
27  	private byte[] data;
28  
29  	public PostResponseDataSource(TimeablePostMethod postMethod)
30  	{
31  		this.postMethod = postMethod;
32  		
33  		try
34  		{
35  			data = Tools.readAll( postMethod.getResponseBodyAsStream(), 0 ).toByteArray();
36  		}
37  		catch (Exception e)
38  		{
39  			e.printStackTrace();
40  		}		
41  	}
42  	
43  	public long getDataSize()
44  	{
45  		return data == null ? -1 : data.length;
46  	}
47  
48  	public String getContentType()
49  	{
50  		return postMethod.getResponseHeader( "Content-Type" ).getValue();
51  	}
52  
53  	public InputStream getInputStream() throws IOException
54  	{
55  		return new ByteArrayInputStream( data );
56  	}
57  
58  	public String getName()
59  	{
60  		return postMethod.getName() + " response for " + postMethod.getPath().toString();
61  	}
62  
63  	public OutputStream getOutputStream() throws IOException
64  	{
65  		return null;
66  	}
67  
68  }