diff --git a/binder/src/test/java/io/grpc/binder/internal/PendingAuthListenerTest.java b/binder/src/test/java/io/grpc/binder/internal/PendingAuthListenerTest.java index f477c3857d..df9fb92af1 100644 --- a/binder/src/test/java/io/grpc/binder/internal/PendingAuthListenerTest.java +++ b/binder/src/test/java/io/grpc/binder/internal/PendingAuthListenerTest.java @@ -17,6 +17,8 @@ import io.grpc.CallOptions; import io.grpc.Channel; import io.grpc.ClientCall; import io.grpc.ClientInterceptor; +import io.grpc.ForwardingClientCall; +import io.grpc.ForwardingClientCallListener; import io.grpc.ManagedChannel; import io.grpc.Metadata; import io.grpc.MethodDescriptor; @@ -157,52 +159,19 @@ public final class PendingAuthListenerTest { public ClientCall interceptCall( MethodDescriptor method, CallOptions callOptions, Channel next) { ClientCall delegate = next.newCall(method, callOptions); - return new ClientCall() { + return new ForwardingClientCall.SimpleForwardingClientCall(delegate){ @Override public void start(Listener responseListener, Metadata headers) { - Listener listenerWrapper = new Listener() { - @Override - public void onHeaders(Metadata headers) { - responseListener.onHeaders(headers); - } - - @Override - public void onMessage(RespT message) { - responseListener.onMessage(message); - } - - @Override - public void onReady() { - responseListener.onReady(); - } - - @Override - public void onClose(Status status, Metadata trailers) { - responseListener.onClose(status, trailers); - closed.set(true); - } - }; - delegate.start(listenerWrapper, headers); - } - - @Override - public void request(int numMessages) { - delegate.request(numMessages); - } - - @Override - public void cancel(@Nullable String message, @Nullable Throwable cause) { - delegate.cancel(message, cause); - } - - @Override - public void halfClose() { - delegate.halfClose(); - } - - @Override - public void sendMessage(ReqT message) { - delegate.sendMessage(message); + ClientCall.Listener wrappedListener = + new ForwardingClientCallListener.SimpleForwardingClientCallListener( + responseListener){ + @Override + public void onClose(Status status, Metadata trailers) { + super.onClose(status, trailers); + closed.set(true); + } + }; + super.start(wrappedListener, headers); } }; } @@ -218,6 +187,7 @@ public final class PendingAuthListenerTest { assertThat(closed.get()).isTrue(); } + private static class StringMarshaller implements MethodDescriptor.Marshaller { public static final StringMarshaller INSTANCE = new StringMarshaller();