Merge "TemporaryBuffer: Fix reading from in-memory InputStream"

This commit is contained in:
Shawn O. Pearce 2011-07-01 10:55:16 -04:00 committed by Code Review
commit 532f9eba3b
2 changed files with 18 additions and 2 deletions

View File

@ -52,6 +52,7 @@
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import org.eclipse.jgit.junit.TestRng;
@ -248,6 +249,20 @@ public void testLarge_SingleWrite() throws IOException {
}
}
@Test
public void testInCoreInputStream() throws IOException {
final int cnt = 256;
final byte[] test = new TestRng(getName()).nextBytes(cnt);
final TemporaryBuffer.Heap b = new TemporaryBuffer.Heap(cnt + 4);
b.write(test);
b.close();
InputStream in = b.openInputStream();
byte[] act = new byte[cnt];
IO.readFully(in, act, 0, cnt);
assertTrue(Arrays.equals(test, act));
}
@Test
public void testInCoreLimit_SwitchOnAppendByte() throws IOException {
final TemporaryBuffer b = new TemporaryBuffer.LocalFile();

View File

@ -548,7 +548,7 @@ public long skip(long cnt) throws IOException {
long skipped = 0;
while (0 < cnt) {
int n = (int) Math.min(block.count - blockPos, cnt);
if (n < 0) {
if (0 < n) {
blockPos += n;
skipped += n;
cnt -= n;
@ -567,11 +567,12 @@ public int read(byte[] b, int off, int len) throws IOException {
int copied = 0;
while (0 < len) {
int c = Math.min(block.count - blockPos, len);
if (c < 0) {
if (0 < c) {
System.arraycopy(block.buffer, blockPos, b, off, c);
blockPos += c;
off += c;
len -= c;
copied += c;
} else if (nextBlock())
continue;
else