flasn4nt
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use voidMethod methoddescriptor

This commit is contained in:
Mateus Azis 2025-07-15 17:55:07 -07:00
parent bbd5983d14
commit 29dd39be85
2 changed files with 10 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
binder/src/test/java/io/grpc/binder/internal/PendingAuthListenerTest.java
View File

@ -35,6 +35,7 @@ import io.grpc.inprocess.InProcessServerBuilder;
import io.grpc.stub.ClientCalls; import io.grpc.stub.ClientCalls;
import io.grpc.stub.ServerCalls; import io.grpc.stub.ServerCalls;
import io.grpc.testing.GrpcCleanupRule; import io.grpc.testing.GrpcCleanupRule;
import io.grpc.testing.TestMethodDescriptors;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Before; import org.junit.Before;
@ -126,15 +127,10 @@ public final class PendingAuthListenerTest {
@Test @Test
public void whenStartCallFails_closesTheCallWithInternalStatus() throws Exception { public void whenStartCallFails_closesTheCallWithInternalStatus() throws Exception {
// Arrange // Arrange
String name = "test_server"; String name = TestMethodDescriptors.SERVICE_NAME;
AtomicBoolean closed = new AtomicBoolean(false); AtomicBoolean closed = new AtomicBoolean(false);
MethodDescriptor<String, String> method = MethodDescriptor<Void, Void> method = TestMethodDescriptors.voidMethod();
MethodDescriptor ServerCallHandler<Void, Void> callHandler =
.newBuilder(StringMarshaller.INSTANCE, StringMarshaller.INSTANCE)
.setFullMethodName("test_server/method")
.setType(MethodDescriptor.MethodType.UNARY)
.build();
ServerCallHandler<String, String> callHandler =
ServerCalls.asyncUnaryCall((req, respObserver) -> { ServerCalls.asyncUnaryCall((req, respObserver) -> {
throw new IllegalStateException("ooops"); throw new IllegalStateException("ooops");
}); });
@ -181,28 +177,9 @@ public final class PendingAuthListenerTest {
// Act // Act
assertThrows(StatusRuntimeException.class, () -> ClientCalls.blockingUnaryCall(channel, 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 // Assert
assertThat(closed.get()).isTrue(); 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);
}
}
}
} }

7
testing/src/main/java/io/grpc/testing/TestMethodDescriptors.java
View File

@ -30,16 +30,19 @@ import java.io.InputStream;
public final class TestMethodDescriptors { public final class TestMethodDescriptors {
private 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 * 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 * @since 1.1.0
*/ */
public static MethodDescriptor<Void, Void> voidMethod() { public static MethodDescriptor<Void, Void> voidMethod() {
return MethodDescriptor.<Void, Void>newBuilder() return MethodDescriptor.<Void, Void>newBuilder()
.setType(MethodType.UNARY) .setType(MethodType.UNARY)
.setFullMethodName(MethodDescriptor.generateFullMethodName("service_foo", "method_bar")) .setFullMethodName(MethodDescriptor.generateFullMethodName(SERVICE_NAME, "method_bar"))
.setRequestMarshaller(TestMethodDescriptors.voidMarshaller()) .setRequestMarshaller(TestMethodDescriptors.voidMarshaller())
.setResponseMarshaller(TestMethodDescriptors.voidMarshaller()) .setResponseMarshaller(TestMethodDescriptors.voidMarshaller())
.build(); .build();