Examples with CallOptions org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions used on opensource projects

Example 16 with CallOptions

use of org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions in project grpc-java by grpc.

the class AbstractBlockingStubTest method defaultCallOptions.

@Test
public void defaultCallOptions() {
    NoopBlockingStub stub = NoopBlockingStub.newStub(new StubFactory<NoopBlockingStub>() {

        @Override
        public NoopBlockingStub newStub(Channel channel, CallOptions callOptions) {
            return create(channel, callOptions);
        }
    }, channel, CallOptions.DEFAULT);
    assertThat(stub.getCallOptions().getOption(ClientCalls.STUB_TYPE_OPTION)).isEqualTo(StubType.BLOCKING);
}

Example 17 with CallOptions

use of org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions in project grpc-java by grpc.

the class ClientCallsTest method blockingUnaryCall_HasBlockingStubType.

@Test
public void blockingUnaryCall_HasBlockingStubType() {
    NoopClientCall<Integer, Integer> call = new NoopClientCall<Integer, Integer>() {

        @Override
        public void start(io.grpc.ClientCall.Listener<Integer> listener, Metadata headers) {
            listener.onMessage(1);
            listener.onClose(Status.OK, new Metadata());
        }
    };
    when(mockChannel.newCall(ArgumentMatchers.<MethodDescriptor<Integer, Integer>>any(), any(CallOptions.class))).thenReturn(call);
    Integer unused = ClientCalls.blockingUnaryCall(mockChannel, UNARY_METHOD, CallOptions.DEFAULT, 1);
    verify(mockChannel).newCall(methodDescriptorCaptor.capture(), callOptionsCaptor.capture());
    CallOptions capturedCallOption = callOptionsCaptor.getValue();
    assertThat(capturedCallOption.getOption(ClientCalls.STUB_TYPE_OPTION)).isEquivalentAccordingToCompareTo(StubType.BLOCKING);
}

Example 18 with CallOptions

use of org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions in project grpc-java by grpc.

the class ClientCallsTest method blockingServerStreamingCall_HasBlockingStubType.

@Test
public void blockingServerStreamingCall_HasBlockingStubType() {
    NoopClientCall<Integer, Integer> call = new NoopClientCall<Integer, Integer>() {

        @Override
        public void start(io.grpc.ClientCall.Listener<Integer> listener, Metadata headers) {
            listener.onMessage(1);
            listener.onClose(Status.OK, new Metadata());
        }
    };
    when(mockChannel.newCall(ArgumentMatchers.<MethodDescriptor<Integer, Integer>>any(), any(CallOptions.class))).thenReturn(call);
    Iterator<Integer> unused = ClientCalls.blockingServerStreamingCall(mockChannel, UNARY_METHOD, CallOptions.DEFAULT, 1);
    verify(mockChannel).newCall(methodDescriptorCaptor.capture(), callOptionsCaptor.capture());
    CallOptions capturedCallOption = callOptionsCaptor.getValue();
    assertThat(capturedCallOption.getOption(ClientCalls.STUB_TYPE_OPTION)).isEquivalentAccordingToCompareTo(StubType.BLOCKING);
}

Example 19 with CallOptions

use of org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions in project grpc-java by grpc.

the class AbstractAsyncStubTest method newStub_futureStub_throwsException.

@Test
@SuppressWarnings("AssertionFailureIgnored")
public void newStub_futureStub_throwsException() {
    try {
        NoopFutureStub unused = NoopAsyncStub.newStub(new StubFactory<NoopFutureStub>() {

            @Override
            public NoopFutureStub newStub(Channel channel, CallOptions callOptions) {
                return new NoopFutureStub(channel, callOptions);
            }
        }, channel, CallOptions.DEFAULT);
        fail("should not reach here");
    } catch (AssertionError e) {
        assertThat(e).hasMessageThat().startsWith("Expected AbstractAsyncStub");
    }
}

Example 20 with CallOptions

use of org.apache.beam.vendor.grpc.v1p43p2.io.grpc.CallOptions in project grpc-java by grpc.

the class AbstractAsyncStubTest method defaultCallOptions.

@Test
public void defaultCallOptions() {
    NoopAsyncStub stub = NoopAsyncStub.newStub(new StubFactory<NoopAsyncStub>() {

        @Override
        public NoopAsyncStub newStub(Channel channel, CallOptions callOptions) {
            return create(channel, callOptions);
        }
    }, channel, CallOptions.DEFAULT);
    assertThat(stub.getCallOptions().getOption(ClientCalls.STUB_TYPE_OPTION)).isEqualTo(StubType.ASYNC);
}