Class ServerImplTest

core/src/test/java/io/grpc/internal/ServerImplTest.java:121–1898

Unit tests for ServerImpl.

/** Unit tests for {@link ServerImpl}. */
@RunWith(JUnit4.class)
public class ServerImplTest {
  private static final IntegerMarshaller INTEGER_MARSHALLER = IntegerMarshaller.INSTANCE;
  private static final StringMarshaller STRING_MARSHALLER = StringMarshaller.INSTANCE;
  private static final MethodDescriptor<String, Integer> METHOD =
      MethodDescriptor.<String, Integer>newBuilder()
          .setType(MethodDescriptor.MethodType.UNKNOWN)
          .setFullMethodName("Waiter/serve")
          .setRequestMarshaller(STRING_MARSHALLER)
          .setResponseMarshaller(INTEGER_MARSHALLER)
          .build();
  private static final MethodDescriptor<String, Integer> GENERATED_METHOD =
      METHOD.toBuilder()
          .setSampledToLocalTracing(true)
          .build();
  private static final Context.Key<String> SERVER_ONLY = Context.key("serverOnly");
  private static final Context.Key<String> SERVER_TRACER_ADDED_KEY = Context.key("tracer-added");
  private static final Context.CancellableContext SERVER_CONTEXT =
      Context.ROOT.withValue(SERVER_ONLY, "yes").withCancellation();
  private static final FakeClock.TaskFilter CONTEXT_CLOSER_TASK_FILTER =
      new FakeClock.TaskFilter() {
        @Override
        public boolean shouldAccept(Runnable runnable) {
          return runnable instanceof ServerImpl.ContextCloser;
        }
      };
  private static final String AUTHORITY = "some_authority";

  private static final class MethodNameCapturingTracer extends ServerStreamTracer
      implements StatsTraceContext.ServerCallMethodListener {
    @Nullable private ServerCallInfo<?, ?> serverCallInfo;
    @Nullable private String recordedMethodName;
    @Nullable private String resolvedMethodName;
    private boolean streamClosed;

    @Override
    public synchronized void serverCallMethodResolved(MethodDescriptor<?, ?> method) {
      resolvedMethodName =
          recordMethodName(method.isSampledToLocalTracing(), method.getFullMethodName());
    }

    @Override
    public synchronized void streamClosed(Status status) {
      streamClosed = true;
      if (serverCallInfo != null) {
        recordedMethodName =
            recordMethodName(
                serverCallInfo.getMethodDescriptor().isSampledToLocalTracing(),
                serverCallInfo.getMethodDescriptor().getFullMethodName());
      } else if (resolvedMethodName != null) {
        recordedMethodName = resolvedMethodName;
      } else {
        recordedMethodName = "other";
      }
    }

    @Override
    public synchronized void serverCallStarted(ServerCallInfo<?, ?> callInfo) {