mirror of https://github.com/grpc/grpc-java.git
interop-testing: add echo-server for proxyless gRPC testing in Istio (#9261)
This commit is contained in:
sanjaypujare
GitHub
parent
7bd0797496
commit
c2d33f15be
|
|
@ -0,0 +1,350 @@
|
|||
package io.grpc.testing.istio;
|
||||
|
||||
import static io.grpc.MethodDescriptor.generateFullMethodName;
|
||||
|
||||
/**
|
||||
*/
|
||||
@javax.annotation.Generated(
|
||||
value = "by gRPC proto compiler",
|
||||
comments = "Source: istio/testing/echo.proto")
|
||||
@io.grpc.stub.annotations.GrpcGenerated
|
||||
public final class EchoTestServiceGrpc {
|
||||
|
||||
private EchoTestServiceGrpc() {}
|
||||
|
||||
public static final String SERVICE_NAME = "istio.testing.EchoTestService";
|
||||
|
||||
// Static method descriptors that strictly reflect the proto.
|
||||
private static volatile io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.EchoRequest,
|
||||
io.grpc.testing.istio.Istio.EchoResponse> getEchoMethod;
|
||||
|
||||
@io.grpc.stub.annotations.RpcMethod(
|
||||
fullMethodName = SERVICE_NAME + '/' + "Echo",
|
||||
requestType = io.grpc.testing.istio.Istio.EchoRequest.class,
|
||||
responseType = io.grpc.testing.istio.Istio.EchoResponse.class,
|
||||
methodType = io.grpc.MethodDescriptor.MethodType.UNARY)
|
||||
public static io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.EchoRequest,
|
||||
io.grpc.testing.istio.Istio.EchoResponse> getEchoMethod() {
|
||||
io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.EchoRequest, io.grpc.testing.istio.Istio.EchoResponse> getEchoMethod;
|
||||
if ((getEchoMethod = EchoTestServiceGrpc.getEchoMethod) == null) {
|
||||
synchronized (EchoTestServiceGrpc.class) {
|
||||
if ((getEchoMethod = EchoTestServiceGrpc.getEchoMethod) == null) {
|
||||
EchoTestServiceGrpc.getEchoMethod = getEchoMethod =
|
||||
io.grpc.MethodDescriptor.<io.grpc.testing.istio.Istio.EchoRequest, io.grpc.testing.istio.Istio.EchoResponse>newBuilder()
|
||||
.setType(io.grpc.MethodDescriptor.MethodType.UNARY)
|
||||
.setFullMethodName(generateFullMethodName(SERVICE_NAME, "Echo"))
|
||||
.setSampledToLocalTracing(true)
|
||||
.setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
|
||||
io.grpc.testing.istio.Istio.EchoRequest.getDefaultInstance()))
|
||||
.setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
|
||||
io.grpc.testing.istio.Istio.EchoResponse.getDefaultInstance()))
|
||||
.setSchemaDescriptor(new EchoTestServiceMethodDescriptorSupplier("Echo"))
|
||||
.build();
|
||||
}
|
||||
}
|
||||
}
|
||||
return getEchoMethod;
|
||||
}
|
||||
|
||||
private static volatile io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.ForwardEchoRequest,
|
||||
io.grpc.testing.istio.Istio.ForwardEchoResponse> getForwardEchoMethod;
|
||||
|
||||
@io.grpc.stub.annotations.RpcMethod(
|
||||
fullMethodName = SERVICE_NAME + '/' + "ForwardEcho",
|
||||
requestType = io.grpc.testing.istio.Istio.ForwardEchoRequest.class,
|
||||
responseType = io.grpc.testing.istio.Istio.ForwardEchoResponse.class,
|
||||
methodType = io.grpc.MethodDescriptor.MethodType.UNARY)
|
||||
public static io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.ForwardEchoRequest,
|
||||
io.grpc.testing.istio.Istio.ForwardEchoResponse> getForwardEchoMethod() {
|
||||
io.grpc.MethodDescriptor<io.grpc.testing.istio.Istio.ForwardEchoRequest, io.grpc.testing.istio.Istio.ForwardEchoResponse> getForwardEchoMethod;
|
||||
if ((getForwardEchoMethod = EchoTestServiceGrpc.getForwardEchoMethod) == null) {
|
||||
synchronized (EchoTestServiceGrpc.class) {
|
||||
if ((getForwardEchoMethod = EchoTestServiceGrpc.getForwardEchoMethod) == null) {
|
||||
EchoTestServiceGrpc.getForwardEchoMethod = getForwardEchoMethod =
|
||||
io.grpc.MethodDescriptor.<io.grpc.testing.istio.Istio.ForwardEchoRequest, io.grpc.testing.istio.Istio.ForwardEchoResponse>newBuilder()
|
||||
.setType(io.grpc.MethodDescriptor.MethodType.UNARY)
|
||||
.setFullMethodName(generateFullMethodName(SERVICE_NAME, "ForwardEcho"))
|
||||
.setSampledToLocalTracing(true)
|
||||
.setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
|
||||
io.grpc.testing.istio.Istio.ForwardEchoRequest.getDefaultInstance()))
|
||||
.setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
|
||||
io.grpc.testing.istio.Istio.ForwardEchoResponse.getDefaultInstance()))
|
||||
.setSchemaDescriptor(new EchoTestServiceMethodDescriptorSupplier("ForwardEcho"))
|
||||
.build();
|
||||
}
|
||||
}
|
||||
}
|
||||
return getForwardEchoMethod;
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new async stub that supports all call types for the service
|
||||
*/
|
||||
public static EchoTestServiceStub newStub(io.grpc.Channel channel) {
|
||||
io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceStub> factory =
|
||||
new io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceStub>() {
|
||||
@java.lang.Override
|
||||
public EchoTestServiceStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceStub(channel, callOptions);
|
||||
}
|
||||
};
|
||||
return EchoTestServiceStub.newStub(factory, channel);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new blocking-style stub that supports unary and streaming output calls on the service
|
||||
*/
|
||||
public static EchoTestServiceBlockingStub newBlockingStub(
|
||||
io.grpc.Channel channel) {
|
||||
io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceBlockingStub> factory =
|
||||
new io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceBlockingStub>() {
|
||||
@java.lang.Override
|
||||
public EchoTestServiceBlockingStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceBlockingStub(channel, callOptions);
|
||||
}
|
||||
};
|
||||
return EchoTestServiceBlockingStub.newStub(factory, channel);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new ListenableFuture-style stub that supports unary calls on the service
|
||||
*/
|
||||
public static EchoTestServiceFutureStub newFutureStub(
|
||||
io.grpc.Channel channel) {
|
||||
io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceFutureStub> factory =
|
||||
new io.grpc.stub.AbstractStub.StubFactory<EchoTestServiceFutureStub>() {
|
||||
@java.lang.Override
|
||||
public EchoTestServiceFutureStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceFutureStub(channel, callOptions);
|
||||
}
|
||||
};
|
||||
return EchoTestServiceFutureStub.newStub(factory, channel);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public static abstract class EchoTestServiceImplBase implements io.grpc.BindableService {
|
||||
|
||||
/**
|
||||
*/
|
||||
public void echo(io.grpc.testing.istio.Istio.EchoRequest request,
|
||||
io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.EchoResponse> responseObserver) {
|
||||
io.grpc.stub.ServerCalls.asyncUnimplementedUnaryCall(getEchoMethod(), responseObserver);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public void forwardEcho(io.grpc.testing.istio.Istio.ForwardEchoRequest request,
|
||||
io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.ForwardEchoResponse> responseObserver) {
|
||||
io.grpc.stub.ServerCalls.asyncUnimplementedUnaryCall(getForwardEchoMethod(), responseObserver);
|
||||
}
|
||||
|
||||
@java.lang.Override public final io.grpc.ServerServiceDefinition bindService() {
|
||||
return io.grpc.ServerServiceDefinition.builder(getServiceDescriptor())
|
||||
.addMethod(
|
||||
getEchoMethod(),
|
||||
io.grpc.stub.ServerCalls.asyncUnaryCall(
|
||||
new MethodHandlers<
|
||||
io.grpc.testing.istio.Istio.EchoRequest,
|
||||
io.grpc.testing.istio.Istio.EchoResponse>(
|
||||
this, METHODID_ECHO)))
|
||||
.addMethod(
|
||||
getForwardEchoMethod(),
|
||||
io.grpc.stub.ServerCalls.asyncUnaryCall(
|
||||
new MethodHandlers<
|
||||
io.grpc.testing.istio.Istio.ForwardEchoRequest,
|
||||
io.grpc.testing.istio.Istio.ForwardEchoResponse>(
|
||||
this, METHODID_FORWARD_ECHO)))
|
||||
.build();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public static final class EchoTestServiceStub extends io.grpc.stub.AbstractAsyncStub<EchoTestServiceStub> {
|
||||
private EchoTestServiceStub(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
super(channel, callOptions);
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
protected EchoTestServiceStub build(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceStub(channel, callOptions);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public void echo(io.grpc.testing.istio.Istio.EchoRequest request,
|
||||
io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.EchoResponse> responseObserver) {
|
||||
io.grpc.stub.ClientCalls.asyncUnaryCall(
|
||||
getChannel().newCall(getEchoMethod(), getCallOptions()), request, responseObserver);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public void forwardEcho(io.grpc.testing.istio.Istio.ForwardEchoRequest request,
|
||||
io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.ForwardEchoResponse> responseObserver) {
|
||||
io.grpc.stub.ClientCalls.asyncUnaryCall(
|
||||
getChannel().newCall(getForwardEchoMethod(), getCallOptions()), request, responseObserver);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public static final class EchoTestServiceBlockingStub extends io.grpc.stub.AbstractBlockingStub<EchoTestServiceBlockingStub> {
|
||||
private EchoTestServiceBlockingStub(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
super(channel, callOptions);
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
protected EchoTestServiceBlockingStub build(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceBlockingStub(channel, callOptions);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public io.grpc.testing.istio.Istio.EchoResponse echo(io.grpc.testing.istio.Istio.EchoRequest request) {
|
||||
return io.grpc.stub.ClientCalls.blockingUnaryCall(
|
||||
getChannel(), getEchoMethod(), getCallOptions(), request);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public io.grpc.testing.istio.Istio.ForwardEchoResponse forwardEcho(io.grpc.testing.istio.Istio.ForwardEchoRequest request) {
|
||||
return io.grpc.stub.ClientCalls.blockingUnaryCall(
|
||||
getChannel(), getForwardEchoMethod(), getCallOptions(), request);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public static final class EchoTestServiceFutureStub extends io.grpc.stub.AbstractFutureStub<EchoTestServiceFutureStub> {
|
||||
private EchoTestServiceFutureStub(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
super(channel, callOptions);
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
protected EchoTestServiceFutureStub build(
|
||||
io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
|
||||
return new EchoTestServiceFutureStub(channel, callOptions);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public com.google.common.util.concurrent.ListenableFuture<io.grpc.testing.istio.Istio.EchoResponse> echo(
|
||||
io.grpc.testing.istio.Istio.EchoRequest request) {
|
||||
return io.grpc.stub.ClientCalls.futureUnaryCall(
|
||||
getChannel().newCall(getEchoMethod(), getCallOptions()), request);
|
||||
}
|
||||
|
||||
/**
|
||||
*/
|
||||
public com.google.common.util.concurrent.ListenableFuture<io.grpc.testing.istio.Istio.ForwardEchoResponse> forwardEcho(
|
||||
io.grpc.testing.istio.Istio.ForwardEchoRequest request) {
|
||||
return io.grpc.stub.ClientCalls.futureUnaryCall(
|
||||
getChannel().newCall(getForwardEchoMethod(), getCallOptions()), request);
|
||||
}
|
||||
}
|
||||
|
||||
private static final int METHODID_ECHO = 0;
|
||||
private static final int METHODID_FORWARD_ECHO = 1;
|
||||
|
||||
private static final class MethodHandlers<Req, Resp> implements
|
||||
io.grpc.stub.ServerCalls.UnaryMethod<Req, Resp>,
|
||||
io.grpc.stub.ServerCalls.ServerStreamingMethod<Req, Resp>,
|
||||
io.grpc.stub.ServerCalls.ClientStreamingMethod<Req, Resp>,
|
||||
io.grpc.stub.ServerCalls.BidiStreamingMethod<Req, Resp> {
|
||||
private final EchoTestServiceImplBase serviceImpl;
|
||||
private final int methodId;
|
||||
|
||||
MethodHandlers(EchoTestServiceImplBase serviceImpl, int methodId) {
|
||||
this.serviceImpl = serviceImpl;
|
||||
this.methodId = methodId;
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
@java.lang.SuppressWarnings("unchecked")
|
||||
public void invoke(Req request, io.grpc.stub.StreamObserver<Resp> responseObserver) {
|
||||
switch (methodId) {
|
||||
case METHODID_ECHO:
|
||||
serviceImpl.echo((io.grpc.testing.istio.Istio.EchoRequest) request,
|
||||
(io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.EchoResponse>) responseObserver);
|
||||
break;
|
||||
case METHODID_FORWARD_ECHO:
|
||||
serviceImpl.forwardEcho((io.grpc.testing.istio.Istio.ForwardEchoRequest) request,
|
||||
(io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.ForwardEchoResponse>) responseObserver);
|
||||
break;
|
||||
default:
|
||||
throw new AssertionError();
|
||||
}
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
@java.lang.SuppressWarnings("unchecked")
|
||||
public io.grpc.stub.StreamObserver<Req> invoke(
|
||||
io.grpc.stub.StreamObserver<Resp> responseObserver) {
|
||||
switch (methodId) {
|
||||
default:
|
||||
throw new AssertionError();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static abstract class EchoTestServiceBaseDescriptorSupplier
|
||||
implements io.grpc.protobuf.ProtoFileDescriptorSupplier, io.grpc.protobuf.ProtoServiceDescriptorSupplier {
|
||||
EchoTestServiceBaseDescriptorSupplier() {}
|
||||
|
||||
@java.lang.Override
|
||||
public com.google.protobuf.Descriptors.FileDescriptor getFileDescriptor() {
|
||||
return io.grpc.testing.istio.Istio.getDescriptor();
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
public com.google.protobuf.Descriptors.ServiceDescriptor getServiceDescriptor() {
|
||||
return getFileDescriptor().findServiceByName("EchoTestService");
|
||||
}
|
||||
}
|
||||
|
||||
private static final class EchoTestServiceFileDescriptorSupplier
|
||||
extends EchoTestServiceBaseDescriptorSupplier {
|
||||
EchoTestServiceFileDescriptorSupplier() {}
|
||||
}
|
||||
|
||||
private static final class EchoTestServiceMethodDescriptorSupplier
|
||||
extends EchoTestServiceBaseDescriptorSupplier
|
||||
implements io.grpc.protobuf.ProtoMethodDescriptorSupplier {
|
||||
private final String methodName;
|
||||
|
||||
EchoTestServiceMethodDescriptorSupplier(String methodName) {
|
||||
this.methodName = methodName;
|
||||
}
|
||||
|
||||
@java.lang.Override
|
||||
public com.google.protobuf.Descriptors.MethodDescriptor getMethodDescriptor() {
|
||||
return getServiceDescriptor().findMethodByName(methodName);
|
||||
}
|
||||
}
|
||||
|
||||
private static volatile io.grpc.ServiceDescriptor serviceDescriptor;
|
||||
|
||||
public static io.grpc.ServiceDescriptor getServiceDescriptor() {
|
||||
io.grpc.ServiceDescriptor result = serviceDescriptor;
|
||||
if (result == null) {
|
||||
synchronized (EchoTestServiceGrpc.class) {
|
||||
result = serviceDescriptor;
|
||||
if (result == null) {
|
||||
serviceDescriptor = result = io.grpc.ServiceDescriptor.newBuilder(SERVICE_NAME)
|
||||
.setSchemaDescriptor(new EchoTestServiceFileDescriptorSupplier())
|
||||
.addMethod(getEchoMethod())
|
||||
.addMethod(getForwardEchoMethod())
|
||||
.build();
|
||||
}
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,370 @@
|
|||
/*
|
||||
* Copyright 2022 The gRPC Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package io.grpc.testing.istio;
|
||||
|
||||
import com.google.common.annotations.VisibleForTesting;
|
||||
import com.google.common.collect.ImmutableMap;
|
||||
import io.grpc.ForwardingServerCall;
|
||||
import io.grpc.Grpc;
|
||||
import io.grpc.InsecureChannelCredentials;
|
||||
import io.grpc.InsecureServerCredentials;
|
||||
import io.grpc.ManagedChannel;
|
||||
import io.grpc.ManagedChannelBuilder;
|
||||
import io.grpc.Metadata;
|
||||
import io.grpc.Server;
|
||||
import io.grpc.ServerCall;
|
||||
import io.grpc.ServerCallHandler;
|
||||
import io.grpc.ServerInterceptor;
|
||||
import io.grpc.ServerInterceptors;
|
||||
import io.grpc.ServerServiceDefinition;
|
||||
import io.grpc.Status;
|
||||
import io.grpc.StatusRuntimeException;
|
||||
import io.grpc.stub.MetadataUtils;
|
||||
import io.grpc.stub.StreamObserver;
|
||||
import io.grpc.testing.istio.EchoTestServiceGrpc.EchoTestServiceImplBase;
|
||||
import io.grpc.testing.istio.Istio.ForwardEchoRequest;
|
||||
import io.grpc.testing.istio.Istio.ForwardEchoResponse;
|
||||
import java.io.IOException;
|
||||
import java.net.InetAddress;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.SocketAddress;
|
||||
import java.time.Duration;
|
||||
import java.time.Instant;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Random;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.logging.Level;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
/**
|
||||
* This class implements the Istio echo server functionality similar to
|
||||
* https://github.com/istio/istio/blob/master/pkg/test/echo/server/endpoint/grpc.go .
|
||||
* Please see Istio framework docs https://github.com/istio/istio/wiki/Istio-Test-Framework .
|
||||
*/
|
||||
public class EchoTestServer {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(EchoTestServer.class.getName());
|
||||
|
||||
private static final String REQUEST_ID = "x-request-id";
|
||||
private static final String STATUS_CODE = "StatusCode";
|
||||
private static final String HOST = "Host";
|
||||
private static final String HOSTNAME = "Hostname";
|
||||
private static final String REQUEST_HEADER = "RequestHeader";
|
||||
private static final String IP = "IP";
|
||||
public static final String GRPC_SCHEME = "grpc://";
|
||||
|
||||
@VisibleForTesting List<Server> servers;
|
||||
|
||||
/**
|
||||
* Preprocess args, for two things:
|
||||
* 1. merge duplicate flags. So "--grpc=8080 --grpc=9090" becomes
|
||||
* "--grpc=8080,9090".
|
||||
* 2. replace '-' to '_'. So "--istio-version=123" becomes
|
||||
* "--istio_version=123" (so exclude the leading "--").
|
||||
**/
|
||||
@VisibleForTesting
|
||||
static Map<String, List<String>> preprocessArgs(String[] args) {
|
||||
HashMap<String, List<String>> argsMap = new HashMap<>();
|
||||
for (String arg : args) {
|
||||
String[] keyValue = arg.split("=", 2);
|
||||
|
||||
if (keyValue.length == 2) {
|
||||
String key = keyValue[0];
|
||||
String value = keyValue[1];
|
||||
|
||||
key = key.substring(0, 2) + key.substring(2).replace('-', '_');
|
||||
List<String> oldValue = argsMap.get(key);
|
||||
if (oldValue == null) {
|
||||
oldValue = new ArrayList<>();
|
||||
}
|
||||
oldValue.add(value);
|
||||
argsMap.put(key, oldValue);
|
||||
}
|
||||
}
|
||||
return ImmutableMap.<String, List<String>>builder().putAll(argsMap).build();
|
||||
}
|
||||
|
||||
/** Turn gRPC ports from a string list to an int list. */
|
||||
@VisibleForTesting
|
||||
static List<Integer> getGrpcPorts(Map<String, List<String>> args) {
|
||||
List<String> grpcPorts = args.get("--grpc");
|
||||
List<Integer> grpcPortsInt = new ArrayList<>(grpcPorts.size());
|
||||
|
||||
for (String port : grpcPorts) {
|
||||
grpcPortsInt.add(Integer.parseInt(port));
|
||||
}
|
||||
return grpcPortsInt;
|
||||
}
|
||||
|
||||
private static String determineHostname() {
|
||||
try {
|
||||
return InetAddress.getLocalHost().getHostName();
|
||||
} catch (IOException ex) {
|
||||
logger.log(Level.INFO, "Failed to determine hostname. Will generate one", ex);
|
||||
}
|
||||
// let's make an identifier for ourselves.
|
||||
return "generated-" + new Random().nextInt();
|
||||
}
|
||||
|
||||
/**
|
||||
* The main application allowing this program to be launched from the command line.
|
||||
*/
|
||||
public static void main(String[] args) throws Exception {
|
||||
Map<String, List<String>> processedArgs = preprocessArgs(args);
|
||||
List<Integer> grpcPorts = getGrpcPorts(processedArgs);
|
||||
|
||||
String hostname = determineHostname();
|
||||
EchoTestServer echoTestServer = new EchoTestServer();
|
||||
echoTestServer.runServers(grpcPorts, hostname);
|
||||
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
|
||||
try {
|
||||
System.out.println("Shutting down");
|
||||
echoTestServer.stopServers();
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.SEVERE, "stopServers", e);
|
||||
throw e;
|
||||
}
|
||||
}));
|
||||
echoTestServer.blockUntilShutdown();
|
||||
}
|
||||
|
||||
void runServers(List<Integer> grpcPorts, String hostname) throws IOException {
|
||||
ServerServiceDefinition service = ServerInterceptors.intercept(
|
||||
new EchoTestServiceImpl(hostname), new EchoTestServerInterceptor());
|
||||
servers = new ArrayList<>(grpcPorts.size() + 1);
|
||||
for (int port : grpcPorts) {
|
||||
runServer(port, service);
|
||||
}
|
||||
}
|
||||
|
||||
void runServer(int port, ServerServiceDefinition service) throws IOException {
|
||||
logger.log(Level.INFO, "Listening GRPC on " + port);
|
||||
servers.add(Grpc.newServerBuilderForPort(port, InsecureServerCredentials.create())
|
||||
.addService(service)
|
||||
.build().start());
|
||||
}
|
||||
|
||||
void stopServers() {
|
||||
for (Server server : servers) {
|
||||
server.shutdownNow();
|
||||
}
|
||||
}
|
||||
|
||||
void blockUntilShutdown() throws InterruptedException {
|
||||
for (Server server : servers) {
|
||||
if (!server.awaitTermination(5, TimeUnit.SECONDS)) {
|
||||
System.err.println("Timed out waiting for server shutdown");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static class EchoTestServerInterceptor implements ServerInterceptor {
|
||||
|
||||
@Override
|
||||
public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(ServerCall<ReqT, RespT> call,
|
||||
final Metadata requestHeaders, ServerCallHandler<ReqT, RespT> next) {
|
||||
final String methodName = call.getMethodDescriptor().getBareMethodName();
|
||||
|
||||
// we need this processing only for Echo
|
||||
if (!"Echo".equals(methodName)) {
|
||||
return next.startCall(call, requestHeaders);
|
||||
}
|
||||
|
||||
final SocketAddress peerAddress = call.getAttributes()
|
||||
.get(Grpc.TRANSPORT_ATTR_REMOTE_ADDR);
|
||||
|
||||
return next.startCall(
|
||||
new ForwardingServerCall.SimpleForwardingServerCall<ReqT, RespT>(call) {
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Override
|
||||
public void sendMessage(RespT message) {
|
||||
io.grpc.testing.istio.Istio.EchoResponse echoResponse =
|
||||
(io.grpc.testing.istio.Istio.EchoResponse) message;
|
||||
String oldMessage = echoResponse.getMessage();
|
||||
|
||||
EchoMessage echoMessage = new EchoMessage();
|
||||
|
||||
for (String key : requestHeaders.keys()) {
|
||||
if (!key.endsWith("-bin")) {
|
||||
|
||||
echoMessage.writeKeyValueForRequest(REQUEST_HEADER, key,
|
||||
requestHeaders.get(Metadata.Key.of(key, Metadata.ASCII_STRING_MARSHALLER)));
|
||||
}
|
||||
}
|
||||
// This is not a complete list. May need to add/remove fields later,
|
||||
// such as "ServiceVersion", "ServicePort", "URL", "Method", "ResponseHeader",
|
||||
// "Cluster", "IstioVersion"
|
||||
// Only keep the fields needed for now.
|
||||
if (peerAddress instanceof InetSocketAddress) {
|
||||
InetSocketAddress inetPeerAddress = (InetSocketAddress) peerAddress;
|
||||
echoMessage.writeKeyValue(IP, inetPeerAddress.getAddress().getHostAddress());
|
||||
}
|
||||
echoMessage.writeKeyValue(STATUS_CODE, "200");
|
||||
echoMessage.writeKeyValue(HOST, call.getAuthority());
|
||||
echoMessage.writeMessage(oldMessage);
|
||||
echoResponse =
|
||||
io.grpc.testing.istio.Istio.EchoResponse.newBuilder()
|
||||
.setMessage(echoMessage.toString())
|
||||
.build();
|
||||
super.sendMessage((RespT) echoResponse);
|
||||
}
|
||||
},
|
||||
requestHeaders);
|
||||
}
|
||||
}
|
||||
|
||||
private static class EchoTestServiceImpl extends EchoTestServiceImplBase {
|
||||
|
||||
private final String hostname;
|
||||
|
||||
EchoTestServiceImpl(String hostname) {
|
||||
this.hostname = hostname;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void echo(io.grpc.testing.istio.Istio.EchoRequest request,
|
||||
io.grpc.stub.StreamObserver<io.grpc.testing.istio.Istio.EchoResponse> responseObserver) {
|
||||
|
||||
EchoMessage echoMessage = new EchoMessage();
|
||||
echoMessage.writeKeyValue(HOSTNAME, hostname);
|
||||
echoMessage.writeKeyValue("Echo", request.getMessage());
|
||||
io.grpc.testing.istio.Istio.EchoResponse echoResponse
|
||||
= io.grpc.testing.istio.Istio.EchoResponse.newBuilder()
|
||||
.setMessage(echoMessage.toString())
|
||||
.build();
|
||||
|
||||
responseObserver.onNext(echoResponse);
|
||||
responseObserver.onCompleted();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void forwardEcho(ForwardEchoRequest request,
|
||||
StreamObserver<ForwardEchoResponse> responseObserver) {
|
||||
try {
|
||||
responseObserver.onNext(buildEchoResponse(request));
|
||||
responseObserver.onCompleted();
|
||||
} catch (InterruptedException e) {
|
||||
responseObserver.onError(e);
|
||||
Thread.currentThread().interrupt();
|
||||
} catch (Exception e) {
|
||||
responseObserver.onError(e);
|
||||
}
|
||||
}
|
||||
|
||||
private ForwardEchoResponse buildEchoResponse(ForwardEchoRequest request)
|
||||
throws InterruptedException {
|
||||
Istio.ForwardEchoResponse.Builder forwardEchoResponseBuilder
|
||||
= io.grpc.testing.istio.Istio.ForwardEchoResponse.newBuilder();
|
||||
String rawUrl = request.getUrl();
|
||||
if (!rawUrl.startsWith(GRPC_SCHEME)) {
|
||||
throw new StatusRuntimeException(
|
||||
Status.UNIMPLEMENTED.withDescription("protocol grpc:// required"));
|
||||
}
|
||||
rawUrl = rawUrl.substring(GRPC_SCHEME.length());
|
||||
|
||||
// May need to use xds security if urlScheme is "xds"
|
||||
ManagedChannelBuilder<?> channelBuilder = Grpc.newChannelBuilder(
|
||||
rawUrl, InsecureChannelCredentials.create());
|
||||
ManagedChannel channel = channelBuilder.build();
|
||||
|
||||
List<Istio.Header> requestHeaders = request.getHeadersList();
|
||||
Metadata metadata = new Metadata();
|
||||
|
||||
for (Istio.Header header : requestHeaders) {
|
||||
metadata.put(Metadata.Key.of(header.getKey(), Metadata.ASCII_STRING_MARSHALLER),
|
||||
header.getValue());
|
||||
}
|
||||
|
||||
int count = request.getCount() == 0 ? 1 : request.getCount();
|
||||
Duration durationPerQuery = Duration.ZERO;
|
||||
if (request.getQps() > 0) {
|
||||
durationPerQuery = Duration.ofNanos(
|
||||
Duration.ofSeconds(1).toNanos() / request.getQps());
|
||||
}
|
||||
logger.info("qps=" + request.getQps());
|
||||
logger.info("durationPerQuery=" + durationPerQuery);
|
||||
io.grpc.testing.istio.Istio.EchoRequest echoRequest
|
||||
= io.grpc.testing.istio.Istio.EchoRequest.newBuilder()
|
||||
.setMessage(request.getMessage())
|
||||
.build();
|
||||
Instant start = Instant.now();
|
||||
logger.info("starting instant=" + start);
|
||||
Duration expected = Duration.ZERO;
|
||||
for (int i = 0; i < count; i++) {
|
||||
Metadata currentMetadata = new Metadata();
|
||||
currentMetadata.merge(metadata);
|
||||
currentMetadata.put(
|
||||
Metadata.Key.of(REQUEST_ID, Metadata.ASCII_STRING_MARSHALLER), "" + i);
|
||||
EchoTestServiceGrpc.EchoTestServiceBlockingStub stub
|
||||
= EchoTestServiceGrpc.newBlockingStub(channel).withInterceptors(
|
||||
MetadataUtils.newAttachHeadersInterceptor(currentMetadata))
|
||||
.withDeadlineAfter(request.getTimeoutMicros(), TimeUnit.MICROSECONDS);
|
||||
String response = callEcho(stub, echoRequest, i);
|
||||
forwardEchoResponseBuilder.addOutput(response);
|
||||
Instant current = Instant.now();
|
||||
logger.info("after rpc instant=" + current);
|
||||
Duration elapsed = Duration.between(start, current);
|
||||
expected = expected.plus(durationPerQuery);
|
||||
Duration timeLeft = expected.minus(elapsed);
|
||||
logger.info("elapsed=" + elapsed + ", expected=" + expected + ", timeLeft=" + timeLeft);
|
||||
if (!timeLeft.isNegative()) {
|
||||
logger.info("sleeping for ms =" + timeLeft);
|
||||
Thread.sleep(timeLeft.toMillis());
|
||||
}
|
||||
}
|
||||
return forwardEchoResponseBuilder.build();
|
||||
}
|
||||
|
||||
private String callEcho(EchoTestServiceGrpc.EchoTestServiceBlockingStub stub,
|
||||
io.grpc.testing.istio.Istio.EchoRequest echoRequest, int count) {
|
||||
try {
|
||||
Istio.EchoResponse echoResponse = stub.echo(echoRequest);
|
||||
return echoResponse.getMessage();
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.INFO, "RPC failed " + count, e);
|
||||
}
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
private static class EchoMessage {
|
||||
private final StringBuilder sb = new StringBuilder();
|
||||
|
||||
void writeKeyValue(String key, String value) {
|
||||
sb.append(key).append("=").append(value).append("\n");
|
||||
}
|
||||
|
||||
void writeKeyValueForRequest(String requestHeader, String key, String value) {
|
||||
if (value != null) {
|
||||
writeKeyValue(requestHeader, key + ":" + value);
|
||||
}
|
||||
}
|
||||
|
||||
void writeMessage(String message) {
|
||||
sb.append(message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,86 @@
|
|||
// Copyright 2022 gRPC authors.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
syntax = "proto3";
|
||||
|
||||
import "google/protobuf/wrappers.proto";
|
||||
|
||||
package istio.testing;
|
||||
|
||||
option java_package = "io.grpc.testing.istio";
|
||||
option java_outer_classname = "Istio";
|
||||
|
||||
service EchoTestService {
|
||||
rpc Echo (EchoRequest) returns (EchoResponse);
|
||||
rpc ForwardEcho (ForwardEchoRequest) returns (ForwardEchoResponse);
|
||||
}
|
||||
|
||||
message EchoRequest {
|
||||
string message = 1;
|
||||
}
|
||||
|
||||
message EchoResponse {
|
||||
string message = 1;
|
||||
}
|
||||
|
||||
message Header {
|
||||
string key = 1;
|
||||
string value = 2;
|
||||
}
|
||||
|
||||
message ForwardEchoRequest {
|
||||
int32 count = 1;
|
||||
int32 qps = 2;
|
||||
int64 timeout_micros = 3;
|
||||
string url = 4;
|
||||
repeated Header headers = 5;
|
||||
string message = 6;
|
||||
// Method for the request. Valid only for HTTP
|
||||
string method = 9;
|
||||
// If true, requests will be sent using h2c prior knowledge
|
||||
bool http2 = 7;
|
||||
// If true, requests will be sent using http3
|
||||
bool http3 = 15;
|
||||
// If true, requests will not be sent until magic string is received
|
||||
bool serverFirst = 8;
|
||||
// If true, 301 redirects will be followed
|
||||
bool followRedirects = 14;
|
||||
// If non-empty, make the request with the corresponding cert and key.
|
||||
string cert = 10;
|
||||
string key = 11;
|
||||
// If non-empty, verify the server CA
|
||||
string caCert = 12;
|
||||
// If non-empty, make the request with the corresponding cert and key file.
|
||||
string certFile = 16;
|
||||
string keyFile = 17;
|
||||
// If non-empty, verify the server CA with the ca cert file.
|
||||
string caCertFile = 18;
|
||||
// Skip verifying peer's certificate.
|
||||
bool insecureSkipVerify = 19;
|
||||
// List of ALPNs to present. If not set, this will be automatically be set based on the protocol
|
||||
Alpn alpn = 13;
|
||||
// Server name (SNI) to present in TLS connections. If not set, Host will be used for http requests.
|
||||
string serverName = 20;
|
||||
// Expected response determines what string to look for in the response to validate TCP requests succeeded.
|
||||
// If not set, defaults to "StatusCode=200"
|
||||
google.protobuf.StringValue expectedResponse = 21;
|
||||
}
|
||||
|
||||
message Alpn {
|
||||
repeated string value = 1;
|
||||
}
|
||||
|
||||
message ForwardEchoResponse {
|
||||
repeated string output = 1;
|
||||
}
|
||||
|
|
@ -0,0 +1,191 @@
|
|||
/*
|
||||
* Copyright 2022 The gRPC Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package io.grpc.testing.istio;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotEquals;
|
||||
|
||||
import com.google.common.collect.ImmutableList;
|
||||
import com.google.common.collect.ImmutableSet;
|
||||
import com.google.common.collect.Range;
|
||||
import io.grpc.Grpc;
|
||||
import io.grpc.InsecureChannelCredentials;
|
||||
import io.grpc.ManagedChannel;
|
||||
import io.grpc.ManagedChannelBuilder;
|
||||
import io.grpc.Metadata;
|
||||
import io.grpc.stub.MetadataUtils;
|
||||
import java.io.IOException;
|
||||
import java.time.Duration;
|
||||
import java.time.Instant;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.JUnit4;
|
||||
|
||||
/**
|
||||
* Unit tests for {@link EchoTestServer}.
|
||||
*/
|
||||
@RunWith(JUnit4.class)
|
||||
public class EchoTestServerTest {
|
||||
|
||||
@Test
|
||||
public void preprocessArgsTest() {
|
||||
String[] splitArgs = TEST_ARGS.split(" ");
|
||||
Map<String, List<String>> processedArgs = EchoTestServer.preprocessArgs(splitArgs);
|
||||
|
||||
assertEquals(processedArgs.keySet(), ImmutableSet.copyOf(EXPECTED_KEY_SET));
|
||||
assertEquals(processedArgs.get("--server_first"), ImmutableList.of("16060", "16061"));
|
||||
assertEquals(processedArgs.get("--bind_ip"), ImmutableList.of("18082"));
|
||||
assertEquals(processedArgs.get("--bind_localhost"), ImmutableList.of("18084"));
|
||||
assertEquals(processedArgs.get("--version"), ImmutableList.of("\"v1\""));
|
||||
assertEquals(processedArgs.get("--grpc"), ImmutableList.of("\"17070\"", "\"17071\""));
|
||||
assertEquals(processedArgs.get("--tls"), ImmutableList.of("18443", "19443"));
|
||||
assertEquals(processedArgs.get("--cluster"), ImmutableList.of("\"cluster-0\""));
|
||||
assertEquals(processedArgs.get("--key"), ImmutableList.of("/cert.key"));
|
||||
assertEquals(processedArgs.get("--tcp"), ImmutableList.of("\"19090\"", "\"16060\"",
|
||||
"\"19091\"","\"16061\"","\"19092\""));
|
||||
assertEquals(processedArgs.get("--istio_version"), ImmutableList.of("3"));
|
||||
assertEquals(processedArgs.get("--crt"), ImmutableList.of("/cert.crt"));
|
||||
assertEquals(processedArgs.get("--metrics"), ImmutableList.of("15014"));
|
||||
assertEquals(
|
||||
processedArgs.get("--port"),
|
||||
ImmutableList.of(
|
||||
"\"18080\"",
|
||||
"\"18085\"",
|
||||
"\"18443\"",
|
||||
"\"18081\"",
|
||||
"\"19443\"",
|
||||
"\"18082\"",
|
||||
"\"18084\"",
|
||||
"\"18083\"",
|
||||
"\"8080\"",
|
||||
"\"3333\""));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void echoTest() throws IOException, InterruptedException {
|
||||
EchoTestServer echoTestServer = new EchoTestServer();
|
||||
|
||||
echoTestServer.runServers(ImmutableList.of(0, 0), "test-host");
|
||||
assertEquals(2, echoTestServer.servers.size());
|
||||
int port = echoTestServer.servers.get(0).getPort();
|
||||
assertNotEquals(0, port);
|
||||
assertNotEquals(0, echoTestServer.servers.get(1).getPort());
|
||||
|
||||
ManagedChannelBuilder<?> channelBuilder =
|
||||
Grpc.newChannelBuilderForAddress("localhost", port, InsecureChannelCredentials.create());
|
||||
ManagedChannel channel = channelBuilder.build();
|
||||
|
||||
Metadata metadata = new Metadata();
|
||||
metadata.put(Metadata.Key.of("header1", Metadata.ASCII_STRING_MARSHALLER), "value1");
|
||||
metadata.put(Metadata.Key.of("header2", Metadata.ASCII_STRING_MARSHALLER), "value2");
|
||||
|
||||
EchoTestServiceGrpc.EchoTestServiceBlockingStub stub =
|
||||
EchoTestServiceGrpc.newBlockingStub(channel)
|
||||
.withInterceptors(MetadataUtils.newAttachHeadersInterceptor(metadata));
|
||||
|
||||
io.grpc.testing.istio.Istio.EchoRequest echoRequest
|
||||
= io.grpc.testing.istio.Istio.EchoRequest.newBuilder()
|
||||
.setMessage("test-message1")
|
||||
.build();
|
||||
Istio.EchoResponse echoResponse = stub.echo(echoRequest);
|
||||
String echoMessage = echoResponse.getMessage();
|
||||
Set<String> lines = ImmutableSet.copyOf(echoMessage.split(System.lineSeparator()));
|
||||
|
||||
assertThat(lines).contains("RequestHeader=header1:value1");
|
||||
assertThat(lines).contains("RequestHeader=header2:value2");
|
||||
assertThat(lines).contains("Echo=test-message1");
|
||||
assertThat(lines).contains("Hostname=test-host");
|
||||
assertThat(lines).contains("Host=localhost:" + port);
|
||||
assertThat(lines).contains("StatusCode=200");
|
||||
|
||||
echoTestServer.stopServers();
|
||||
echoTestServer.blockUntilShutdown();
|
||||
}
|
||||
|
||||
static final int COUNT_OF_REQUESTS_TO_FORWARD = 60;
|
||||
|
||||
@Test
|
||||
public void forwardEchoTest() throws IOException {
|
||||
EchoTestServer echoTestServer = new EchoTestServer();
|
||||
|
||||
echoTestServer.runServers(ImmutableList.of(0, 0), "test-host");
|
||||
assertEquals(2, echoTestServer.servers.size());
|
||||
int port1 = echoTestServer.servers.get(0).getPort();
|
||||
int port2 = echoTestServer.servers.get(1).getPort();
|
||||
|
||||
ManagedChannelBuilder<?> channelBuilder =
|
||||
Grpc.newChannelBuilderForAddress("localhost", port1, InsecureChannelCredentials.create());
|
||||
ManagedChannel channel = channelBuilder.build();
|
||||
|
||||
Istio.ForwardEchoRequest forwardEchoRequest =
|
||||
Istio.ForwardEchoRequest.newBuilder()
|
||||
.setCount(COUNT_OF_REQUESTS_TO_FORWARD)
|
||||
.setQps(100)
|
||||
.setTimeoutMicros(100_000L) // 100 millis
|
||||
.setUrl("grpc://localhost:" + port2)
|
||||
.addHeaders(
|
||||
Istio.Header.newBuilder().setKey("test-key1").setValue("test-value1").build())
|
||||
.addHeaders(
|
||||
Istio.Header.newBuilder().setKey("test-key2").setValue("test-value2").build())
|
||||
.setMessage("forward-echo-test-message")
|
||||
.build();
|
||||
|
||||
EchoTestServiceGrpc.EchoTestServiceBlockingStub stub =
|
||||
EchoTestServiceGrpc.newBlockingStub(channel);
|
||||
|
||||
Instant start = Instant.now();
|
||||
Istio.ForwardEchoResponse forwardEchoResponse = stub.forwardEcho(forwardEchoRequest);
|
||||
Instant end = Instant.now();
|
||||
List<String> outputs = forwardEchoResponse.getOutputList();
|
||||
assertEquals(COUNT_OF_REQUESTS_TO_FORWARD, outputs.size());
|
||||
for (int i = 0; i < COUNT_OF_REQUESTS_TO_FORWARD; i++) {
|
||||
validateOutput(outputs.get(i), i);
|
||||
}
|
||||
long duration = Duration.between(start, end).toMillis();
|
||||
assertThat(duration).isIn(Range.closed(
|
||||
COUNT_OF_REQUESTS_TO_FORWARD * 10L, 2 * COUNT_OF_REQUESTS_TO_FORWARD * 10L));
|
||||
}
|
||||
|
||||
private static void validateOutput(String output, int i) {
|
||||
Set<String> lines = ImmutableSet.copyOf(output.split(System.lineSeparator()));
|
||||
assertThat(lines).contains("RequestHeader=x-request-id:" + i);
|
||||
assertThat(lines).contains("RequestHeader=test-key1:test-value1");
|
||||
assertThat(lines).contains("RequestHeader=test-key2:test-value2");
|
||||
assertThat(lines).contains("Hostname=test-host");
|
||||
assertThat(lines).contains("StatusCode=200");
|
||||
assertThat(lines).contains("Echo=forward-echo-test-message");
|
||||
}
|
||||
|
||||
private static final String[] EXPECTED_KEY_SET = {
|
||||
"--server_first",
|
||||
"--bind_ip", "--istio_version", "--bind_localhost", "--version", "--grpc", "--tls",
|
||||
"--cluster", "--key", "--tcp", "--crt", "--metrics", "--port"
|
||||
};
|
||||
|
||||
private static final String TEST_ARGS =
|
||||
"--metrics=15014 --cluster=\"cluster-0\" --port=\"18080\" --grpc=\"17070\" --port=\"18085\""
|
||||
+ " --tcp=\"19090\" --port=\"18443\" --tls=18443 --tcp=\"16060\" --server_first=16060"
|
||||
+ " --tcp=\"19091\" --tcp=\"16061\" --server_first=16061 --port=\"18081\""
|
||||
+ " --grpc=\"17071\" --port=\"19443\" --tls=19443 --port=\"18082\" --bind_ip=18082"
|
||||
+ " --port=\"18084\" --bind_localhost=18084 --tcp=\"19092\" --port=\"18083\""
|
||||
+ " --port=\"8080\" --port=\"3333\" --version=\"v1\" --istio-version=3 --crt=/cert.crt"
|
||||
+ " --key=/cert.key";
|
||||
}
|
||||
Loading…
Reference in New Issue