Search in sources :

Example 1 with AuthenticatedUserInfo

use of alluxio.security.authentication.AuthenticatedUserInfo 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)

Aggregations

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