mirror of https://github.com/grpc/grpc-java.git
Added unit test for Bootstrapper.
This commit is contained in:
Chengyuan Zhang
parent
27ec907e68
commit
3ae88de5fe
|
|
@ -0,0 +1,219 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2019 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.xds;
|
||||||
|
|
||||||
|
import static com.google.common.truth.Truth.assertThat;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
import com.google.protobuf.InvalidProtocolBufferException;
|
||||||
|
import com.google.protobuf.Struct;
|
||||||
|
import com.google.protobuf.Value;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.ApiConfigSource;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.ApiConfigSource.ApiType;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.GrpcService;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.GrpcService.GoogleGrpc;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.Locality;
|
||||||
|
import io.envoyproxy.envoy.api.v2.core.Node;
|
||||||
|
import io.grpc.xds.Bootstrapper.FileBasedBootstrapper;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.junit.runners.JUnit4;
|
||||||
|
|
||||||
|
/** Unit tests for {@link Bootstrapper}. */
|
||||||
|
@RunWith(JUnit4.class)
|
||||||
|
public class BootstrapperTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void validBootstrap() {
|
||||||
|
Bootstrap config =
|
||||||
|
Bootstrap.newBuilder()
|
||||||
|
.setNode(
|
||||||
|
Node.newBuilder()
|
||||||
|
.setId("ENVOY_NODE_ID")
|
||||||
|
.setLocality(
|
||||||
|
Locality.newBuilder().setZone("ENVOY_ZONE").setRegion("ENVOY_REGION"))
|
||||||
|
.setMetadata(
|
||||||
|
Struct.newBuilder()
|
||||||
|
.putFields("TRAFFICDIRECTOR_INTERCEPTION_PORT",
|
||||||
|
Value.newBuilder().setStringValue("ENVOY_PORT").build())
|
||||||
|
.putFields("TRAFFICDIRECTOR_NETWORK_NAME",
|
||||||
|
Value.newBuilder().setStringValue("VPC_NETWORK_NAME").build())))
|
||||||
|
.setXdsServer(ApiConfigSource.newBuilder()
|
||||||
|
.setApiType(ApiType.GRPC)
|
||||||
|
.addGrpcServices(
|
||||||
|
GrpcService.newBuilder()
|
||||||
|
.setGoogleGrpc(
|
||||||
|
GoogleGrpc.newBuilder()
|
||||||
|
.setTargetUri("trafficdirector.googleapis.com:443").build())))
|
||||||
|
.build();
|
||||||
|
|
||||||
|
Bootstrapper bootstrapper = new FileBasedBootstrapper(config);
|
||||||
|
assertThat(bootstrapper.getBalancerName()).isEqualTo("trafficdirector.googleapis.com:443");
|
||||||
|
assertThat(bootstrapper.getNode())
|
||||||
|
.isEqualTo(
|
||||||
|
Node.newBuilder()
|
||||||
|
.setId("ENVOY_NODE_ID")
|
||||||
|
.setLocality(Locality.newBuilder().setZone("ENVOY_ZONE").setRegion("ENVOY_REGION"))
|
||||||
|
.setMetadata(
|
||||||
|
Struct.newBuilder()
|
||||||
|
.putFields("TRAFFICDIRECTOR_INTERCEPTION_PORT",
|
||||||
|
Value.newBuilder().setStringValue("ENVOY_PORT").build())
|
||||||
|
.putFields("TRAFFICDIRECTOR_NETWORK_NAME",
|
||||||
|
Value.newBuilder().setStringValue("VPC_NETWORK_NAME").build())
|
||||||
|
.build()).build());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void unsupportedApiType() {
|
||||||
|
Bootstrap config =
|
||||||
|
Bootstrap.newBuilder()
|
||||||
|
.setNode(
|
||||||
|
Node.newBuilder()
|
||||||
|
.setId("ENVOY_NODE_ID")
|
||||||
|
.setLocality(
|
||||||
|
Locality.newBuilder().setZone("ENVOY_ZONE").setRegion("ENVOY_REGION"))
|
||||||
|
.setMetadata(
|
||||||
|
Struct.newBuilder()
|
||||||
|
.putFields("TRAFFICDIRECTOR_INTERCEPTION_PORT",
|
||||||
|
Value.newBuilder().setStringValue("ENVOY_PORT").build())
|
||||||
|
.putFields("TRAFFICDIRECTOR_NETWORK_NAME",
|
||||||
|
Value.newBuilder().setStringValue("VPC_NETWORK_NAME").build())))
|
||||||
|
.setXdsServer(ApiConfigSource.newBuilder()
|
||||||
|
.setApiType(ApiType.REST)
|
||||||
|
.addGrpcServices(
|
||||||
|
GrpcService.newBuilder()
|
||||||
|
.setGoogleGrpc(
|
||||||
|
GoogleGrpc.newBuilder()
|
||||||
|
.setTargetUri("trafficdirector.googleapis.com:443").build())))
|
||||||
|
.build();
|
||||||
|
|
||||||
|
try {
|
||||||
|
new FileBasedBootstrapper(config);
|
||||||
|
fail("RuntimeException should have been thrown");
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
// Expected.
|
||||||
|
assertThat(e.getMessage()).isEqualTo("Unexpected api type: REST");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void tooManyGrpcServices() {
|
||||||
|
Bootstrap config =
|
||||||
|
Bootstrap.newBuilder()
|
||||||
|
.setNode(
|
||||||
|
Node.newBuilder()
|
||||||
|
.setId("ENVOY_NODE_ID")
|
||||||
|
.setLocality(
|
||||||
|
Locality.newBuilder().setZone("ENVOY_ZONE").setRegion("ENVOY_REGION"))
|
||||||
|
.setMetadata(
|
||||||
|
Struct.newBuilder()
|
||||||
|
.putFields("TRAFFICDIRECTOR_INTERCEPTION_PORT",
|
||||||
|
Value.newBuilder().setStringValue("ENVOY_PORT").build())
|
||||||
|
.putFields("TRAFFICDIRECTOR_NETWORK_NAME",
|
||||||
|
Value.newBuilder().setStringValue("VPC_NETWORK_NAME").build())))
|
||||||
|
.setXdsServer(ApiConfigSource.newBuilder()
|
||||||
|
.setApiType(ApiType.GRPC)
|
||||||
|
.addGrpcServices(
|
||||||
|
GrpcService.newBuilder()
|
||||||
|
.setGoogleGrpc(
|
||||||
|
GoogleGrpc.newBuilder()
|
||||||
|
.setTargetUri("trafficdirector.googleapis.com:443").build()))
|
||||||
|
.addGrpcServices(
|
||||||
|
GrpcService.newBuilder()
|
||||||
|
.setGoogleGrpc(
|
||||||
|
GoogleGrpc.newBuilder()
|
||||||
|
.setTargetUri("foobar.googleapis.com:443").build()))
|
||||||
|
)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
try {
|
||||||
|
new FileBasedBootstrapper(config);
|
||||||
|
fail("RuntimeException should have been thrown");
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
// Expected.
|
||||||
|
assertThat(e.getMessage())
|
||||||
|
.isEqualTo("Unexpected number of gRPC services: expected: 1, actual: 2");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseBootstrap_emptyData() {
|
||||||
|
String rawData = "";
|
||||||
|
try {
|
||||||
|
Bootstrapper.parseConfig(rawData);
|
||||||
|
fail("InvalidProtocolBufferException should have been thrown");
|
||||||
|
} catch (InvalidProtocolBufferException e) {
|
||||||
|
// Expected.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseBootstrap_invalidNodeProto() {
|
||||||
|
String rawData = "{"
|
||||||
|
+ "\"node\": {"
|
||||||
|
+ "\"id\": \"ENVOY_NODE_ID\","
|
||||||
|
+ "\"bad_field\": \"bad_value\""
|
||||||
|
+ "\"locality\": {"
|
||||||
|
+ "\"zone\": \"ENVOY_ZONE\"},"
|
||||||
|
+ "\"metadata\": {"
|
||||||
|
+ "\"TRAFFICDIRECTOR_INTERCEPTION_PORT\": \"ENVOY_PORT\", "
|
||||||
|
+ "\"TRAFFICDIRECTOR_NETWORK_NAME\": \"VPC_NETWORK_NAME\""
|
||||||
|
+ "}"
|
||||||
|
+ "},"
|
||||||
|
+ "\"xds_server\": {"
|
||||||
|
+ "\"api_type\": \"GRPC\","
|
||||||
|
+ "\"grpc_services\": "
|
||||||
|
+ "[ {\"google_grpc\": {\"target_uri\": \"trafficdirector.googleapis.com:443\"} } ]"
|
||||||
|
+ "} "
|
||||||
|
+ "}";
|
||||||
|
|
||||||
|
try {
|
||||||
|
Bootstrapper.parseConfig(rawData);
|
||||||
|
fail("InvalidProtocolBufferException should have been thrown");
|
||||||
|
} catch (InvalidProtocolBufferException e) {
|
||||||
|
// Expected.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseBootstrap_invalidApiConfigSourceProto() {
|
||||||
|
String rawData = "{"
|
||||||
|
+ "\"node\": {"
|
||||||
|
+ "\"id\": \"ENVOY_NODE_ID\","
|
||||||
|
+ "\"locality\": {"
|
||||||
|
+ "\"zone\": \"ENVOY_ZONE\"},"
|
||||||
|
+ "\"metadata\": {"
|
||||||
|
+ "\"TRAFFICDIRECTOR_INTERCEPTION_PORT\": \"ENVOY_PORT\", "
|
||||||
|
+ "\"TRAFFICDIRECTOR_NETWORK_NAME\": \"VPC_NETWORK_NAME\""
|
||||||
|
+ "}"
|
||||||
|
+ "},"
|
||||||
|
+ "\"xds_server\": {"
|
||||||
|
+ "\"api_type\": \"GRPC\","
|
||||||
|
+ "\"bad_field\": \"bad_value\""
|
||||||
|
+ "\"grpc_services\": "
|
||||||
|
+ "[ {\"google_grpc\": {\"target_uri\": \"trafficdirector.googleapis.com:443\"} } ]"
|
||||||
|
+ "} "
|
||||||
|
+ "}";
|
||||||
|
|
||||||
|
try {
|
||||||
|
Bootstrapper.parseConfig(rawData);
|
||||||
|
fail("InvalidProtocolBufferException should have been thrown");
|
||||||
|
} catch (InvalidProtocolBufferException e) {
|
||||||
|
// Expected.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue