Example usage for io.netty.channel.embedded EmbeddedChannel EmbeddedChannel

Introduction

In this page you can find the example usage for io.netty.channel.embedded EmbeddedChannel EmbeddedChannel.

Prototype

public EmbeddedChannel(ChannelHandler... handlers)

Source Link

Document

Create a new instance with the pipeline initialized with the specified handlers.

Usage

From source file:alluxio.worker.netty.BlockReadHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    ResourceLeakDetector.setLevel(ResourceLeakDetector.Level.ADVANCED);
    mBlockWorker = mock(BlockWorker.class);
    doNothing().when(mBlockWorker).accessBlock(anyLong(), anyLong());
    mChannel = new EmbeddedChannel(
            new BlockReadHandler(NettyExecutors.BLOCK_READER_EXECUTOR, mBlockWorker, FileTransferType.MAPPED));
    mChannelNoException = new EmbeddedNoExceptionChannel(
            new BlockReadHandler(NettyExecutors.BLOCK_READER_EXECUTOR, mBlockWorker, FileTransferType.MAPPED));
}

From source file:alluxio.worker.netty.BlockReadHandlerTest.java

License:Apache License

/**
 * Tests the {@link FileTransferType#TRANSFER} type.
 *///from   w ww.  j ava 2s .co m
@Test
public void transferType() throws Exception {
    mChannel = new EmbeddedChannel(new BlockReadHandler(NettyExecutors.BLOCK_READER_EXECUTOR, mBlockWorker,
            FileTransferType.TRANSFER));

    long fileSize = PACKET_SIZE * 2;
    long checksumExpected = populateInputFile(fileSize, 0, fileSize - 1);

    BlockReader blockReader = spy(mBlockReader);
    // Do not call close here so that we can check result. It will be closed explicitly.
    doNothing().when(blockReader).close();
    when(mBlockWorker.readBlockRemote(anyLong(), anyLong(), anyLong())).thenReturn(blockReader);
    mChannel.writeInbound(buildReadRequest(0, fileSize));
    checkAllReadResponses(mChannel, checksumExpected);
    mBlockReader.close();
}

From source file:alluxio.worker.netty.BlockWriteHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    mFile = mTestFolder.newFile();//from   w ww  . ja  v  a2  s  .  c  om
    mBlockWorker = Mockito.mock(BlockWorker.class);
    Mockito.doNothing().when(mBlockWorker).createBlockRemote(Mockito.anyLong(), Mockito.anyLong(),
            Mockito.anyString(), Mockito.anyLong());
    Mockito.doNothing().when(mBlockWorker).requestSpace(Mockito.anyLong(), Mockito.anyLong(),
            Mockito.anyLong());
    Mockito.doNothing().when(mBlockWorker).abortBlock(Mockito.anyLong(), Mockito.anyLong());
    Mockito.doNothing().when(mBlockWorker).commitBlock(Mockito.anyLong(), Mockito.anyLong());
    mBlockWriter = new LocalFileBlockWriter(mFile.getPath());
    Mockito.when(mBlockWorker.getTempBlockWriterRemote(Mockito.anyLong(), Mockito.anyLong()))
            .thenReturn(mBlockWriter).thenReturn(new LocalFileBlockWriter(mTestFolder.newFile().getPath()));
    mChannel = new EmbeddedChannel(new BlockWriteHandler(NettyExecutors.BLOCK_WRITER_EXECUTOR, mBlockWorker));
}

From source file:alluxio.worker.netty.DataServerBlockReadHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    ResourceLeakDetector.setLevel(ResourceLeakDetector.Level.ADVANCED);
    mBlockWorker = Mockito.mock(BlockWorker.class);
    Mockito.doNothing().when(mBlockWorker).accessBlock(Mockito.anyLong(), Mockito.anyLong());
    mChannel = new EmbeddedChannel(new DataServerBlockReadHandler(NettyExecutors.BLOCK_READER_EXECUTOR,
            mBlockWorker, FileTransferType.MAPPED));
    mChannelNoException = new EmbeddedNoExceptionChannel(new DataServerBlockReadHandler(
            NettyExecutors.BLOCK_READER_EXECUTOR, mBlockWorker, FileTransferType.MAPPED));
}

From source file:alluxio.worker.netty.DataServerBlockReadHandlerTest.java

License:Apache License

/**
 * Tests the {@link FileTransferType#TRANSFER} type.
 *///from w  ww.j  av  a2 s  . com
@Test
public void transferType() throws Exception {
    mChannel = new EmbeddedChannel(new DataServerBlockReadHandler(NettyExecutors.BLOCK_READER_EXECUTOR,
            mBlockWorker, FileTransferType.TRANSFER));

    long fileSize = PACKET_SIZE * 2;
    long checksumExpected = populateInputFile(fileSize, 0, fileSize - 1);

    BlockReader blockReader = Mockito.spy(mBlockReader);
    // Do not call close here so that we can check result. It will be closed explicitly.
    Mockito.doNothing().when(blockReader).close();
    Mockito.when(mBlockWorker.readBlockRemote(Mockito.anyLong(), Mockito.anyLong(), Mockito.anyLong()))
            .thenReturn(blockReader);
    mChannel.writeInbound(buildReadRequest(0, fileSize));
    checkAllReadResponses(mChannel, checksumExpected);
    mBlockReader.close();
}

From source file:alluxio.worker.netty.DataServerBlockWriteHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    mBlockWorker = Mockito.mock(BlockWorker.class);
    Mockito.doNothing().when(mBlockWorker).createBlockRemote(Mockito.anyLong(), Mockito.anyLong(),
            Mockito.anyString(), Mockito.anyLong());
    Mockito.doNothing().when(mBlockWorker).requestSpace(Mockito.anyLong(), Mockito.anyLong(),
            Mockito.anyLong());/*from  ww w  . ja v  a2 s  .com*/
    mFile = mTestFolder.newFile().getPath();
    mBlockWriter = new LocalFileBlockWriter(mFile);
    Mockito.when(mBlockWorker.getTempBlockWriterRemote(Mockito.anyLong(), Mockito.anyLong()))
            .thenReturn(mBlockWriter);
    mChecksum = 0;

    mChannel = new EmbeddedChannel(
            new DataServerBlockWriteHandler(NettyExecutors.BLOCK_WRITER_EXECUTOR, mBlockWorker));
    mChannelNoException = new EmbeddedNoExceptionChannel(
            new DataServerBlockWriteHandler(NettyExecutors.BLOCK_WRITER_EXECUTOR, mBlockWorker));
}

From source file:alluxio.worker.netty.DataServerUFSFileReadHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    ResourceLeakDetector.setLevel(ResourceLeakDetector.Level.ADVANCED);
    mFileSystemWorker = Mockito.mock(FileSystemWorker.class);
    mChannel = new EmbeddedChannel(
            new DataServerUFSFileReadHandler(NettyExecutors.UFS_BLOCK_READER_EXECUTOR, mFileSystemWorker));
    mChannelNoException = new EmbeddedNoExceptionChannel(
            new DataServerUFSFileReadHandler(NettyExecutors.UFS_BLOCK_READER_EXECUTOR, mFileSystemWorker));
}

From source file:alluxio.worker.netty.DataServerUFSFileWriteHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    mFileSystemWorker = Mockito.mock(FileSystemWorker.class);
    mFile = mTestFolder.newFile().getPath();
    mOutputStream = new FileOutputStream(mFile);
    Mockito.when(mFileSystemWorker.getUfsOutputStream(Mockito.anyLong())).thenReturn(mOutputStream);
    mChecksum = 0;/*from  w w w.  j a  v  a 2 s.c o m*/
    mChannel = new EmbeddedChannel(
            new DataServerUFSFileWriteHandler(NettyExecutors.FILE_WRITER_EXECUTOR, mFileSystemWorker));
    mChannelNoException = new EmbeddedNoExceptionChannel(
            new DataServerUFSFileWriteHandler(NettyExecutors.FILE_WRITER_EXECUTOR, mFileSystemWorker));
}

From source file:alluxio.worker.netty.DataServerUnsupportedMessageHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    mChannel = new EmbeddedChannel(new DataServerUnsupportedMessageHandler());
}

From source file:alluxio.worker.netty.UfsFallbackBlockWriteHandlerTest.java

License:Apache License

@Before
public void before() throws Exception {
    mFile = mTestFolder.newFile();/*  www  .  ja v a2 s.  c o  m*/
    mOutputStream = new FileOutputStream(mFile);
    mBlockStore = new TieredBlockStore();
    mBlockWorker = Mockito.mock(BlockWorker.class);
    Mockito.when(mBlockWorker.getBlockStore()).thenReturn(mBlockStore);
    Mockito.when(mBlockWorker.getWorkerId()).thenReturn(new AtomicReference<>(TEST_WORKER_ID));
    UnderFileSystem mockUfs = Mockito.mock(UnderFileSystem.class);
    UfsManager ufsManager = Mockito.mock(UfsManager.class);
    UfsManager.UfsClient ufsClient = new UfsManager.UfsClient(() -> mockUfs, AlluxioURI.EMPTY_URI);
    Mockito.when(ufsManager.get(Mockito.anyLong())).thenReturn(ufsClient);
    Mockito.when(mockUfs.create(Mockito.anyString(), Mockito.any(CreateOptions.class)))
            .thenReturn(mOutputStream).thenReturn(new FileOutputStream(mFile, true));

    mChannel = new EmbeddedChannel(
            new UfsFallbackBlockWriteHandler(NettyExecutors.FILE_WRITER_EXECUTOR, mBlockWorker, ufsManager));

    // create a partial block in block store first
    mBlockStore.createBlock(TEST_SESSION_ID, TEST_BLOCK_ID, BlockStoreLocation.anyDirInTier("MEM"),
            PACKET_SIZE);
    BlockWriter writer = mBlockStore.getBlockWriter(TEST_SESSION_ID, TEST_BLOCK_ID);
    DataBuffer buffer = newDataBuffer(PARTIAL_WRITTEN);
    mPartialChecksum = getChecksum(buffer);
    writer.append((ByteBuf) buffer.getNettyOutput());
    writer.close();
}