< prev index next >

test/jdk/java/io/ByteArrayOutputStream/Write.java

Print this page

        

@@ -1,7 +1,7 @@
 /*
- * Copyright (c) 1997, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 1997, 2018, 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.

@@ -19,58 +19,124 @@
  * 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 4017158
-   @summary Check for correct implementation of ByteArrayInputStream.write
+/*
+ * @test
+ * @bug 4017158 8180410
+ * @library /test/lib
+ * @build jdk.test.lib.RandomFactory
+ * @run testng Write
+ * @summary Check for correct implementation of ByteArrayInputStream.write
+ * @key randomness
    */
 
-import java.io.*;
-
+import java.io.ByteArrayOutputStream;
+import java.util.Arrays;
+import java.util.Random;
+import jdk.test.lib.RandomFactory;
+import org.testng.annotations.Test;
+import static org.testng.Assert.*;
 
-public class WriteBounds{
-
-    private static void dotest(byte[] b, int off, int len,
+public class Write {
+    private static void doBoundsTest(byte[] b, int off, int len,
                                ByteArrayOutputStream baos)
-        throws Exception
-    {
-
+        throws Exception {
         if (b != null) {
-            System.err.println("ByteArrayOutStream.write -- b.length = " +
+            System.out.println("ByteArrayOutStream.write: b.length = " +
                                b.length + " off = " + off + " len = " + len);
-        }
-        else{
-            System.err.println("ByteArrayOutStream.write - b is null off = " +
+        } else{
+            System.out.println("ByteArrayOutStream.write: b is null off = " +
                                off + " len = " + len);
         }
 
         try {
             baos.write(b, off, len);
         } catch (IndexOutOfBoundsException e) {
-            System.err.println("IndexOutOfBoundsException is thrown -- OKAY");
+            System.out.println("IndexOutOfBoundsException is thrown: OKAY");
         } catch (NullPointerException e) {
-            System.err.println("NullPointerException is thrown -- OKAY");
+            System.out.println("NullPointerException is thrown: OKAY");
         } catch (Throwable e){
-            throw new RuntimeException("Unexpected Exception is thrown");
+            throw new RuntimeException("Unexpected Exception is thrown", e);
         }
 
+        if (b != null) {
+            System.out.println("ByteArrayOutStream.writeBytes: b.length = " +
+                               b.length);
+        } else{
+            System.out.println("ByteArrayOutStream.writeBytes: b is null");
     }
 
-    public static void main( String argv[] ) throws Exception {
+        try {
+            baos.writeBytes(b);
+        } catch (NullPointerException e) {
+            System.out.println("NullPointerException is thrown: OKAY");
+        } catch (Throwable e){
+            throw new RuntimeException("Unexpected Exception is thrown", e);
+        }
+    }
 
-        ByteArrayOutputStream y1;
-        byte array1[]={1 , 2 , 3 , 4 , 5};     // Simple array
+    @Test
+    public static void boundsTest() throws Exception {
+        byte array1[] = {1 , 2 , 3 , 4 , 5};     // Simple array
 
         //Create new ByteArrayOutputStream object
-        y1 = new ByteArrayOutputStream(5);
-
-        dotest(array1, 0, Integer.MAX_VALUE , y1);
-        dotest(array1, 0, array1.length+100, y1);
-        dotest(array1, -1, 2, y1);
-        dotest(array1, 0, -1, y1);
-        dotest(null, 0, 2, y1);
+        ByteArrayOutputStream y1 = new ByteArrayOutputStream(5);
 
+        doBoundsTest(array1, 0, Integer.MAX_VALUE , y1);
+        doBoundsTest(array1, 0, array1.length+100, y1);
+        doBoundsTest(array1, -1, 2, y1);
+        doBoundsTest(array1, 0, -1, y1);
+        doBoundsTest(null, 0, 2, y1);
     }
 
+    @Test
+    public static void writeTest() throws Exception {
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        Random rnd = RandomFactory.getRandom();
+        final int size = 17 + rnd.nextInt(128);
+
+        byte[] b = new byte[size];
+        rnd.nextBytes(b);
+
+        int off1 = rnd.nextInt(size / 4) + 1;
+        int len1 = Math.min(rnd.nextInt(size / 4) + 1, size - off1);
+        int off2 = rnd.nextInt(size / 2) + 1;
+        int len2 = Math.min(rnd.nextInt(size / 2) + 1, size - off2);
+
+        System.out.format("size: %d, off1: %d, len1: %d, off2: %d, len2: %d%n",
+            size, off1, len1, off2, len2);
+
+        baos.write(b, off1, len1);
+        byte[] b1 = baos.toByteArray();
+        assertEquals(b1.length, len1, "Array length test 1 failed.");
+        assertEquals(b1, Arrays.copyOfRange(b, off1, off1 + len1),
+            "Array equality test 1 failed.");
+
+        baos.write(b, off2, len2);
+        byte[] b2 = baos.toByteArray();
+        assertEquals(b2.length, len1 + len2, "Array length test 2 failed.");
+        assertEquals(Arrays.copyOfRange(b2, 0, len1),
+             Arrays.copyOfRange(b, off1, off1 + len1),
+            "Array equality test 2A failed.");
+        assertEquals(Arrays.copyOfRange(b2, len1, len1 + len2),
+            Arrays.copyOfRange(b, off2, off2 + len2),
+            "Array equality test 2B failed.");
+
+        baos.writeBytes(b);
+        byte[] b3 = baos.toByteArray();
+        int len3 = len1 + len2 + b.length;
+        if (b3.length != len1 + len2 + b.length) {
+            throw new RuntimeException("Array length test 3 failed.");
+        }
+        assertEquals(b3.length, len3, "Array length test 3 failed.");
+        assertEquals(Arrays.copyOfRange(b3, 0, len1),
+             Arrays.copyOfRange(b, off1, off1 + len1),
+            "Array equality test 3A failed.");
+        assertEquals(Arrays.copyOfRange(b3, len1, len1 + len2),
+            Arrays.copyOfRange(b, off2, off2 + len2),
+            "Array equality test 3B failed.");
+        assertEquals(Arrays.copyOfRange(b3, len1 + len2, len3), b,
+            "Array equality test 3C failed.");
+    }
 }
< prev index next >