1 /* 2 * Copyright (c) 2014, 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 package stream.XMLStreamWriterTest; 25 26 import java.io.ByteArrayOutputStream; 27 28 import javax.xml.stream.XMLOutputFactory; 29 import javax.xml.stream.XMLStreamWriter; 30 31 import org.testng.Assert; 32 import org.testng.annotations.Listeners; 33 import org.testng.annotations.Test; 34 35 /* 36 * @summary Test XMLStreamWriter writes a document with encoding setting. 37 */ 38 @Listeners({jaxp.library.BasePolicy.class}) 39 public class EncodingTest { 40 41 private static final XMLOutputFactory XML_OUTPUT_FACTORY = XMLOutputFactory.newInstance(); 42 43 /* 44 * Tests writing a document with UTF-8 encoding, by setting UTF-8 on writer. 45 */ 46 @Test 47 public void testWriteStartDocumentUTF8() { 48 49 final String EXPECTED_OUTPUT = "<?xml version=\"1.0\" encoding=\"UTF-8\"?><root></root>"; 50 XMLStreamWriter writer = null; 51 ByteArrayOutputStream byteArrayOutputStream = null; 52 53 try { 54 byteArrayOutputStream = new ByteArrayOutputStream(); 55 writer = XML_OUTPUT_FACTORY.createXMLStreamWriter(byteArrayOutputStream, "UTF-8"); 56 57 writer.writeStartDocument("UTF-8", "1.0"); 58 writer.writeStartElement("root"); 59 writer.writeEndElement(); 60 writer.writeEndDocument(); 61 writer.flush(); 62 63 String actualOutput = byteArrayOutputStream.toString(); 64 Assert.assertEquals(EXPECTED_OUTPUT, actualOutput); 65 66 } catch (Exception e) { 67 e.printStackTrace(); 68 Assert.fail(e.toString()); 69 } 70 71 } 72 73 /* 74 * Tests writing a document with UTF-8 encoding on default enocding writer. 75 * This scenario should result in an exception as default encoding is ASCII. 76 */ 77 @Test 78 public void testWriteStartDocumentUTF8Fail() { 79 80 XMLStreamWriter writer = null; 81 ByteArrayOutputStream byteArrayOutputStream = null; 82 83 // pick a different encoding to use v. default encoding 84 String defaultCharset = java.nio.charset.Charset.defaultCharset().name(); 85 String useCharset = "UTF-8"; 86 if (useCharset.equals(defaultCharset)) { 87 useCharset = "US-ASCII"; 88 } 89 90 System.out.println("defaultCharset = " + defaultCharset + ", useCharset = " + useCharset); 91 92 try { 93 byteArrayOutputStream = new ByteArrayOutputStream(); 94 writer = XML_OUTPUT_FACTORY.createXMLStreamWriter(byteArrayOutputStream); 95 96 writer.writeStartDocument(useCharset, "1.0"); 97 writer.writeStartElement("root"); 98 writer.writeEndElement(); 99 writer.writeEndDocument(); 100 writer.flush(); 101 102 Assert.fail("Expected XMLStreamException as default underlying stream encoding of " + defaultCharset 103 + " differs from explicitly specified encoding of " + useCharset); 104 } catch (Exception e) { 105 e.printStackTrace(); 106 } 107 } 108 }