Examples with RefCnt org.apache.hadoop.hbase.nio.RefCnt used on opensource projects

Example 1 with RefCnt

use of org.apache.hadoop.hbase.nio.RefCnt in project hbase by apache.

the class TestFileIOEngine method testReadFailedShouldReleaseByteBuff.

@Test
public void testReadFailedShouldReleaseByteBuff() {
    ByteBuffAllocator alloc = Mockito.mock(ByteBuffAllocator.class);
    final RefCnt refCnt = RefCnt.create();
    Mockito.when(alloc.allocate(Mockito.anyInt())).thenAnswer(new Answer<ByteBuff>() {

        @Override
        public ByteBuff answer(InvocationOnMock invocation) throws Throwable {
            int len = invocation.getArgument(0);
            return ByteBuff.wrap(new ByteBuffer[] { ByteBuffer.allocate(len + 1) }, refCnt);
        }
    });
    int len = 10;
    byte[] data1 = new byte[len];
    assertEquals(1, refCnt.refCnt());
    try {
        fileIOEngine.write(ByteBuffer.wrap(data1), 0);
        BucketEntry be = createBucketEntry(0, len, alloc);
        fileIOEngine.read(be);
        fail();
    } catch (IOException ioe) {
    // expected exception.
    }
    assertEquals(0, refCnt.refCnt());
}