IO-84 - add tests for files > 2GB to IOUtils.copy(), CountingInputStream and CountingOutputStream

git-svn-id: https://svn.apache.org/repos/asf/jakarta/commons/proper/io/trunk@463537 13f79535-47bb-0310-9956-ffa450edef68

Author: Niall Pemberton

src/test/org/apache/commons/io/IOUtilsCopyTestCase.java

@@ -21,6 +21,7 @@
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.io.OutputStreamWriter;
+import java.io.IOException;
 import java.io.Reader;
 import java.io.Writer;
 import java.util.Arrays;
@@ -29,7 +30,11 @@
 import junit.framework.TestSuite;
 import junit.textui.TestRunner;
 
+import org.apache.commons.io.input.NullInputStream;
+import org.apache.commons.io.input.NullReader;
 import org.apache.commons.io.output.ByteArrayOutputStream;
+import org.apache.commons.io.output.NullOutputStream;
+import org.apache.commons.io.output.NullWriter;
 import org.apache.commons.io.testtools.FileBasedTestCase;
 import org.apache.commons.io.testtools.YellOnCloseInputStream;
 import org.apache.commons.io.testtools.YellOnFlushAndCloseOutputStream;
@@ -111,6 +116,29 @@ public void testCopy_inputStreamToOutputStream_nullOut() throws Exception {
         } catch (NullPointerException ex) {}
     }
 
+    /**
+     * Test Copying file > 2GB  - see issue# IO-84
+     */
+    public void testCopy_inputStreamToOutputStream_IO84() throws Exception {
+        long size = (long)Integer.MAX_VALUE + (long)1;
+        InputStream  in  = new NullInputStream(size);
+        OutputStream out = new NullOutputStream();
+
+        // Test copy() method
+        try {
+            IOUtils.copy(in, out);
+            fail("Expected copy() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+
+        // reset the input
+        in.close();
+
+        // Test copyLarge() method
+        assertEquals("copyLarge()", size, IOUtils.copyLarge(in, out));
+    }
+
     //-----------------------------------------------------------------------
     public void testCopy_inputStreamToWriter() throws Exception {
         InputStream in = new ByteArrayInputStream(inData);
@@ -332,4 +360,28 @@ public void testCopy_readerToWriter_nullOut() throws Exception {
         } catch (NullPointerException ex) {}
     }
 
+    /**
+     * Test Copying file > 2GB  - see issue# IO-84
+     */
+    public void testCopy_readerToWriter_IO84() throws Exception {
+        long size = (long)Integer.MAX_VALUE + (long)1;
+        Reader reader = new NullReader(size);
+        Writer writer = new NullWriter();
+
+        // Test copy() method
+        try {
+            IOUtils.copy(reader, writer);
+            fail("Expected copy() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+
+        // reset the input
+        reader.close();
+
+        // Test copyLarge() method
+        assertEquals("copyLarge()", size, IOUtils.copyLarge(reader, writer));
+
+    }
+
 }

src/test/org/apache/commons/io/input/CountingInputStreamTest.java

@@ -17,9 +17,12 @@
 package org.apache.commons.io.input;
 
 import java.io.ByteArrayInputStream;
+import java.io.OutputStream;
 import java.io.IOException;
 
 import junit.framework.TestCase;
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.io.output.NullOutputStream;
 
 /**
  * Tests the CountingInputStream.
@@ -67,6 +70,39 @@ public void testCounting() throws Exception {
         assertEquals(textResult, text);
     }
 
+
+    /**
+     * Test for files > 2GB in size - see issue IO-84
+     */
+    public void testLargeFiles_IO84() throws Exception {
+        long size = (long)Integer.MAX_VALUE + (long)1;
+        NullInputStream mock    = new NullInputStream(size);
+        CountingInputStream cis = new CountingInputStream(mock);
+        OutputStream out        = new NullOutputStream();
+
+        // Test integer methods
+        IOUtils.copyLarge(cis, out);
+        try {
+            cis.getCount();
+            fail("Expected getCount() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+        try {
+            cis.resetCount();
+            fail("Expected resetCount() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+
+        mock.close();
+
+        // Test long methods
+        IOUtils.copyLarge(cis, out);
+        assertEquals("getByteCount()",   size, cis.getByteCount());
+        assertEquals("resetByteCount()", size, cis.resetByteCount());
+    }
+
     public void testResetting() throws Exception {
         String text = "A piece of text";
         byte[] bytes = text.getBytes();

src/test/org/apache/commons/io/output/CountingOutputStreamTest.java

@@ -17,10 +17,13 @@
 package org.apache.commons.io.output;
 
 
+import java.io.OutputStream;
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 
 import junit.framework.TestCase;
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.io.input.NullInputStream;
 
 
 /**
@@ -70,6 +73,39 @@ public void testCounting() throws IOException {
 
     }
 
+    /**
+     * Test for files > 2GB in size - see issue IO-84
+     */
+    public void testLargeFiles_IO84() throws Exception {
+        long size = (long)Integer.MAX_VALUE + (long)1;
+
+        NullInputStream mock     = new NullInputStream(size);
+        OutputStream nos         = new NullOutputStream();
+        CountingOutputStream cos = new CountingOutputStream(nos);
+
+        // Test integer methods
+        IOUtils.copyLarge(mock, cos);
+        try {
+            cos.getCount();
+            fail("Expected getCount() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+        try {
+            cos.resetCount();
+            fail("Expected resetCount() to throw an ArithmeticException");
+        } catch (ArithmeticException ae) {
+            // expected result
+        }
+
+        mock.close();
+
+        // Test long methods
+        IOUtils.copyLarge(mock, cos);
+        assertEquals("getByteCount()",   size, cos.getByteCount());
+        assertEquals("resetByteCount()", size, cos.resetByteCount());
+    }
+
     private void assertByteArrayEquals(String msg, byte[] array, int start, int end) {
         for (int i = start; i < end; i++) {
             assertEquals(msg+": array[" + i + "] mismatch", array[i], i-start);