io.grpc.testing.integration.ReconnectTestClient Maven / Gradle / Ivy
/*
* Copyright 2015, Google Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* * Neither the name of Google Inc. nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package io.grpc.testing.integration;
import static org.junit.Assert.assertTrue;
import com.google.protobuf.EmptyProtos.Empty;
import io.grpc.ManagedChannel;
import io.grpc.StatusRuntimeException;
import io.grpc.netty.NegotiationType;
import io.grpc.netty.NettyChannelBuilder;
import io.grpc.okhttp.OkHttpChannelBuilder;
import io.grpc.testing.integration.Messages.ReconnectInfo;
/**
* Verifies the client is reconnecting the server with correct backoffs
*
* See the Test Spec.
*/
public class ReconnectTestClient {
private static final int TEST_TIME_MS = 540 * 1000;
private int serverControlPort = 8080;
private int serverRetryPort = 8081;
private boolean useOkhttp = false;
private ManagedChannel controlChannel;
private ManagedChannel retryChannel;
private ReconnectServiceGrpc.ReconnectServiceBlockingStub controlStub;
private ReconnectServiceGrpc.ReconnectServiceBlockingStub retryStub;
private void parseArgs(String[] args) {
for (String arg : args) {
if (!arg.startsWith("--")) {
System.err.println("All arguments must start with '--': " + arg);
System.exit(1);
}
String[] parts = arg.substring(2).split("=", 2);
String key = parts[0];
String value = parts[1];
if ("server_control_port".equals(key)) {
serverControlPort = Integer.parseInt(value);
} else if ("server_retry_port".equals(key)) {
serverRetryPort = Integer.parseInt(value);
} else if ("use_okhttp".equals(key)) {
useOkhttp = Boolean.parseBoolean(value);
} else {
System.err.println("Unknown argument: " + key);
System.exit(1);
}
}
}
private void runTest() throws Exception {
try {
controlChannel = NettyChannelBuilder.forAddress("127.0.0.1", serverControlPort)
.negotiationType(NegotiationType.PLAINTEXT).build();
controlStub = ReconnectServiceGrpc.newBlockingStub(controlChannel);
if (useOkhttp) {
retryChannel = OkHttpChannelBuilder.forAddress("127.0.0.1", serverRetryPort)
.negotiationType(io.grpc.okhttp.NegotiationType.TLS).build();
} else {
retryChannel = NettyChannelBuilder.forAddress("127.0.0.1", serverRetryPort)
.negotiationType(NegotiationType.TLS).build();
}
retryStub = ReconnectServiceGrpc.newBlockingStub(retryChannel);
controlStub.start(Empty.getDefaultInstance());
long startTimeStamp = System.currentTimeMillis();
while ((System.currentTimeMillis() - startTimeStamp) < TEST_TIME_MS) {
try {
retryStub.start(Empty.getDefaultInstance());
} catch (StatusRuntimeException expected) {
// Make CheckStyle happy.
}
Thread.sleep(50);
}
ReconnectInfo info = controlStub.stop(Empty.getDefaultInstance());
assertTrue(info.getPassed());
} finally {
controlChannel.shutdownNow();
retryChannel.shutdownNow();
}
}
/**
* The main application allowing this client to be launched from the command line.
*/
public static void main(String[] args) {
ReconnectTestClient client = new ReconnectTestClient();
client.parseArgs(args);
System.out.println("Starting test:");
try {
client.runTest();
System.out.println("Finished successfully");
System.exit(0);
} catch (Throwable e) {
e.printStackTrace();
System.err.println("Test failed!");
System.exit(1);
}
}
}