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