stub: add withWaitForReady

Author: dapengzhang0

stub/src/main/java/io/grpc/stub/AbstractStub.java

@@ -181,4 +181,12 @@ public final S withInterceptors(ClientInterceptor... interceptors) {
   public final S withCallCredentials(CallCredentials credentials) {
     return build(channel, callOptions.withCallCredentials(credentials));
   }
+
+  /**
+   * Returns a new stub that uses the 'wait for ready' call option.
+   */
+  @ExperimentalApi("https://github.com/grpc/grpc-java/issues/1915")
+  public final S withWaitForReady() {
+    return build(channel, callOptions.withWaitForReady());
+  }
 }

stub/src/test/java/io/grpc/stub/AbstractStubTest.java

@@ -31,19 +31,30 @@
 
 package io.grpc.stub;
 
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
 import io.grpc.CallOptions;
 import io.grpc.Channel;
+
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
 
 @RunWith(JUnit4.class)
 public class AbstractStubTest {
 
   @Mock
   Channel channel;
 
+  @Before
+  public void setup() {
+    MockitoAnnotations.initMocks(this);
+  }
+
   @Test(expected = NullPointerException.class)
   public void channelMustNotBeNull() {
     new NoopStub(null);
@@ -55,8 +66,19 @@ public void callOptionsMustNotBeNull() {
   }
 
   @Test(expected = NullPointerException.class)
-  public void callOptionsAndChannelMustNotBeNull() {
-    new NoopStub(null, null);
+  public void channelMustNotBeNull2() {
+    new NoopStub(null, CallOptions.DEFAULT);
+  }
+
+  @Test()
+  public void withWaitForReady() {
+    NoopStub stub = new NoopStub(channel);
+    CallOptions callOptions = stub.getCallOptions();
+    assertFalse(callOptions.isWaitForReady());
+
+    stub = stub.withWaitForReady();
+    callOptions = stub.getCallOptions();
+    assertTrue(callOptions.isWaitForReady());
   }
 
   class NoopStub extends AbstractStub<NoopStub> {