Examples with TestSocket io.grpc.protobuf.services.ChannelzTestHelper.TestSocket used on opensource projects

Example 1 with TestSocket

use of io.grpc.protobuf.services.ChannelzTestHelper.TestSocket in project grpc-java by grpc.

the class ChannelzProtoUtilTest method toGetServerSocketsResponse.

@Test
public void toGetServerSocketsResponse() {
    // empty results
    assertEquals(GetServerSocketsResponse.getDefaultInstance(), ChannelzProtoUtil.toGetServerSocketsResponse(new ServerSocketsList(Collections.<InternalWithLogId>emptyList(), false)));
    // 1 result, paginated
    assertEquals(GetServerSocketsResponse.newBuilder().addSocketRef(socketRef).build(), ChannelzProtoUtil.toGetServerSocketsResponse(new ServerSocketsList(ImmutableList.<InternalWithLogId>of(socket), false)));
    // 1 result, end
    assertEquals(GetServerSocketsResponse.newBuilder().addSocketRef(socketRef).setEnd(true).build(), ChannelzProtoUtil.toGetServerSocketsResponse(new ServerSocketsList(ImmutableList.<InternalWithLogId>of(socket), true)));
    TestSocket socket2 = new TestSocket();
    // 2 results, end
    assertEquals(GetServerSocketsResponse.newBuilder().addSocketRef(socketRef).addSocketRef(ChannelzProtoUtil.toSocketRef(socket2)).setEnd(true).build(), ChannelzProtoUtil.toGetServerSocketsResponse(new ServerSocketsList(ImmutableList.<InternalWithLogId>of(socket, socket2), true)));
}

Example 2 with TestSocket

use of io.grpc.protobuf.services.ChannelzTestHelper.TestSocket in project grpc-java by grpc.

the class ChannelzServiceTest method getSocket.

@Test
public void getSocket() throws Exception {
    TestSocket socket = new TestSocket();
    assertSocketNotFound(socket.getLogId().getId());
    channelz.addClientSocket(socket);
    assertEquals(GetSocketResponse.newBuilder().setSocket(ChannelzProtoUtil.toSocket(socket)).build(), getSocketHelper(socket.getLogId().getId()));
    channelz.removeClientSocket(socket);
    assertSocketNotFound(socket.getLogId().getId());
}

Example 3 with TestSocket

use of io.grpc.protobuf.services.ChannelzTestHelper.TestSocket in project grpc-java by grpc.

the class ChannelzProtoUtilTest method toSubchannel_socketChildren.

@Test
public void toSubchannel_socketChildren() throws Exception {
    subchannel.stats = toBuilder(subchannel.stats).setSockets(ImmutableList.<InternalWithLogId>of(socket)).build();
    assertEquals(subchannelProto.toBuilder().addSocketRef(socketRef).build(), ChannelzProtoUtil.toSubchannel(subchannel));
    TestSocket otherSocket = new TestSocket();
    subchannel.stats = toBuilder(subchannel.stats).setSockets(ImmutableList.<InternalWithLogId>of(socket, otherSocket)).build();
    assertEquals(subchannelProto.toBuilder().addSocketRef(socketRef).addSocketRef(ChannelzProtoUtil.toSocketRef(otherSocket)).build(), ChannelzProtoUtil.toSubchannel(subchannel));
}