Java DefaultChannelPromise.setFailure方法代码示例

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test(expected = RuntimeException.class)
public void testIdentifyWhenWriteFailsAndChannelIsActiveClosesChannelAndThrows() throws Exception {
    DefaultChannelPromise promise = new DefaultChannelPromise(channel, eventExecutors.next());
    promise.setFailure(new IOException("test"));

    when(channel.writeAndFlush(any())).thenReturn(promise);
    when(channel.isActive()).thenReturn(true);
    when(channel.close()).thenReturn(promise);

    Admin admin = new Admin();
    admin.setBoxId("test");
    admin.setAdminCommand(AdminCommand.IDENTIFY);

    try {
        clientSession.identify(admin);
    } finally {
        verify(channel).writeAndFlush(admin);
        verify(channel).close();

        assertTrue(clientSession.isClosed());
    }
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test(expected = RuntimeException.class)
public void testIdentifyWhenWriteFailsAndChannelIsInactiveSetsClosedState() throws Exception {
    DefaultChannelPromise promise = new DefaultChannelPromise(channel, eventExecutors.next());
    promise.setFailure(new IOException("test"));

    when(channel.writeAndFlush(any())).thenReturn(promise);
    when(channel.isActive()).thenReturn(false);

    Admin admin = new Admin();
    admin.setBoxId("test");
    admin.setAdminCommand(AdminCommand.IDENTIFY);

    try {
        clientSession.identify(admin);
    } finally {
        verify(channel).writeAndFlush(admin);
        verify(channel, times(0)).close();

        assertTrue(clientSession.isClosed());
    }
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test
public void testSendSmsAndWaitThrowsWhenWriteFails() throws Exception {
    DefaultChannelPromise promise = new DefaultChannelPromise(channel, eventExecutors.next());
    promise.setFailure(new IOException());

    when(channel.writeAndFlush(any())).thenReturn(promise);

    Sms sms = new Sms();
    sms.setId(UUID.randomUUID());
    sms.setBoxId("test box");

    try {
        clientSession.sendSmsAndWait(sms, 5000);
    } catch (ExecutionException e) {
        assertTrue(e.getCause() instanceof IOException);
    }
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
private void assertExceptionsFiredOnFailure() throws Exception {
  // get the listener added when the channel was written to
  ArgumentCaptor<ChannelFutureListener> captor = ArgumentCaptor.forClass(ChannelFutureListener.class);
  verify(writeFuture, atLeast(1)).addListener(captor.capture());
  ChannelFutureListener addedListener = captor.getValue();

  // tell the listener the write failed
  DefaultChannelPromise promise = new DefaultChannelPromise(channel, ImmediateEventExecutor.INSTANCE);
  promise.setFailure(new Exception());
  addedListener.operationComplete(promise);

  verify(pipeline).fireExceptionCaught(promise.cause());
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test(expected = IOException.class)
public void testSendSmsReturnsFailedFutureWhenWriteFails() throws Exception {
    DefaultChannelPromise promise = new DefaultChannelPromise(channel, eventExecutors.next());
    promise.setFailure(new IOException());

    when(channel.writeAndFlush(any())).thenReturn(promise);

    Sms sms = new Sms();
    sms.setId(UUID.randomUUID());
    sms.setBoxId("test box");

    WindowFuture<Sms, Ack> future = clientSession.sendSms(sms, 5000);

    Futures.getChecked(future, IOException.class);
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test
public void testEncodeMethodCallFailure() throws InvalidProtocolBufferException,
    InterruptedException {
  Channel channel = Mockito.mock(Channel.class);
  ArgumentCaptor<Object> captor = ArgumentCaptor.forClass(Object.class);

  DefaultChannelPromise failure = new DefaultChannelPromise(
      channel, ImmediateEventExecutor.INSTANCE);
  failure.setFailure(new Exception("OMGWTF"));
  Mockito.when(channel.writeAndFlush(captor.capture())).thenReturn(failure);

  RpcClientHandler handler = new RpcClientHandler();
  RpcClient client = new RpcClient(channel, handler, new NullClientLogger());

  ClientMethod<TimeResponse> method = Mockito.mock(ClientMethod.class);
  Mockito.when(method.serviceName()).thenReturn("TimeService");
  Mockito.when(method.name()).thenReturn("GetTime");
  Mockito.when(method.outputParser()).thenReturn(TimeResponse.PARSER);

  final CountDownLatch latch = new CountDownLatch(1);
  FutureCallback<TimeResponse> callback = new FutureCallback<TimeResponse>() {
    @Override
    public void onSuccess(@Nullable TimeResponse result) {
    }

    @Override
    public void onFailure(Throwable t) {
      Assert.assertEquals("OMGWTF", t.getMessage());
      latch.countDown();
    }
  };

  ListenableFuture<TimeResponse> future = client.encodeMethodCall(
      method, TimeRequest.newBuilder().setTimezone("UTC").build());

  Futures.addCallback(future, callback);
  latch.await(5, TimeUnit.SECONDS);

  Assert.assertEquals(0, handler.inFlightRequests().size());
}
 

import io.netty.channel.DefaultChannelPromise; //导入方法依赖的package包/类
@Test
public void testEncodeMethodCallFailure() throws InvalidProtocolBufferException,
    InterruptedException {
  Channel channel = Mockito.mock(Channel.class);

  Mockito.when(channel.remoteAddress()).thenReturn(
      new InetSocketAddress(InetAddress.getLoopbackAddress(), 10000));

  ArgumentCaptor<Object> captor = ArgumentCaptor.forClass(Object.class);

  DefaultChannelPromise failure = new DefaultChannelPromise(
      channel, ImmediateEventExecutor.INSTANCE);
  failure.setFailure(new Exception("OMGWTF"));
  Mockito.when(channel.writeAndFlush(captor.capture())).thenReturn(failure);

  JsonRpcClientHandler handler = new JsonRpcClientHandler();
  HttpJsonRpcClient client = new HttpJsonRpcClient(channel, handler, "/rpc",
      new NullClientLogger());

  ClientMethod<TimeResponse> method = Mockito.mock(ClientMethod.class);
  Mockito.when(method.serviceName()).thenReturn("TimeService");
  Mockito.when(method.name()).thenReturn("GetTime");
  Mockito.when(method.outputParser()).thenReturn(TimeResponse.PARSER);

  final CountDownLatch latch = new CountDownLatch(1);
  FutureCallback<TimeResponse> callback = new FutureCallback<TimeResponse>() {
    @Override
    public void onSuccess(@Nullable TimeResponse result) {
    }

    @Override
    public void onFailure(Throwable t) {
      Assert.assertEquals("OMGWTF", t.getMessage());
      latch.countDown();
    }
  };

  ListenableFuture<TimeResponse> future = client.encodeMethodCall(
      method, TimeRequest.newBuilder().setTimezone("UTC").build());

  Futures.addCallback(future, callback);
  latch.await(5, TimeUnit.SECONDS);

  Assert.assertEquals(0, handler.inFlightRequests().size());
}