Search in sources :

Example 1 with NoopBlockWorker

use of alluxio.worker.block.NoopBlockWorker in project alluxio by Alluxio.

the class BlockReadHandlerTest method before.

@Before
public void before() throws Exception {
    ResourceLeakDetector.setLevel(ResourceLeakDetector.Level.ADVANCED);
    mFile = mTestFolder.newFile();
    mBlockReader = new LocalFileBlockReader(mFile.getPath());
    mBlockWorker = new NoopBlockWorker() {

        @Override
        public BlockReader createBlockReader(BlockReadRequest request) throws IOException {
            ((FileChannel) mBlockReader.getChannel()).position(request.getStart());
            return mBlockReader;
        }
    };
    mResponseObserver = Mockito.mock(ServerCallStreamObserver.class);
    Mockito.when(mResponseObserver.isReady()).thenReturn(true);
    doAnswer(args -> {
        mResponseCompleted = true;
        return null;
    }).when(mResponseObserver).onCompleted();
    doAnswer(args -> {
        mResponseCompleted = true;
        mError = args.getArgument(0, Throwable.class);
        return null;
    }).when(mResponseObserver).onError(any(Throwable.class));
    doAnswer((args) -> {
        // make a copy of response data before it is released
        mResponses.add(ReadResponse.parseFrom(args.getArgument(0, ReadResponse.class).toByteString()));
        return null;
    }).when(mResponseObserver).onNext(any(ReadResponse.class));
    mReadHandler = new BlockReadHandler(GrpcExecutors.BLOCK_READER_EXECUTOR, mBlockWorker, mResponseObserver, new AuthenticatedUserInfo(), false);
}
Also used : LocalFileBlockReader(alluxio.worker.block.io.LocalFileBlockReader) ServerCallStreamObserver(io.grpc.stub.ServerCallStreamObserver) BlockReadRequest(alluxio.wire.BlockReadRequest) ReadResponse(alluxio.grpc.ReadResponse) LocalFileBlockReader(alluxio.worker.block.io.LocalFileBlockReader) BlockReader(alluxio.worker.block.io.BlockReader) AuthenticatedUserInfo(alluxio.security.authentication.AuthenticatedUserInfo) NoopBlockWorker(alluxio.worker.block.NoopBlockWorker) IOException(java.io.IOException) Before(org.junit.Before)

Example 2 with NoopBlockWorker

use of alluxio.worker.block.NoopBlockWorker in project alluxio by Alluxio.

the class BlockWriteHandlerTest method before.

@Before
public void before() throws Exception {
    mFile = mTestFolder.newFile();
    mBlockWorker = new NoopBlockWorker() {

        @Override
        public BlockWriter createBlockWriter(long sessionId, long blockId) {
            return mBlockWriter;
        }
    };
    mBlockWriter = new LocalFileBlockWriter(mFile.getPath());
    mResponseObserver = Mockito.mock(StreamObserver.class);
    mWriteHandler = new BlockWriteHandler(mBlockWorker, mResponseObserver, mUserInfo, false);
    setupResponseTrigger();
}
Also used : StreamObserver(io.grpc.stub.StreamObserver) LocalFileBlockWriter(alluxio.worker.block.io.LocalFileBlockWriter) NoopBlockWorker(alluxio.worker.block.NoopBlockWorker) BlockWriter(alluxio.worker.block.io.BlockWriter) LocalFileBlockWriter(alluxio.worker.block.io.LocalFileBlockWriter) Before(org.junit.Before)

Aggregations

NoopBlockWorker (alluxio.worker.block.NoopBlockWorker)2 Before (org.junit.Before)2 ReadResponse (alluxio.grpc.ReadResponse)1 AuthenticatedUserInfo (alluxio.security.authentication.AuthenticatedUserInfo)1 BlockReadRequest (alluxio.wire.BlockReadRequest)1 BlockReader (alluxio.worker.block.io.BlockReader)1 BlockWriter (alluxio.worker.block.io.BlockWriter)1 LocalFileBlockReader (alluxio.worker.block.io.LocalFileBlockReader)1 LocalFileBlockWriter (alluxio.worker.block.io.LocalFileBlockWriter)1 ServerCallStreamObserver (io.grpc.stub.ServerCallStreamObserver)1 StreamObserver (io.grpc.stub.StreamObserver)1 IOException (java.io.IOException)1