--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/javax/imageio/stream/StreamFlush.java Mon May 22 19:54:23 2017 -0700
@@ -0,0 +1,108 @@
+/*
+ * Copyright (c) 2001, 2017, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+/*
+ * @test
+ * @bug 4414990 4415041
+ * @summary Checks that the output is flushed properly when using various
+ * ImageOutputStreams and writers
+ */
+
+import java.awt.image.BufferedImage;
+import java.io.BufferedOutputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+
+import javax.imageio.ImageIO;
+import javax.imageio.stream.ImageOutputStream;
+
+public class StreamFlush {
+
+ public static void main(String[] args) throws IOException {
+ ImageIO.setUseCache(true);
+
+ // Create a FileImageOutputStream from a FileOutputStream
+ File temp1 = File.createTempFile("imageio", ".tmp");
+ temp1.deleteOnExit();
+ ImageOutputStream fios = ImageIO.createImageOutputStream(temp1);
+
+ // Create a FileCacheImageOutputStream from a BufferedOutputStream
+ File temp2 = File.createTempFile("imageio", ".tmp");
+ temp2.deleteOnExit();
+ FileOutputStream fos2 = new FileOutputStream(temp2);
+ BufferedOutputStream bos = new BufferedOutputStream(fos2);
+ ImageOutputStream fcios1 = ImageIO.createImageOutputStream(bos);
+
+ // Create a FileCacheImageOutputStream from a ByteArrayOutputStream
+ ByteArrayOutputStream baos = new ByteArrayOutputStream();
+ ImageOutputStream fcios2 = ImageIO.createImageOutputStream(baos);
+
+ BufferedImage bi =
+ new BufferedImage(10, 10, BufferedImage.TYPE_3BYTE_BGR);
+
+ ImageIO.write(bi, "jpg", fios); // No bug, check it anyway
+ ImageIO.write(bi, "png", fcios1); // Bug 4414990
+ ImageIO.write(bi, "jpg", fcios2); // Bug 4415041
+
+ // It should not be necessary to flush any of the streams
+ // If flushing does make a difference, it indicates a bug
+ // in the writer or the stream implementation
+
+ // Get length of temp1 before and after flushing
+ long file1NoFlushLength = temp1.length();
+ fios.flush();
+ long file1FlushLength = temp1.length();
+
+ // Get length of temp2 before and after flushing
+ long file2NoFlushLength = temp2.length();
+ fcios1.flush();
+ bos.flush();
+ long file2FlushLength = temp2.length();
+
+ byte[] b0 = baos.toByteArray();
+ int cacheNoFlushLength = b0.length;
+ fcios2.flush();
+ byte[] b1 = baos.toByteArray();
+ int cacheFlushLength = b1.length;
+
+ if (file1NoFlushLength != file1FlushLength) {
+ // throw new RuntimeException
+ System.out.println
+ ("FileImageOutputStream not flushed!");
+ }
+
+ if (file2NoFlushLength != file2FlushLength) {
+ // throw new RuntimeException
+ System.out.println
+ ("FileCacheImageOutputStream/BufferedOutputStream not flushed!");
+ }
+
+ if (cacheNoFlushLength != cacheFlushLength) {
+ // throw new RuntimeException
+ System.out.println
+ ("FileCacheImageOutputStream/ByteArrayOutputStream not flushed!");
+ }
+ }
+}