honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

vary offset/length (i hope) in compressed stream test

This commit is contained in:
Robert Muir 2014-08-08 09:54:37 -04:00
parent 34830368d9
commit 1a09e7180a
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/test/java/org/elasticsearch/common/compress/CompressedStreamTests.java
View File

@ -399,10 +399,13 @@ public class CompressedStreamTests extends ElasticsearchTestCase {
Random r = getRandom(); Random r = getRandom();
int bufferSize = r.nextBoolean() ? 65535 : TestUtil.nextInt(getRandom(), 1, 70000); int bufferSize = r.nextBoolean() ? 65535 : TestUtil.nextInt(getRandom(), 1, 70000);
byte buffer[] = new byte[bufferSize]; int prepadding = r.nextInt(70000);
int postpadding = r.nextInt(70000);
byte buffer[] = new byte[prepadding + bufferSize + postpadding];
r.nextBytes(buffer); // fill block completely with junk
int len; int len;
while ((len = rawIn.read(buffer)) != -1) { while ((len = rawIn.read(buffer, prepadding, bufferSize)) != -1) {
os.write(buffer, 0, len); os.write(buffer, prepadding, len);
} }
os.close(); os.close();
rawIn.close(); rawIn.close();
@ -414,9 +417,16 @@ public class CompressedStreamTests extends ElasticsearchTestCase {
StreamInput compressedIn = new ByteBufferStreamInput(bb2); StreamInput compressedIn = new ByteBufferStreamInput(bb2);
StreamInput in = c.streamInput(compressedIn); StreamInput in = c.streamInput(compressedIn);
// randomize constants again
bufferSize = r.nextBoolean() ? 65535 : TestUtil.nextInt(getRandom(), 1, 70000);
prepadding = r.nextInt(70000);
postpadding = r.nextInt(70000);
buffer = new byte[prepadding + bufferSize + postpadding];
r.nextBytes(buffer); // fill block completely with junk
ByteArrayOutputStream uncompressedOut = new ByteArrayOutputStream(); ByteArrayOutputStream uncompressedOut = new ByteArrayOutputStream();
while ((len = in.read(buffer)) != -1) { while ((len = in.read(buffer, prepadding, bufferSize)) != -1) {
uncompressedOut.write(buffer, 0, len); uncompressedOut.write(buffer, prepadding, len);
} }
uncompressedOut.close(); uncompressedOut.close();