com.eviware.soapui.impl.wsdl.submit.transports.http.MimeMessageRequestEntity.java Source code

Introduction

Here is the source code for com.eviware.soapui.impl.wsdl.submit.transports.http.MimeMessageRequestEntity.java

Source

/*
 *  soapui, copyright (C) 2006 eviware.com 
 *
 *  SoapUI is free software; you can redistribute it and/or modify it under the 
 *  terms of the GNU Lesser General Public License as published by the Free Software Foundation; 
 *  either version 2.1 of the License, or (at your option) any later version.
 *
 *  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.impl.wsdl.submit.transports.http;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;

import org.apache.commons.httpclient.methods.RequestEntity;

import com.eviware.soapui.impl.wsdl.WsdlRequest;

public class MimeMessageRequestEntity implements RequestEntity {
    private final MimeMessage message;
    private byte[] buffer;
    private final boolean isXOP;
    private final WsdlRequest wsdlRequest;

    public MimeMessageRequestEntity(MimeMessage message, boolean isXOP, WsdlRequest wsdlRequest) {
        this.message = message;
        this.isXOP = isXOP;
        this.wsdlRequest = wsdlRequest;
    }

    public long getContentLength() {
        try {
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            writeRequest(out);
            buffer = out.toByteArray();
            return buffer.length;
        } catch (Exception e) {
            e.printStackTrace();
            return -1;
        }
    }

    public String getContentType() {
        try {
            if (isXOP) {
                String header = message.getHeader("Content-Type")[0];
                return AttachmentUtils.buildMTOMContentType(header, wsdlRequest.getOperation().getAction());
            } else {
                String header = message.getHeader("Content-Type")[0];
                int ix = header.indexOf("boundary");
                return "multipart/related; type=\"text/xml\"; start=\""
                        + HttpClientRequestTransport.ROOTPART_SOAPUI_ORG + "\"; " + header.substring(ix);
            }
        } catch (MessagingException e) {
            e.printStackTrace();
        }

        return null;
    }

    public boolean isRepeatable() {
        return true;
    }

    public void writeRequest(OutputStream arg0) throws IOException {
        try {
            if (buffer == null) {
                arg0.write("\r\n".getBytes());
                ((MimeMultipart) message.getContent()).writeTo(arg0);
            } else
                arg0.write(buffer);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}