diff --git a/lucene/src/test/org/apache/lucene/store/TestCopyBytes.java b/lucene/src/test/org/apache/lucene/store/TestCopyBytes.java new file mode 100644 index 00000000000..ba014fb8f57 --- /dev/null +++ b/lucene/src/test/org/apache/lucene/store/TestCopyBytes.java @@ -0,0 +1,110 @@ +package org.apache.lucene.store; + +/** + * 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. + */ + + +import java.io.IOException; +import java.util.Random; +import org.apache.lucene.util.LuceneTestCaseJ4; +import org.apache.lucene.util._TestUtil; + +import org.junit.Test; +import static org.junit.Assert.*; + +public class TestCopyBytes extends LuceneTestCaseJ4 { + + private byte value(int idx) { + return (byte) ((idx%256) * (1+(idx/256))); + } + + + @Test + public void testCopyBytes() throws Exception { + Random rand = newRandom(); + for(int iter=0;iter<10*RANDOM_MULTIPLIER;iter++) { + Directory dir = newDirectory(rand); + if (VERBOSE) { + System.out.println("TEST: iter=" + iter + " dir=" + dir); + } + + // make random file + IndexOutput out = dir.createOutput("test"); + byte[] bytes = new byte[_TestUtil.nextInt(rand, 1, 77777)]; + final int size = _TestUtil.nextInt(rand, 1, 1777777); + int upto = 0; + int byteUpto = 0; + while(upto < size) { + bytes[byteUpto++] = value(upto); + upto++; + if (byteUpto == bytes.length) { + out.writeBytes(bytes, 0, bytes.length); + byteUpto = 0; + } + } + + out.writeBytes(bytes, 0, byteUpto); + assertEquals(size, out.getFilePointer()); + out.close(); + assertEquals(size, dir.fileLength("test")); + + // copy from test -> test2 + final IndexInput in = dir.openInput("test"); + + out = dir.createOutput("test2"); + + upto = 0; + while(upto < size) { + if (rand.nextBoolean()) { + out.writeByte(in.readByte()); + upto++; + } else { + final int chunk = Math.min(_TestUtil.nextInt(rand, 1, bytes.length), size-upto); + out.copyBytes(in, chunk); + upto += chunk; + } + } + assertEquals(size, upto); + out.close(); + in.close(); + + // verify + IndexInput in2 = dir.openInput("test2"); + upto = 0; + while(upto < size) { + if (rand.nextBoolean()) { + final byte v = in2.readByte(); + assertEquals(value(upto), v); + upto++; + } else { + final int limit = Math.min(_TestUtil.nextInt(rand, 1, bytes.length), size-upto); + in2.readBytes(bytes, 0, limit); + for(int byteIdx=0;byteIdx