1 /* 2 * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 import javax.xml.soap.AttachmentPart; 25 import javax.xml.soap.MessageFactory; 26 import javax.xml.soap.Name; 27 import javax.xml.soap.SOAPEnvelope; 28 import javax.xml.soap.SOAPException; 29 import javax.xml.soap.SOAPMessage; 30 import javax.xml.transform.stream.StreamSource; 31 import java.io.ByteArrayInputStream; 32 import java.io.ByteArrayOutputStream; 33 import java.io.File; 34 import java.io.FileInputStream; 35 import java.io.FileOutputStream; 36 import java.io.IOException; 37 import java.util.Iterator; 38 39 /* 40 * @test 41 * @compile -addmods java.xml.ws XmlTest.java 42 * @run main/othervm -addmods java.xml.ws XmlTest 43 * @summary tests JAF DataHandler can be instantiated; test serialize and 44 * deserialize SOAP message containing xml attachment 45 */ 46 public class XmlTest { 47 48 public void test() throws Exception { 49 50 File file = new File("message.xml"); 51 file.deleteOnExit(); 52 53 MessageFactory mf = MessageFactory.newInstance(); 54 SOAPMessage msg = createMessage(mf); 55 56 // Save the soap message to file 57 try (FileOutputStream sentFile = new FileOutputStream(file)) { 58 msg.writeTo(sentFile); 59 } 60 61 // See if we get the image object back 62 try (FileInputStream fin = new FileInputStream(file)) { 63 SOAPMessage newMsg = mf.createMessage(msg.getMimeHeaders(), fin); 64 65 newMsg.writeTo(new ByteArrayOutputStream()); 66 67 Iterator<?> i = newMsg.getAttachments(); 68 while (i.hasNext()) { 69 AttachmentPart att = (AttachmentPart) i.next(); 70 Object obj = att.getContent(); 71 if (!(obj instanceof StreamSource)) { 72 fail("Got incorrect attachment type [" + obj.getClass() + "], " + 73 "expected [javax.xml.transform.stream.StreamSource]"); 74 } 75 } 76 } 77 78 } 79 80 private SOAPMessage createMessage(MessageFactory mf) throws SOAPException, IOException { 81 SOAPMessage msg = mf.createMessage(); 82 SOAPEnvelope envelope = msg.getSOAPPart().getEnvelope(); 83 Name name = envelope.createName("hello", "ex", "http://example.com"); 84 envelope.getBody().addChildElement(name).addTextNode("THERE!"); 85 86 String s = "<root><hello>THERE!</hello></root>"; 87 88 AttachmentPart ap = msg.createAttachmentPart( 89 new StreamSource(new ByteArrayInputStream(s.getBytes())), 90 "text/xml" 91 ); 92 msg.addAttachmentPart(ap); 93 msg.saveChanges(); 94 95 return msg; 96 } 97 98 private void fail(String s) { 99 throw new RuntimeException(s); 100 } 101 102 public static void main(String[] args) throws Exception { 103 new XmlTest().test(); 104 } 105 106 }