mirror of https://github.com/grpc/grpc-java.git
use voidMethod methoddescriptor
This commit is contained in:
parent
bbd5983d14
commit
29dd39be85
|
@ -35,6 +35,7 @@ import io.grpc.inprocess.InProcessServerBuilder;
|
|||
import io.grpc.stub.ClientCalls;
|
||||
import io.grpc.stub.ServerCalls;
|
||||
import io.grpc.testing.GrpcCleanupRule;
|
||||
import io.grpc.testing.TestMethodDescriptors;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
|
@ -126,15 +127,10 @@ public final class PendingAuthListenerTest {
|
|||
@Test
|
||||
public void whenStartCallFails_closesTheCallWithInternalStatus() throws Exception {
|
||||
// Arrange
|
||||
String name = "test_server";
|
||||
String name = TestMethodDescriptors.SERVICE_NAME;
|
||||
AtomicBoolean closed = new AtomicBoolean(false);
|
||||
MethodDescriptor<String, String> method =
|
||||
MethodDescriptor
|
||||
.newBuilder(StringMarshaller.INSTANCE, StringMarshaller.INSTANCE)
|
||||
.setFullMethodName("test_server/method")
|
||||
.setType(MethodDescriptor.MethodType.UNARY)
|
||||
.build();
|
||||
ServerCallHandler<String, String> callHandler =
|
||||
MethodDescriptor<Void, Void> method = TestMethodDescriptors.voidMethod();
|
||||
ServerCallHandler<Void, Void> callHandler =
|
||||
ServerCalls.asyncUnaryCall((req, respObserver) -> {
|
||||
throw new IllegalStateException("ooops");
|
||||
});
|
||||
|
@ -181,28 +177,9 @@ public final class PendingAuthListenerTest {
|
|||
|
||||
// Act
|
||||
assertThrows(StatusRuntimeException.class, () -> ClientCalls.blockingUnaryCall(channel,
|
||||
method, CallOptions.DEFAULT.withDeadlineAfter(Duration.ofSeconds(5)), "foo"));
|
||||
method, CallOptions.DEFAULT.withDeadlineAfter(Duration.ofSeconds(5)), /* request= */ (Void)null));
|
||||
|
||||
// Assert
|
||||
assertThat(closed.get()).isTrue();
|
||||
}
|
||||
|
||||
|
||||
private static class StringMarshaller implements MethodDescriptor.Marshaller<String> {
|
||||
public static final StringMarshaller INSTANCE = new StringMarshaller();
|
||||
|
||||
@Override
|
||||
public InputStream stream(String value) {
|
||||
return new ByteArrayInputStream(value.getBytes(UTF_8));
|
||||
}
|
||||
|
||||
@Override
|
||||
public String parse(InputStream stream) {
|
||||
try {
|
||||
return new String(ByteStreams.toByteArray(stream), UTF_8);
|
||||
} catch (IOException ex) {
|
||||
throw new RuntimeException(ex);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -30,16 +30,19 @@ import java.io.InputStream;
|
|||
public final class TestMethodDescriptors {
|
||||
private TestMethodDescriptors() {}
|
||||
|
||||
/** The name of the service that the method returned by {@link #voidMethod()} uses. */
|
||||
public static final String SERVICE_NAME = "service_foo";
|
||||
|
||||
/**
|
||||
* Creates a new method descriptor that always creates zero length messages, and always parses to
|
||||
* null objects.
|
||||
* null objects. It is part of the service named {@link #SERVICE_NAME}.
|
||||
*
|
||||
* @since 1.1.0
|
||||
*/
|
||||
public static MethodDescriptor<Void, Void> voidMethod() {
|
||||
return MethodDescriptor.<Void, Void>newBuilder()
|
||||
.setType(MethodType.UNARY)
|
||||
.setFullMethodName(MethodDescriptor.generateFullMethodName("service_foo", "method_bar"))
|
||||
.setFullMethodName(MethodDescriptor.generateFullMethodName(SERVICE_NAME, "method_bar"))
|
||||
.setRequestMarshaller(TestMethodDescriptors.voidMarshaller())
|
||||
.setResponseMarshaller(TestMethodDescriptors.voidMarshaller())
|
||||
.build();
|
||||
|
|
Loading…
Reference in New Issue