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.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.SoapUI;
    23  import com.eviware.soapui.impl.wsdl.support.http.HttpClientSupport;
    24  import com.eviware.soapui.support.Tools;
    25
    26  /***
    27   * DataSource for a standard POST response
    28   *
    29   * @author ole.matzura
    30   */
    31
    32  public class PostResponseDataSource implements DataSource
    33  {
    34  	private final TimeablePostMethod postMethod;
    35  	private byte[] data;
    36
    37  	public PostResponseDataSource(TimeablePostMethod postMethod)
    38  	{
    39  		this.postMethod = postMethod;
    40
    41  		try
    42  		{
    43  			data = Tools.readAll( postMethod.getResponseBodyAsStream(), 0 ).toByteArray();
    44
    45  			if( HttpClientSupport.isZippedResponse( postMethod ))
    46  			{
    47  				data = HttpClientSupport.decompress( data );
    48  			}
    49  		}
    50  		catch (Exception e)
    51  		{
    52  			SoapUI.logError( e );
    53  		}
    54  	}
    55
    56  	public long getDataSize()
    57  	{
    58  		return data == null ? -1 : data.length;
    59  	}
    60
    61  	public String getContentType()
    62  	{
    63  		return postMethod.getResponseHeader( "Content-Type" ).getValue();
    64  	}
    65
    66  	public InputStream getInputStream() throws IOException
    67  	{
    68  		return new ByteArrayInputStream( data );
    69  	}
    70
    71  	public String getName()
    72  	{
    73  		return postMethod.getName() + " response for " + postMethod.getPath().toString();
    74  	}
    75
    76  	public OutputStream getOutputStream() throws IOException
    77  	{
    78  		return null;
    79  	}
    80
    81  }