@Test(timeout = 120000)
@SetupRpcConnection(setupRpcServer = false, setupRpcClient = false)
public void testUnresolvedAddress() throws Exception {
InetSocketAddress address = new InetSocketAddress("test", 0);
boolean expected = false;
BlockingRpcClient client = null;
try {
RpcConnectionKey rpcConnectionKey = new RpcConnectionKey(address, DummyProtocol.class, true);
Properties connParams = new Properties();
connParams.setProperty(RpcConstants.CLIENT_RETRY_NUM, retries + "");
client =
new BlockingRpcClient(
NettyUtils.getDefaultEventLoopGroup(), rpcConnectionKey, connParams);
client.connect();
fail();
} catch (ConnectException e) {
expected = true;
} catch (Throwable throwable) {
fail(throwable.getMessage());
} finally {
client.close();
}
assertTrue(expected);
}
@Test(timeout = 120000)
@SetupRpcConnection(setupRpcClient = false)
public void testUnresolvedAddress2() throws Exception {
String hostAndPort = RpcUtils.normalizeInetSocketAddress(server.getListenAddress());
RpcConnectionKey rpcConnectionKey =
new RpcConnectionKey(RpcUtils.createUnresolved(hostAndPort), DummyProtocol.class, false);
Properties connParams = new Properties();
connParams.setProperty(RpcConstants.CLIENT_RETRY_NUM, retries + "");
BlockingRpcClient client =
new BlockingRpcClient(NettyUtils.getDefaultEventLoopGroup(), rpcConnectionKey, connParams);
client.connect();
assertTrue(client.isConnected());
try {
BlockingInterface stub = client.getStub();
EchoMessage message = EchoMessage.newBuilder().setMessage(MESSAGE).build();
EchoMessage response2 = stub.echo(null, message);
assertEquals(MESSAGE, response2.getMessage());
} finally {
client.close();
}
}
@Test(timeout = 60000)
@SetupRpcConnection(setupRpcServer = false, setupRpcClient = false)
public void testClientRetryFailureOnStartup() throws Exception {
retries = 2;
ServerSocket serverSocket = new ServerSocket(0);
final InetSocketAddress address =
new InetSocketAddress("127.0.0.1", serverSocket.getLocalPort());
serverSocket.close();
EchoMessage message = EchoMessage.newBuilder().setMessage(MESSAGE).build();
RpcConnectionKey rpcConnectionKey = new RpcConnectionKey(address, DummyProtocol.class, false);
Properties connParams = new Properties();
connParams.setProperty(RpcConstants.CLIENT_RETRY_NUM, retries + "");
BlockingRpcClient client =
new BlockingRpcClient(NettyUtils.getDefaultEventLoopGroup(), rpcConnectionKey, connParams);
try {
client.connect();
fail();
} catch (ConnectTimeoutException e) {
assertFalse(e.getMessage(), client.isConnected());
}
BlockingInterface stub = client.getStub();
try {
EchoMessage response = stub.echo(null, message);
fail();
} catch (TajoServiceException e) {
assertFalse(e.getMessage(), client.isConnected());
}
RpcClientManager.cleanup(client);
}