modules/saaj/test/org/apache/axis2/saaj/AttachmentTest.java - axis-axis2-java-core - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements. See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership. The ASF licenses this file
  * to you under the Apache License, Version 2.0 (the
  * "License"); you may not use this file except in compliance
  * with the License. You may obtain a copy of the License at
  *
  * http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  * KIND, either express or implied. See the License for the
  * specific language governing permissions and limitations
  * under the License.
  */

 package org.apache.axis2.saaj;

 import junit.framework.Assert;
 import org.apache.commons.codec.binary.Base64;
 import org.apache.commons.io.IOUtils;
 import org.junit.Test;
 import org.junit.runner.RunWith;

 import javax.activation.DataHandler;
 import javax.activation.DataSource;
 import javax.imageio.ImageIO;
 import javax.xml.soap.AttachmentPart;
 import javax.xml.soap.MessageFactory;
 import javax.xml.soap.MimeHeader;
 import javax.xml.soap.MimeHeaders;
 import javax.xml.soap.SOAPException;
 import javax.xml.soap.SOAPMessage;
 import javax.xml.transform.stream.StreamSource;

 import static org.assertj.core.api.Assertions.assertThat;

 import java.awt.*;
 import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.Iterator;
 import java.util.Random;

 @RunWith(SAAJTestRunner.class)
 public class AttachmentTest extends Assert {
     @Validated @Test
     public void testStringAttachment() throws Exception {

         MessageFactory factory = MessageFactory.newInstance();
         SOAPMessage message = factory.createMessage();
         AttachmentPart attachment = message.createAttachmentPart();
         String stringContent = "Update address for Sunny Skies " +
                 "Inc., to 10 Upbeat Street, Pleasant Grove, CA 95439";

         attachment.setContent(stringContent, "text/plain");
         attachment.setContentId("update_address");
         message.addAttachmentPart(attachment);

         assertTrue(message.countAttachments() == 1);

         Iterator it = message.getAttachments();
         while (it.hasNext()) {
             attachment = (AttachmentPart)it.next();
             Object content = attachment.getContent();
             String id = attachment.getContentId();
             assertEquals(content, stringContent);
         }
         message.removeAllAttachments();
         assertTrue(message.countAttachments() == 0);
     }

     @Validated @Test
     public void testMultipleAttachments() throws Exception {

         MessageFactory factory = MessageFactory.newInstance();
         SOAPMessage msg = factory.createMessage();

         AttachmentPart a1 = msg.createAttachmentPart(new DataHandler("<some_xml/>", "text/xml"));
         a1.setContentType("text/xml");
         msg.addAttachmentPart(a1);
         AttachmentPart a2 = msg.createAttachmentPart(new DataHandler("<some_xml/>", "text/xml"));
         a2.setContentType("text/xml");
         msg.addAttachmentPart(a2);
         AttachmentPart a3 = msg.createAttachmentPart(new DataHandler("text", "text/plain"));
         a3.setContentType("text/plain");
         msg.addAttachmentPart(a3);

         assertTrue(msg.countAttachments() == 3);

         MimeHeaders mimeHeaders = new MimeHeaders();
         mimeHeaders.addHeader("Content-Type", "text/xml");

         int nAttachments = 0;
         Iterator iterator = msg.getAttachments(mimeHeaders);
         while (iterator.hasNext()) {
             nAttachments++;
             AttachmentPart ap = (AttachmentPart)iterator.next();
             assertTrue(ap.equals(a1) || ap.equals(a2));
         }
         assertTrue(nAttachments == 2);
     }

     // Note: This test case fails with Sun's SAAJ implementation
     //       and can't be @Validated.
     @Test
     public void testBadAttSize() throws Exception {
         MessageFactory factory = MessageFactory.newInstance();
         SOAPMessage message = factory.createMessage();

         ByteArrayInputStream ins = new ByteArrayInputStream(new byte[5]);
         DataHandler dh = new DataHandler(new Src(ins, "text/plain"));
         AttachmentPart part = message.createAttachmentPart(dh);
         assertEquals("Size should match", 5, part.getSize());
     }

     class Src implements DataSource {
         InputStream m_src;
         String m_type;

         public Src(InputStream data, String type) {
             m_src = data;
             m_type = type;
         }

         public String getContentType() {
             return m_type;
         }

         public InputStream getInputStream() throws IOException {
             m_src.reset();
             return m_src;
         }

         public String getName() {
             return "Some-Data";
         }

         public OutputStream getOutputStream() {
             throw new UnsupportedOperationException("I don't give output streams");
         }
     }

     @Validated @Test
     public void testClearContent() throws Exception {
         try {
             InputStream in1 = TestUtils.getTestFile("attach.xml");

             MessageFactory factory = MessageFactory.newInstance();
             SOAPMessage message = factory.createMessage();
             AttachmentPart ap = message.createAttachmentPart();
             MimeHeader mh = null;

             //Setting Mime Header
             ap.setMimeHeader("Content-Description", "some text");

             //Setting Content Id Header
             ap.setContentId("id@abc.com");

             //Setting Content
             ap.setContent(new StreamSource(in1), "text/xml");

             //Clearing Content
             ap.clearContent();

             try {

                 //Getting Content
                 InputStream is = (InputStream)ap.getContent();
                 fail("Error: SOAPException should have been thrown");
             } catch (SOAPException e) {
                 //Error thrown.(expected)
             }

             Iterator iterator = ap.getAllMimeHeaders();
             int cnt = 0;
             boolean foundHeader1 = false;
             boolean foundHeader2 = false;
             boolean foundDefaultHeader = false;
             while (iterator.hasNext()) {
                 cnt++;
                 mh = (MimeHeader)iterator.next();
                 String name = mh.getName();
                 String value = mh.getValue();
                 if (name.equals("Content-Description") && value.equals("some text")) {
                     if (!foundHeader1) {
                         foundHeader1 = true;
                         //MimeHeaders do match for header1
                     } else {
                         fail("Error: Received the same header1 header twice");
                     }
                 } else if (name.equalsIgnoreCase("Content-Id") && value.equals("id@abc.com")) {
                     if (!foundHeader2) {
                         foundHeader2 = true;
                         //MimeHeaders do match for header2
                     } else {
                         fail("Error: Received the same header2 header twice");
                     }
                 } else if (name.equals("Content-Type") && value.equals("text/xml")) {
                     if (!foundDefaultHeader) {
                         foundDefaultHeader = true;
                         //MimeHeaders do match for default header
                     } else {
                         fail("Error: Received the same default header header twice");
                     }
                 } else {
                     fail("Error: Received an invalid header");
                 }
             }

             if (!(foundHeader1 && foundHeader2)) {
                 fail("Error: did not receive both headers");
             }

         } catch (Exception e) {
             fail("Exception: " + e);
         }

     }

     @Validated @Test
     public void testGetContent() throws Exception {
         MessageFactory factory = MessageFactory.newInstance();
         SOAPMessage msg = factory.createMessage();
         AttachmentPart ap = msg.createAttachmentPart();
         Image image = ImageIO.read(TestUtils.getTestFileURL("attach.gif"));
         ap = msg.createAttachmentPart(image, "image/gif");

         //Getting Content should return an Image object
         Object o = ap.getContent();
         if (o != null) {
             if (o instanceof Image) {
                 //Image object was returned (ok)
             } else {
                 fail("Unexpected object was returned");
             }
         }
     }

     @Validated @Test
     public void testGetRawContents() {
         try {
             MessageFactory factory = MessageFactory.newInstance();
             SOAPMessage msg = factory.createMessage();
             AttachmentPart ap = msg.createAttachmentPart();
             ap = msg.createAttachmentPart();
             byte data1[] = null;
             data1 = ap.getRawContentBytes();

         } catch (SOAPException e) {
             //Caught expected SOAPException
         } catch (NullPointerException e) {
             //Caught expected NullPointerException
         } catch (Exception e) {
             fail();
         }
     }

     @Validated @Test
     public void testSetBase64Content() throws Exception {
         MessageFactory factory = MessageFactory.newInstance();
         SOAPMessage msg = factory.createMessage();
         AttachmentPart ap = msg.createAttachmentPart();

         byte[] bytes = new byte[4096];
         new Random(1234).nextBytes(bytes);
         ap.setBase64Content(
                 new ByteArrayInputStream(Base64.encodeBase64(bytes, false)),
                 "application/octet-stream");

         InputStream r = ap.getBase64Content();
         try {
             assertThat(Base64.decodeBase64(IOUtils.toByteArray(r))).isEqualTo(bytes);
         } finally {
             r.close();
         }
     }
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one
	* or more contributor license agreements. See the NOTICE file
	* distributed with this work for additional information
	* regarding copyright ownership. The ASF licenses this file
	* to you under the Apache License, Version 2.0 (the
	* "License"); you may not use this file except in compliance
	* with the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing,
	* software distributed under the License is distributed on an
	* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
	* KIND, either express or implied. See the License for the
	* specific language governing permissions and limitations
	* under the License.
	*/

	package org.apache.axis2.saaj;

	import junit.framework.Assert;
	import org.apache.commons.codec.binary.Base64;
	import org.apache.commons.io.IOUtils;
	import org.junit.Test;
	import org.junit.runner.RunWith;

	import javax.activation.DataHandler;
	import javax.activation.DataSource;
	import javax.imageio.ImageIO;
	import javax.xml.soap.AttachmentPart;
	import javax.xml.soap.MessageFactory;
	import javax.xml.soap.MimeHeader;
	import javax.xml.soap.MimeHeaders;
	import javax.xml.soap.SOAPException;
	import javax.xml.soap.SOAPMessage;
	import javax.xml.transform.stream.StreamSource;

	import static org.assertj.core.api.Assertions.assertThat;

	import java.awt.*;
	import java.io.ByteArrayInputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.io.OutputStream;
	import java.util.Iterator;
	import java.util.Random;

	@RunWith(SAAJTestRunner.class)
	public class AttachmentTest extends Assert {
	@Validated @Test
	public void testStringAttachment() throws Exception {

	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage message = factory.createMessage();
	AttachmentPart attachment = message.createAttachmentPart();
	String stringContent = "Update address for Sunny Skies " +
	"Inc., to 10 Upbeat Street, Pleasant Grove, CA 95439";

	attachment.setContent(stringContent, "text/plain");
	attachment.setContentId("update_address");
	message.addAttachmentPart(attachment);

	assertTrue(message.countAttachments() == 1);

	Iterator it = message.getAttachments();
	while (it.hasNext()) {
	attachment = (AttachmentPart)it.next();
	Object content = attachment.getContent();
	String id = attachment.getContentId();
	assertEquals(content, stringContent);
	}
	message.removeAllAttachments();
	assertTrue(message.countAttachments() == 0);
	}

	@Validated @Test
	public void testMultipleAttachments() throws Exception {

	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage msg = factory.createMessage();

	AttachmentPart a1 = msg.createAttachmentPart(new DataHandler("<some_xml/>", "text/xml"));
	a1.setContentType("text/xml");
	msg.addAttachmentPart(a1);
	AttachmentPart a2 = msg.createAttachmentPart(new DataHandler("<some_xml/>", "text/xml"));
	a2.setContentType("text/xml");
	msg.addAttachmentPart(a2);
	AttachmentPart a3 = msg.createAttachmentPart(new DataHandler("text", "text/plain"));
	a3.setContentType("text/plain");
	msg.addAttachmentPart(a3);

	assertTrue(msg.countAttachments() == 3);

	MimeHeaders mimeHeaders = new MimeHeaders();
	mimeHeaders.addHeader("Content-Type", "text/xml");

	int nAttachments = 0;
	Iterator iterator = msg.getAttachments(mimeHeaders);
	while (iterator.hasNext()) {
	nAttachments++;
	AttachmentPart ap = (AttachmentPart)iterator.next();
	assertTrue(ap.equals(a1) \|\| ap.equals(a2));
	}
	assertTrue(nAttachments == 2);
	}

	// Note: This test case fails with Sun's SAAJ implementation
	// and can't be @Validated.
	@Test
	public void testBadAttSize() throws Exception {
	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage message = factory.createMessage();

	ByteArrayInputStream ins = new ByteArrayInputStream(new byte[5]);
	DataHandler dh = new DataHandler(new Src(ins, "text/plain"));
	AttachmentPart part = message.createAttachmentPart(dh);
	assertEquals("Size should match", 5, part.getSize());
	}

	class Src implements DataSource {
	InputStream m_src;
	String m_type;

	public Src(InputStream data, String type) {
	m_src = data;
	m_type = type;
	}

	public String getContentType() {
	return m_type;
	}

	public InputStream getInputStream() throws IOException {
	m_src.reset();
	return m_src;
	}

	public String getName() {
	return "Some-Data";
	}

	public OutputStream getOutputStream() {
	throw new UnsupportedOperationException("I don't give output streams");
	}
	}

	@Validated @Test
	public void testClearContent() throws Exception {
	try {
	InputStream in1 = TestUtils.getTestFile("attach.xml");

	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage message = factory.createMessage();
	AttachmentPart ap = message.createAttachmentPart();
	MimeHeader mh = null;

	//Setting Mime Header
	ap.setMimeHeader("Content-Description", "some text");

	//Setting Content Id Header
	ap.setContentId("id@abc.com");

	//Setting Content
	ap.setContent(new StreamSource(in1), "text/xml");

	//Clearing Content
	ap.clearContent();

	try {

	//Getting Content
	InputStream is = (InputStream)ap.getContent();
	fail("Error: SOAPException should have been thrown");
	} catch (SOAPException e) {
	//Error thrown.(expected)
	}

	Iterator iterator = ap.getAllMimeHeaders();
	int cnt = 0;
	boolean foundHeader1 = false;
	boolean foundHeader2 = false;
	boolean foundDefaultHeader = false;
	while (iterator.hasNext()) {
	cnt++;
	mh = (MimeHeader)iterator.next();
	String name = mh.getName();
	String value = mh.getValue();
	if (name.equals("Content-Description") && value.equals("some text")) {
	if (!foundHeader1) {
	foundHeader1 = true;
	//MimeHeaders do match for header1
	} else {
	fail("Error: Received the same header1 header twice");
	}
	} else if (name.equalsIgnoreCase("Content-Id") && value.equals("id@abc.com")) {
	if (!foundHeader2) {
	foundHeader2 = true;
	//MimeHeaders do match for header2
	} else {
	fail("Error: Received the same header2 header twice");
	}
	} else if (name.equals("Content-Type") && value.equals("text/xml")) {
	if (!foundDefaultHeader) {
	foundDefaultHeader = true;
	//MimeHeaders do match for default header
	} else {
	fail("Error: Received the same default header header twice");
	}
	} else {
	fail("Error: Received an invalid header");
	}
	}

	if (!(foundHeader1 && foundHeader2)) {
	fail("Error: did not receive both headers");
	}

	} catch (Exception e) {
	fail("Exception: " + e);
	}

	}

	@Validated @Test
	public void testGetContent() throws Exception {
	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage msg = factory.createMessage();
	AttachmentPart ap = msg.createAttachmentPart();
	Image image = ImageIO.read(TestUtils.getTestFileURL("attach.gif"));
	ap = msg.createAttachmentPart(image, "image/gif");

	//Getting Content should return an Image object
	Object o = ap.getContent();
	if (o != null) {
	if (o instanceof Image) {
	//Image object was returned (ok)
	} else {
	fail("Unexpected object was returned");
	}
	}
	}

	@Validated @Test
	public void testGetRawContents() {
	try {
	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage msg = factory.createMessage();
	AttachmentPart ap = msg.createAttachmentPart();
	ap = msg.createAttachmentPart();
	byte data1[] = null;
	data1 = ap.getRawContentBytes();

	} catch (SOAPException e) {
	//Caught expected SOAPException
	} catch (NullPointerException e) {
	//Caught expected NullPointerException
	} catch (Exception e) {
	fail();
	}
	}

	@Validated @Test
	public void testSetBase64Content() throws Exception {
	MessageFactory factory = MessageFactory.newInstance();
	SOAPMessage msg = factory.createMessage();
	AttachmentPart ap = msg.createAttachmentPart();

	byte[] bytes = new byte[4096];
	new Random(1234).nextBytes(bytes);
	ap.setBase64Content(
	new ByteArrayInputStream(Base64.encodeBase64(bytes, false)),
	"application/octet-stream");

	InputStream r = ap.getBase64Content();
	try {
	assertThat(Base64.decodeBase64(IOUtils.toByteArray(r))).isEqualTo(bytes);
	} finally {
	r.close();
	}
	}
	}