S2AStub uses withWaitForReady.

grpc · Sep 9, 2024 · 655f0bd · 655f0bd
1 parent da330cd
commit 655f0bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/s2a/src/main/java/io/grpc/s2a/handshaker/S2AStub.java b/s2a/src/main/java/io/grpc/s2a/handshaker/S2AStub.java
@@ -142,7 +142,10 @@ public void close() {
   private void createWriterIfNull() {
     if (writer == null) {
       writer =
-          serviceStub.withDeadlineAfter(HANDSHAKE_RPC_DEADLINE_SECS, SECONDS).setUpSession(reader);
+          serviceStub
+              .withWaitForReady()
+              .withDeadlineAfter(HANDSHAKE_RPC_DEADLINE_SECS, SECONDS)
+              .setUpSession(reader);
     }
   }
 

diff --git a/s2a/src/test/java/io/grpc/s2a/handshaker/S2AStubTest.java b/s2a/src/test/java/io/grpc/s2a/handshaker/S2AStubTest.java
@@ -62,7 +62,7 @@ public void send_receiveOkStatus() throws Exception {
     IOException expected =
         assertThrows(IOException.class, () -> newStub.send(SessionReq.getDefaultInstance()));
 
-    assertThat(expected).hasMessageThat().contains("UNAVAILABLE");
+    assertThat(expected).hasMessageThat().contains("DEADLINE_EXCEEDED");
   }
 
   @Test