Merged HEAD-QA to HEAD (4.2) (including moving test classes into separate folders)

51903 to 54309 


git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@54310 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261

source/test-java/org/alfresco/repo/content/LimitedStreamCopierTest.java

@@ -0,0 +1,121 @@
+/*
+ * Copyright (C) 2005-2011 Alfresco Software Limited.
+ *
+ * This file is part of Alfresco
+ *
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Alfresco 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.alfresco.repo.content;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+
+import org.junit.Before;
+import org.junit.Test;
+
+/**
+ * Unit test for {@link LimitedStreamCopier}.
+ * 
+ * @author Neil Mc Erlean
+ * @since Thor
+ */
+public class LimitedStreamCopierTest
+{
+    private final static byte[] ZERO_BYTE_ARRAY =        "".getBytes();
+    private final static byte[] SHORT_BYTE_ARRAY =       "This string is shorter than the limit".getBytes();
+    private final static byte[] LIMIT_SIZED_BYTE_ARRAY = "This string's length exactly equals the limit".getBytes();
+    private final static byte[] LONG_BYTE_ARRAY =        "This test string is longer than the limit. Tum te tum te tum te tum.".getBytes();
+    
+    private final static int SIZE_LIMIT = LIMIT_SIZED_BYTE_ARRAY.length;
+    
+    private LimitedStreamCopier streamCopier = new LimitedStreamCopier();
+    private InputStream in;
+    private ByteArrayOutputStream out;
+    
+    @Before public void initStreamCopier()
+    {
+        out = new ByteArrayOutputStream();
+    }
+    
+    /**
+     * Test copying a simple byte[]
+     */
+    @Test public void copyStreamSuccessful() throws Exception
+    {
+        in = new ByteArrayInputStream(SHORT_BYTE_ARRAY);
+        
+        int bytesCopied = streamCopier.copyStreams(in, out, SIZE_LIMIT);
+        
+        assertArrayEquals(SHORT_BYTE_ARRAY, out.toByteArray());
+        assertEquals(SHORT_BYTE_ARRAY.length, bytesCopied);
+    }
+    
+    /**
+     * Test copying an empty byte[]
+     */
+    @Test public void copyStreamSuccessfulZeroBytes() throws Exception
+    {
+        in = new ByteArrayInputStream(ZERO_BYTE_ARRAY);
+        
+        int bytesCopied = streamCopier.copyStreams(in, out, SIZE_LIMIT);
+        
+        assertArrayEquals(ZERO_BYTE_ARRAY, out.toByteArray());
+        assertEquals(ZERO_BYTE_ARRAY.length, bytesCopied);
+    }
+    
+    /**
+     * Test copying a byte[] that exceeds the limit.
+     */
+    @Test (expected=ContentLimitViolationException.class)
+    public void copyStreamUnsuccessfulLimitExceeded() throws Exception
+    {
+        in = new ByteArrayInputStream(LONG_BYTE_ARRAY);
+        
+        int bytesCopied = streamCopier.copyStreams(in, out, SIZE_LIMIT);
+        
+        assertArrayEquals(LONG_BYTE_ARRAY, out.toByteArray());
+        assertEquals(LONG_BYTE_ARRAY.length, bytesCopied);
+    }
+    
+    /**
+     * Test copying a byte[] that equals the limit.
+     */
+    @Test
+    public void copyStreamSuccessfulLimitHit() throws Exception
+    {
+        in = new ByteArrayInputStream(LIMIT_SIZED_BYTE_ARRAY);
+        
+        int bytesCopied = streamCopier.copyStreams(in, out, SIZE_LIMIT);
+        
+        assertArrayEquals(LIMIT_SIZED_BYTE_ARRAY, out.toByteArray());
+        assertEquals(LIMIT_SIZED_BYTE_ARRAY.length, bytesCopied);
+    }
+    
+    /**
+     * Test copying a simple byte[] with no limit sent.
+     */
+    @Test public void copyStreamSuccessfulBecauseLimitNotImposed() throws Exception
+    {
+        in = new ByteArrayInputStream(LONG_BYTE_ARRAY);
+        
+        int bytesCopied = streamCopier.copyStreams(in, out, -1);
+        
+        assertArrayEquals(LONG_BYTE_ARRAY, out.toByteArray());
+        assertEquals(LONG_BYTE_ARRAY.length, bytesCopied);
+    }
+}