Examples with GetProvisionInfoResponse org.apache.beam.model.fnexecution.v1.ProvisionApi.GetProvisionInfoResponse used on opensource projects

Example 1 with GetProvisionInfoResponse

use of org.apache.beam.model.fnexecution.v1.ProvisionApi.GetProvisionInfoResponse in project beam by apache.

the class StaticGrpcProvisionServiceTest method returnsProvisionInfo.

@Test
public void returnsProvisionInfo() throws Exception {
    Struct options = Struct.newBuilder().putFields("foo", Value.newBuilder().setBoolValue(true).build()).putFields("bar", Value.newBuilder().setNumberValue(2.5).build()).putFields("baz", Value.newBuilder().setListValue(ListValue.newBuilder().addValues(Value.newBuilder().setStructValue(Struct.newBuilder().putFields("spam", Value.newBuilder().setNullValue(NullValue.NULL_VALUE).build()))).build()).build()).build();
    ProvisionInfo info = ProvisionInfo.newBuilder().setPipelineOptions(options).build();
    GrpcFnServer<StaticGrpcProvisionService> server = GrpcFnServer.allocatePortAndCreateFor(StaticGrpcProvisionService.create(info, GrpcContextHeaderAccessorProvider.getHeaderAccessor()), InProcessServerFactory.create());
    ProvisionServiceBlockingStub stub = ProvisionServiceGrpc.newBlockingStub(InProcessChannelBuilder.forName(server.getApiServiceDescriptor().getUrl()).intercept(AddHarnessIdInterceptor.create("test_worker")).build());
    GetProvisionInfoResponse provisionResponse = stub.getProvisionInfo(GetProvisionInfoRequest.getDefaultInstance());
    assertThat(provisionResponse.getInfo(), equalTo(info));
}