Example usage for io.netty.channel.local LocalChannel LocalChannel

List of usage examples for io.netty.channel.local LocalChannel LocalChannel

Introduction

In this page you can find the example usage for io.netty.channel.local LocalChannel LocalChannel.

Prototype

public LocalChannel() 

Source Link

Usage

From source file:com.schibsted.triathlon.main.TriathlonRouterTest.java

License:Apache License

private HttpServerRequest getRequest(String path, HttpRequestHeaders headers, HttpMethod method) {
    HttpServerRequest request = Mockito.mock(HttpServerRequest.class);
    when(request.getHeaders()).thenReturn(headers);
    when(request.getHttpMethod()).thenReturn(method);
    when(request.getHttpVersion()).thenReturn(HttpVersion.HTTP_1_1);
    when(request.getPath()).thenReturn(path);
    when(request.getQueryString()).thenReturn("");
    when(request.getUri()).thenReturn(path);
    when(request.getQueryParameters()).thenReturn(new HashMap<String, List<String>>());
    when(request.getCookies()).thenReturn(new HashMap<String, Set<String>>());
    when(request.getNettyChannel()).thenReturn(new LocalChannel());
    return request;
}

From source file:com.schibsted.triathlon.main.TriathlonRouterTest.java

License:Apache License

private HttpServerResponse getResponse() {
    HttpServerResponse response = Mockito.mock(HttpServerResponse.class);
    when(response.getChannel()).thenReturn(new LocalChannel());

    when(response.close()).thenReturn(Observable.empty());

    return response;
}

From source file:io.reactivex.netty.contexts.NoOpChannelHandlerContext.java

License:Apache License

public NoOpChannelHandlerContext() {
    channel = new LocalChannel() {
        @Override//from   w w  w . j  av  a  2s  . c  o m
        public <T> Attribute<T> attr(AttributeKey<T> key) {
            return attributeMap.attr(key);
        }
    };
    failedPromise = new DefaultChannelPromise(channel, executor());
    failedPromise.setFailure(new OperationNotSupportedException());
}

From source file:io.reactivex.netty.NoOpChannelHandlerContext.java

License:Apache License

public NoOpChannelHandlerContext() {
    this(new LocalChannel());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void handleSuccessfulFetch() throws Exception {
    StreamChunkId streamChunkId = new StreamChunkId(1, 0);

    TransportResponseHandler handler = new TransportResponseHandler(new LocalChannel());
    ChunkReceivedCallback callback = mock(ChunkReceivedCallback.class);
    handler.addFetchRequest(streamChunkId, callback);
    assertEquals(1, handler.numOutstandingRequests());

    handler.handle(new ChunkFetchSuccess(streamChunkId, new TestManagedBuffer(123)));
    verify(callback, times(1)).onSuccess(eq(0), any());
    assertEquals(0, handler.numOutstandingRequests());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void handleFailedFetch() throws Exception {
    StreamChunkId streamChunkId = new StreamChunkId(1, 0);
    TransportResponseHandler handler = new TransportResponseHandler(new LocalChannel());
    ChunkReceivedCallback callback = mock(ChunkReceivedCallback.class);
    handler.addFetchRequest(streamChunkId, callback);
    assertEquals(1, handler.numOutstandingRequests());

    handler.handle(new ChunkFetchFailure(streamChunkId, "some error msg"));
    verify(callback, times(1)).onFailure(eq(0), any());
    assertEquals(0, handler.numOutstandingRequests());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void clearAllOutstandingRequests() throws Exception {
    TransportResponseHandler handler = new TransportResponseHandler(new LocalChannel());
    ChunkReceivedCallback callback = mock(ChunkReceivedCallback.class);
    handler.addFetchRequest(new StreamChunkId(1, 0), callback);
    handler.addFetchRequest(new StreamChunkId(1, 1), callback);
    handler.addFetchRequest(new StreamChunkId(1, 2), callback);
    assertEquals(3, handler.numOutstandingRequests());

    handler.handle(new ChunkFetchSuccess(new StreamChunkId(1, 0), new TestManagedBuffer(12)));
    handler.exceptionCaught(new Exception("duh duh duhhhh"));

    // should fail both b2 and b3
    verify(callback, times(1)).onSuccess(eq(0), any());
    verify(callback, times(1)).onFailure(eq(1), any());
    verify(callback, times(1)).onFailure(eq(2), any());
    assertEquals(0, handler.numOutstandingRequests());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void handleSuccessfulRPC() throws Exception {
    TransportResponseHandler handler = new TransportResponseHandler(new LocalChannel());
    RpcResponseCallback callback = mock(RpcResponseCallback.class);
    handler.addRpcRequest(12345, callback);
    assertEquals(1, handler.numOutstandingRequests());

    // This response should be ignored.
    handler.handle(new RpcResponse(54321, new NioManagedBuffer(ByteBuffer.allocate(7))));
    assertEquals(1, handler.numOutstandingRequests());

    ByteBuffer resp = ByteBuffer.allocate(10);
    handler.handle(new RpcResponse(12345, new NioManagedBuffer(resp)));
    verify(callback, times(1)).onSuccess(eq(ByteBuffer.allocate(10)));
    assertEquals(0, handler.numOutstandingRequests());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void handleFailedRPC() throws Exception {
    TransportResponseHandler handler = new TransportResponseHandler(new LocalChannel());
    RpcResponseCallback callback = mock(RpcResponseCallback.class);
    handler.addRpcRequest(12345, callback);
    assertEquals(1, handler.numOutstandingRequests());

    handler.handle(new RpcFailure(54321, "uh-oh!")); // should be ignored
    assertEquals(1, handler.numOutstandingRequests());

    handler.handle(new RpcFailure(12345, "oh no"));
    verify(callback, times(1)).onFailure(any());
    assertEquals(0, handler.numOutstandingRequests());
}

From source file:org.apache.spark.network.TransportResponseHandlerSuite.java

License:Apache License

@Test
public void testActiveStreams() throws Exception {
    Channel c = new LocalChannel();
    c.pipeline().addLast(TransportFrameDecoder.HANDLER_NAME, new TransportFrameDecoder());
    TransportResponseHandler handler = new TransportResponseHandler(c);

    StreamResponse response = new StreamResponse("stream", 1234L, null);
    StreamCallback cb = mock(StreamCallback.class);
    handler.addStreamCallback("stream", cb);
    assertEquals(1, handler.numOutstandingRequests());
    handler.handle(response);/*from w  w  w.j  a  va  2 s  . c o m*/
    assertEquals(1, handler.numOutstandingRequests());
    handler.deactivateStream();
    assertEquals(0, handler.numOutstandingRequests());

    StreamFailure failure = new StreamFailure("stream", "uh-oh");
    handler.addStreamCallback("stream", cb);
    assertEquals(1, handler.numOutstandingRequests());
    handler.handle(failure);
    assertEquals(0, handler.numOutstandingRequests());
}