HttpTunnelClientSendHandlerTest.java 文件源码

@Test
public void testDisconnect() {
    emulateConnectAndOpen();

    channel.write(NettyTestUtils.createData(1234L));
    assertEquals(1, sink.events.size());
    checkIsSendDataRequestWithData(sink.events.poll(),
            NettyTestUtils.createData(1234L));

    channel.disconnect();
    Channels.fireMessageReceived(channel,
            HttpTunnelMessageUtils.createSendDataResponse());
    assertEquals(1, sink.events.size());

    HttpRequest request = NettyTestUtils.checkIsDownstreamMessageEvent(
            sink.events.poll(), HttpRequest.class);
    assertTrue(HttpTunnelMessageUtils.isCloseTunnelRequest(request, USER_AGENT));
    assertEquals("newTunnel",
            HttpTunnelMessageUtils.extractTunnelId(request));
    Channels.fireMessageReceived(channel,
            HttpTunnelMessageUtils.createTunnelCloseResponse());
    assertEquals(1, sink.events.size());
    NettyTestUtils.checkIsStateEvent(sink.events.poll(),
            ChannelState.CONNECTED, null);
}