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