X-Git-Url: https://git.mdrn.pl/pylucene.git/blobdiff_plain/a2e61f0c04805cfcb8706176758d1283c7e3a55c..aaeed5504b982cf3545252ab528713250aa33eed:/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/util/UnsafeByteArrayOutputStreamTest.java?ds=sidebyside

diff --git a/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/util/UnsafeByteArrayOutputStreamTest.java b/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/util/UnsafeByteArrayOutputStreamTest.java
new file mode 100644
index 0000000..a472c44
--- /dev/null
+++ b/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/util/UnsafeByteArrayOutputStreamTest.java
@@ -0,0 +1,211 @@
+package org.apache.lucene.util;
+
+import java.io.IOException;
+
+import org.junit.Test;
+
+import org.apache.lucene.util.LuceneTestCase;
+import org.apache.lucene.util.UnsafeByteArrayOutputStream;
+
+/**
+ * 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.
+ */
+
+public class UnsafeByteArrayOutputStreamTest extends LuceneTestCase {
+
+  @Test
+  public void testSimpleWrite() throws IOException {
+    int length = 100;
+    byte[] buffer = new byte[length];
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream(buffer);
+
+    for (int i = 0; i < 100; i++) {
+      ubaos.write((byte) i);
+    }
+
+    byte[] result = ubaos.toByteArray();
+
+    assertEquals(length, ubaos.length());
+
+    for (int j = 0; j < length; ++j) {
+      assertEquals(result[j], j);
+    }
+  }
+
+  @Test
+  public void testArrayWrite() throws IOException {
+    int length = 100;
+    byte[] buffer = new byte[length];
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream(buffer);
+
+    for (int i = 0; i < 100; i++) {
+      ubaos.write((byte) i);
+    }
+
+    int length2 = 10;
+    byte[] buffer2 = new byte[length2];
+    for (int i = 0; i < length2; i++) {
+      buffer2[i] = (byte) (8 + i);
+    }
+
+    ubaos.write(buffer2);
+
+    byte[] result = ubaos.toByteArray();
+
+    assertEquals(length + length2, ubaos.length());
+
+    for (int j = 0; j < length; ++j) {
+      assertEquals(result[j], j);
+    }
+    for (int j = 0; j < length2; ++j) {
+      assertEquals(result[j + length], buffer2[j]);
+    }
+  }
+
+  @Test
+  public void testArrayWriteStartNotZero() throws IOException {
+    int length = 100;
+    byte[] buffer = new byte[length];
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream(buffer);
+
+    for (int i = 0; i < 100; i++) {
+      ubaos.write((byte) i);
+    }
+
+    int length2 = 1000;
+    byte[] buffer2 = new byte[length2];
+    for (int i = 0; i < length2; i++) {
+      buffer2[i] = (byte) (8 + i);
+    }
+
+    int length3 = 5;
+    int start = 2;
+    ubaos.write(buffer2, start, length3);
+
+    byte[] result = ubaos.toByteArray();
+
+    assertEquals(length + length3, ubaos.length());
+
+    for (int j = 0; j < length; ++j) {
+      assertEquals(result[j], j);
+    }
+    for (int j = 0; j < length3; ++j) {
+      assertEquals(result[j + length], buffer2[j + start]);
+    }
+  }
+
+  @Test
+  public void testBufferGrow() throws IOException {
+    int length = 100;
+    byte[] buffer = new byte[length / 10];
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream(buffer);
+
+    for (int i = 0; i < length; i++) {
+      ubaos.write((byte) i);
+    }
+
+    byte[] result = ubaos.toByteArray();
+
+    assertEquals(length, ubaos.length());
+
+    for (int j = 0; j < length; ++j) {
+      assertEquals(result[j], j);
+    }
+
+    buffer = ubaos.toByteArray();
+
+    int length2 = 10;
+    byte[] buffer2 = new byte[length2];
+    for (int i = 0; i < length2; i++) {
+      buffer2[i] = (byte) (8 + i);
+    }
+
+    ubaos.reInit(buffer2);
+    for (int i = 0; i < length2; i++) {
+      ubaos.write(7 + i);
+    }
+
+    byte[] result2 = ubaos.toByteArray();
+
+    assertEquals(length2, ubaos.length());
+
+    for (int j = 0; j < length2; ++j) {
+      assertEquals(result2[j], j + 7);
+    }
+
+    for (int i = 0; i < length; i++) {
+      assertEquals(buffer[i], i);
+    }
+  }
+  
+  @Test
+  public void testStartPos() throws Exception {
+    byte[] buf = new byte[10];
+    for (int i = 0; i < buf.length; i++) {
+      buf[i] = (byte) i;
+    }
+    
+    int startPos = 3;
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream(buf, startPos);
+    int numValues = 5;
+    for (int i = 0; i < numValues; i++) {
+      ubaos.write((i + 1) * 2);
+    }
+
+    // the length of the buffer should be whatever was written after startPos
+    // and before that.
+    assertEquals("invalid buffer length", startPos + numValues, ubaos.length());
+
+    assertEquals("invalid startPos", startPos, ubaos.getStartPos());
+
+    byte[] bytes = ubaos.toByteArray();
+    for (int i = 0; i < startPos; i++) {
+      assertEquals(i, bytes[i]);
+    }
+    
+    for (int i = startPos, j = 0; j < numValues; i++, j++) {
+      assertEquals((j + 1) * 2, bytes[i]);
+    }
+
+    for (int i = startPos + numValues; i < buf.length; i++) {
+      assertEquals(i, bytes[i]);
+    }
+
+  }
+
+  @Test
+  public void testDefaultCtor() throws Exception {
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream();
+    int numValues = 5;
+    for (int i = 0; i < numValues; i++) {
+      ubaos.write(i);
+    }
+
+    assertEquals("invalid buffer length", numValues, ubaos.length());
+    
+    byte[] bytes = ubaos.toByteArray();
+    for (int i = 0; i < numValues; i++) {
+      assertEquals(i, bytes[i]);
+    }
+  }
+  
+  @Test(expected=IllegalArgumentException.class)
+  public void testIllegalBufferSize() throws Exception {
+    UnsafeByteArrayOutputStream ubaos = new UnsafeByteArrayOutputStream();
+    ubaos.reInit(new byte[0]);
+  }
+
+}