Skip to content
Snippets Groups Projects
Commit 448685ca authored by m1ch1's avatar m1ch1
Browse files

gh-69 Added a unit test for RandomByteIterator toArray() and toString() methods.

parent d461c890
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
core/pom.xml
+ 6
− 0
View file @ 448685ca
...@@ -26,6 +26,12 @@ ...@@ -26,6 +26,12 @@
<artifactId>jackson-core-asl</artifactId> <artifactId>jackson-core-asl</artifactId>
<version>${jackson.api.version}</version> <version>${jackson.api.version}</version>
</dependency> </dependency>
<dependency>
<groupId>org.testng</groupId>
<artifactId>testng</artifactId>
<version>6.1.1</version>
<scope>test</scope>
</dependency>
</dependencies> </dependencies>
</project> </project>
core/src/main/java/com/yahoo/ycsb/RandomByteIterator.java
+ 80
− 77
View file @ 448685ca
/** /**
* Copyright (c) 2010 Yahoo! Inc. All rights reserved. * Copyright (c) 2010 Yahoo! Inc. All rights reserved.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); you * Licensed under the Apache License, Version 2.0 (the "License"); you
* may not use this file except in compliance with the License. You * may not use this file except in compliance with the License. You
* may obtain a copy of the License at * may obtain a copy of the License at
* *
* http://www.apache.org/licenses/LICENSE-2.0 * http://www.apache.org/licenses/LICENSE-2.0
* *
* Unless required by applicable law or agreed to in writing, software * Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, * distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied. See the License for the specific language governing * implied. See the License for the specific language governing
* permissions and limitations under the License. See accompanying * permissions and limitations under the License. See accompanying
* LICENSE file. * LICENSE file.
*/ */
package com.yahoo.ycsb; package com.yahoo.ycsb;
/**
* A ByteIterator that generates a random sequence of bytes.
*/
public class RandomByteIterator extends ByteIterator { public class RandomByteIterator extends ByteIterator {
long len; private long len;
long off; private long off;
int buf_off; private int bufOff;
byte[] buf; private byte[] buf;
@Override @Override
public boolean hasNext() { public boolean hasNext() {
return (off + buf_off) < len; return (off + bufOff) < len;
} }
private void fillBytesImpl(byte[] buffer, int base) {
int bytes = Utils.random().nextInt();
try {
buffer[base+0] = (byte)(((bytes) & 31) + ' ');
buffer[base+1] = (byte)(((bytes >> 5) & 31) + ' ');
buffer[base+2] = (byte)(((bytes >> 10) & 31) + ' ');
buffer[base+3] = (byte)(((bytes >> 15) & 31) + ' ');
buffer[base+4] = (byte)(((bytes >> 20) & 31) + ' ');
buffer[base+5] = (byte)(((bytes >> 25) & 31) + ' ');
} catch (ArrayIndexOutOfBoundsException e) { /* ignore it */ }
}
private void fillBytes() {
if(bufOff == buf.length) {
fillBytesImpl(buf, 0);
bufOff = 0;
off += buf.length;
}
}
public RandomByteIterator(long len) {
this.len = len;
this.buf = new byte[6];
this.bufOff = buf.length;
fillBytes();
this.off = 0;
}
public byte nextByte() {
fillBytes();
bufOff++;
return buf[bufOff-1];
}
private void fillBytesImpl(byte[] buf, int base) { @Override
int bytes = Utils.random().nextInt(); public int nextBuf(byte[] buffer, int bufferOffset) {
try { int ret;
buf[base+0] = (byte)(((bytes ) & 31) + ' '); if(len - off < buffer.length - bufferOffset) {
buf[base+1] = (byte)(((bytes >> 5 ) & 31) + ' '); ret = (int)(len - off);
buf[base+2] = (byte)(((bytes >> 10) & 31) + ' '); } else {
buf[base+3] = (byte)(((bytes >> 15) & 31) + ' '); ret = buffer.length - bufferOffset;
buf[base+4] = (byte)(((bytes >> 20) & 31) + ' '); }
buf[base+5] = (byte)(((bytes >> 25) & 31) + ' '); int i;
} catch (ArrayIndexOutOfBoundsException e) { /* ignore it */ } for(i = 0; i < ret; i+=6) {
} fillBytesImpl(buffer, i + bufferOffset);
}
private void fillBytes() { off+=ret;
if(buf_off == buf.length) { return ret + bufferOffset;
fillBytesImpl(buf, 0); }
buf_off = 0;
off += buf.length;
}
}
public RandomByteIterator(long len) {
this.len = len;
this.buf = new byte[6];
this.buf_off = buf.length;
fillBytes();
this.off = 0;
}
public byte nextByte() { @Override
fillBytes(); public long bytesLeft() {
buf_off++; return len - off - bufOff;
return buf[buf_off-1]; }
}
@Override
public int nextBuf(byte[] b, int buf_off) {
int ret;
if(len - off < b.length - buf_off) {
ret = (int)(len - off);
} else {
ret = b.length - buf_off;
}
int i;
for(i = 0; i < ret; i+=6) {
fillBytesImpl(b, i+buf_off);
}
off+=ret;
return ret + buf_off;
}
@Override
public long bytesLeft() {
return len - off;
}
} }
core/src/test/java/com/yahoo/ycsb/TestByteIterator.java 0 → 100644
+ 22
− 0
View file @ 448685ca
package com.yahoo.ycsb;
import org.testng.annotations.Test;
import static org.testng.AssertJUnit.*;
public class TestByteIterator {
@Test
public void testRandomByteIterator() {
int size = 100;
ByteIterator itor = new RandomByteIterator(size);
assertTrue(itor.hasNext());
assertEquals(size, itor.bytesLeft());
assertEquals(size, itor.toString().getBytes().length);
assertFalse(itor.hasNext());
assertEquals(0, itor.bytesLeft());
itor = new RandomByteIterator(size);
assertEquals(size, itor.toArray().length);
assertFalse(itor.hasNext());
assertEquals(0, itor.bytesLeft());
}
}
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment