Class AbstractStubTest

stub/src/test/java/io/grpc/stub/AbstractStubTest.java:36–83

import org.junit.runners.JUnit4;

@RunWith(JUnit4.class)
public class AbstractStubTest extends BaseAbstractStubTest<NoopStub> {

  @Override
  NoopStub create(Channel channel, CallOptions callOptions) {
    return new NoopStub(channel, callOptions);
  }

  @Test
  public void defaultCallOptions() {
    NoopStub stub = NoopStub.newStub(new StubFactory<NoopStub>() {
      @Override
      public NoopStub newStub(Channel channel, CallOptions callOptions) {
        return create(channel, callOptions);
      }
    }, channel, CallOptions.DEFAULT);

    assertThat(stub.getCallOptions().getOption(ClientCalls.STUB_TYPE_OPTION))
        .isNull();
  }

  @Test
  @IgnoreJRERequirement
  public void testDuration() {
    NoopStub stub = NoopStub.newStub(new StubFactory<NoopStub>() {
      @Override
      public NoopStub newStub(Channel channel, CallOptions callOptions) {
        return create(channel, callOptions);
      }
    }, channel, CallOptions.DEFAULT);
    NoopStub stubInstance = stub.withDeadlineAfter(Duration.ofMinutes(1L));
    Deadline actual = stubInstance.getCallOptions().getDeadline();
    Deadline expected = Deadline.after(1, MINUTES);

    assertAbout(deadline()).that(actual).isWithin(10, MILLISECONDS).of(expected);
  }

  class NoopStub extends AbstractStub<NoopStub> {
    NoopStub(Channel channel, CallOptions options) {
      super(channel, options);
    }

    @Override
    protected NoopStub build(Channel channel, CallOptions callOptions) {
      return new NoopStub(channel, callOptions);
    }
  }
}