/*
* The Apache Software License, Version 1.1
*
*
* Copyright (c) 2001 The Apache Software Foundation. All rights
* reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* 3. The end-user documentation included with the redistribution,
* if any, must include the following acknowledgment:
* "This product includes software developed by the
* Apache Software Foundation (http://www.apache.org/)."
* Alternately, this acknowledgment may appear in the software itself,
* if and wherever such third-party acknowledgments normally appear.
*
* 4. The names "Axis" and "Apache Software Foundation" must
* not be used to endorse or promote products derived from this
* software without prior written permission. For written
* permission, please contact apache@apache.org.
*
* 5. Products derived from this software may not be called "Apache",
* nor may "Apache" appear in their name, without prior written
* permission of the Apache Software Foundation.
*
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* <http://www.apache.org/>.
*/
package test.saaj;
import junit.framework.TestCase;
import org.apache.axis.components.logger.LogFactory;
import org.apache.commons.logging.Log;
import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.xml.soap.AttachmentPart;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.MimeHeaders;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPElement;
import javax.xml.soap.SOAPEnvelope;
import javax.xml.soap.SOAPHeader;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.SOAPPart;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
/** Test the attachments load/save sample code.
*/
public class TestAttachmentSerialization extends TestCase {
static Log log = LogFactory.getLog(TestAttachmentSerialization.class.getName());
public TestAttachmentSerialization(String name) {
super(name);
}
public static void main(String args[]) throws Exception {
TestAttachmentSerialization tester = new TestAttachmentSerialization("tester");
tester.testAttachments();
}
public void testAttachments() throws Exception {
try {
ByteArrayOutputStream bais = new ByteArrayOutputStream();
int count1 = saveMsgWithAttachments(bais);
int count2 = loadMsgWithAttachments(new ByteArrayInputStream(bais.toByteArray()));
assertEquals(count1, count2);
} catch (Exception e) {
e.printStackTrace();
throw new Exception("Fault returned from test: " + e);
}
}
public static final String MIME_MULTIPART_RELATED = "multipart/related";
public static final String MIME_APPLICATION_DIME = "application/dime";
public static final String NS_PREFIX = "jaxmtst";
public static final String NS_URI = "http://www.jcommerce.net/soap/jaxm/TestJaxm";
public int saveMsgWithAttachments(OutputStream os) throws Exception {
MessageFactory mf = MessageFactory.newInstance();
SOAPMessage msg = mf.createMessage();
SOAPPart sp = msg.getSOAPPart();
SOAPEnvelope envelope = sp.getEnvelope();
SOAPHeader header = envelope.getHeader();
SOAPBody body = envelope.getBody();
SOAPElement el = header.addHeaderElement(envelope.createName("field4", NS_PREFIX, NS_URI));
SOAPElement el2 = el.addChildElement("field4b", NS_PREFIX);
SOAPElement el3 = el2.addTextNode("field4value");
el = body.addBodyElement(envelope.createName("bodyfield3", NS_PREFIX, NS_URI));
el2 = el.addChildElement("bodyfield3a", NS_PREFIX);
el2.addTextNode("bodyvalue3a");
el2 = el.addChildElement("bodyfield3b", NS_PREFIX);
el2.addTextNode("bodyvalue3b");
el2 = el.addChildElement("datefield", NS_PREFIX);
AttachmentPart ap = msg.createAttachmentPart();
ap.setContent("some attachment text...", "text/plain");
msg.addAttachmentPart(ap);
String jpgfilename = "docs/images/axis.jpg";
File myfile = new File(jpgfilename);
FileDataSource fds = new FileDataSource(myfile);
DataHandler dh = new DataHandler(fds);
AttachmentPart ap2 = msg.createAttachmentPart(dh);
ap2.setContentType("image/jpg");
msg.addAttachmentPart(ap2);
// Test for Bug #17664
if(msg.saveRequired()) {
msg.saveChanges();
}
MimeHeaders headers = msg.getMimeHeaders();
assertTrue(headers != null);
String [] contentType = headers.getHeader("Content-Type");
assertTrue(contentType != null);
msg.writeTo(os);
os.flush();
return msg.countAttachments();
}
public int loadMsgWithAttachments(InputStream is) throws Exception {
MimeHeaders headers = new MimeHeaders();
headers.setHeader("Content-Type", MIME_MULTIPART_RELATED);
MessageFactory mf = MessageFactory.newInstance();
SOAPMessage msg = mf.createMessage(headers, is);
SOAPPart sp = msg.getSOAPPart();
SOAPEnvelope envelope = sp.getEnvelope();
assertTrue(sp != null);
assertTrue(envelope != null);
return msg.countAttachments();
}
}