From c636b3798f368ebfdd6fed7839e24f2072746e51 Mon Sep 17 00:00:00 2001
From: "Adrian T. Co" <aco@apache.org>
Date: Tue, 7 Nov 2006 21:00:46 +0000
Subject: [PATCH] Added test case to test resizing of internal buffer of
 DataByteArrayOutputStream. Fix bug during resizing of internal buffer of
 DataByteArrayOutputStream.

git-svn-id: https://svn.apache.org/repos/asf/incubator/activemq/trunk@472258 13f79535-47bb-0310-9956-ffa450edef68
---
 .../util/DataByteArrayOutputStream.java       |  14 +--
 .../util/DataByteArrayOutputStreamTest.java   | 101 ++++++++++++++++++
 2 files changed, 108 insertions(+), 7 deletions(-)
 create mode 100644 activemq-core/src/test/java/org/apache/activemq/util/DataByteArrayOutputStreamTest.java

diff --git a/activemq-core/src/main/java/org/apache/activemq/util/DataByteArrayOutputStream.java b/activemq-core/src/main/java/org/apache/activemq/util/DataByteArrayOutputStream.java
index a90d0fb6e5..fbbab73d0c 100755
--- a/activemq-core/src/main/java/org/apache/activemq/util/DataByteArrayOutputStream.java
+++ b/activemq-core/src/main/java/org/apache/activemq/util/DataByteArrayOutputStream.java
@@ -138,29 +138,29 @@ public final class DataByteArrayOutputStream extends OutputStream implements Dat
     
 
     public void writeBoolean(boolean v){
-        ensureEnoughBuffer(1);
+        ensureEnoughBuffer(pos + 1);
         buf[pos++]=(byte) (v?1:0);
     }
 
     public void writeByte(int v){
-        ensureEnoughBuffer(1);
+        ensureEnoughBuffer(pos + 1);
         buf[pos++]=(byte) (v>>>0);
     }
 
     public void writeShort(int v){
-        ensureEnoughBuffer(2);
+        ensureEnoughBuffer(pos + 2);
         buf[pos++]=(byte) (v>>>8);
         buf[pos++]=(byte) (v>>>0);
     }
 
     public void writeChar(int v){
-        ensureEnoughBuffer(2);
+        ensureEnoughBuffer(pos + 2);
         buf[pos++]=(byte) (v>>>8);
         buf[pos++]=(byte) (v>>>0);
     }
 
     public void writeInt(int v){
-        ensureEnoughBuffer(4);
+        ensureEnoughBuffer(pos + 4);
         buf[pos++]=(byte) (v>>>24);
         buf[pos++]=(byte) (v>>>16);
         buf[pos++]=(byte) (v>>>8);
@@ -168,7 +168,7 @@ public final class DataByteArrayOutputStream extends OutputStream implements Dat
     }
 
     public void writeLong(long v){
-        ensureEnoughBuffer(8);
+        ensureEnoughBuffer(pos + 8);
         buf[pos++]=(byte) (v>>>56);
         buf[pos++]=(byte) (v>>>48);
         buf[pos++]=(byte) (v>>>40);
@@ -219,7 +219,7 @@ public final class DataByteArrayOutputStream extends OutputStream implements Dat
         }
         if(encodedsize>65535)
             throw new UTFDataFormatException("encoded string too long: "+encodedsize+" bytes");
-        ensureEnoughBuffer(encodedsize+2);
+        ensureEnoughBuffer(pos + encodedsize+2);
         writeShort(encodedsize);
         int i=0;
         for(i=0;i<strlen;i++){
diff --git a/activemq-core/src/test/java/org/apache/activemq/util/DataByteArrayOutputStreamTest.java b/activemq-core/src/test/java/org/apache/activemq/util/DataByteArrayOutputStreamTest.java
new file mode 100644
index 0000000000..6dafc80702
--- /dev/null
+++ b/activemq-core/src/test/java/org/apache/activemq/util/DataByteArrayOutputStreamTest.java
@@ -0,0 +1,101 @@
+/**
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.activemq.util;
+
+import junit.framework.TestCase;
+
+import java.io.IOException;
+
+public class DataByteArrayOutputStreamTest extends TestCase {
+
+    /**
+     * This test case assumes that an ArrayIndexOutOfBoundsException will be thrown when the buffer fails to resize
+     * @throws IOException
+     */
+    public void testResize() throws IOException {
+        int initSize = 64;
+        DataByteArrayOutputStream out = new DataByteArrayOutputStream();
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeBoolean(true);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeByte(1);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeBytes("test");
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeChar('C');
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeChars("test");
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeDouble(3.1416);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeFloat((float)3.1416);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeInt(12345);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeLong(12345);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeShort(1234);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.writeUTF("test");
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.write(1234);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.write(new byte[10], 5, 5);
+
+        fillOut(out, initSize);
+        // Should resized here
+        out.write(new byte[10]);
+    }
+
+    /**
+     * This method restarts the stream to the init size, and fills it up with data
+     * @param out
+     * @param size
+     * @throws IOException
+     */
+    public void fillOut(DataByteArrayOutputStream out, int size) throws IOException {
+        out.restart(size);
+        out.write(new byte[size]);
+    }
+}