1/*
2 * Copyright (c) 2008, 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/* @test
25 * @bug 5005426
26 * @summary Check if StreamEncoder close() method works correctly from
27 *          error recovery after the underneath OutputStream failed to
28 *          close the first time.
29 * @modules jdk.charsets
30 */
31
32import java.io.*;
33public class StreamEncoderClose {
34    public static void main( String arg[] ) throws Exception {
35        byte[] expected = {(byte)0x1b,(byte)0x24,(byte)0x42,
36                           (byte)0x30,(byte)0x6c,
37                           (byte)0x1b,(byte)0x28,(byte)0x42};
38        ByteArrayOutputStream baos = new ByteArrayOutputStream();
39        MyBufferedOutputStream mbos = new MyBufferedOutputStream(baos);
40        PrintWriter pw = new PrintWriter(new OutputStreamWriter(mbos, "ISO-2022-JP"));
41        mbos.dontClose();
42        pw.write("\u4e00");
43        pw.close();             //  1st PrintWriter Close
44        mbos.canClose();
45        pw.close();             //  2nd PrintWriter Close
46
47        //double check, probably not necessary
48        byte[] out = baos.toByteArray();
49        if (out.length != expected.length) {
50            throw new IOException("Failed");
51        }
52        for (int i = 0; i < out.length; i++) {
53            //System.out.printf("(byte)0x%x,", out[i] & 0xff);
54            if (out[i] != expected[i])
55                throw new IOException("Failed");
56        }
57    }
58
59    static class MyBufferedOutputStream extends BufferedOutputStream {
60        MyBufferedOutputStream(OutputStream os) {
61            super(os);
62        }
63        private boolean status;
64        public void dontClose() {
65            status = false;
66        }
67        public void canClose() {
68            status = true;
69        }
70        public void close() throws IOException {
71            if ( status == false ) {
72                throw new IOException("Can't close ");
73            }
74            super.close();
75        }
76    }
77}
78