Add test for supplying CallOptions to transport state

grpc · Mar 21, 2024 · f0cf726 · f0cf726
1 parent 63a454d
commit f0cf726
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/core/src/test/java/io/grpc/internal/Http2ClientStreamTransportStateTest.java b/core/src/test/java/io/grpc/internal/Http2ClientStreamTransportStateTest.java
@@ -348,9 +348,21 @@ public void transportTrailersReceived_missingStatusAfterHeadersIgnoresHttpStatus
     assertEquals(Code.UNKNOWN, statusCaptor.getValue().getCode());
   }
 
+  @Test
+  public void transportStateWithOnReadyThreshold() {
+    BaseTransportState state = new BaseTransportState(transportTracer,
+        CallOptions.DEFAULT.withOnReadyThreshold(Integer.MAX_VALUE));
+    assertEquals(Integer.MAX_VALUE, state.onReadyThreshold);
+  }
+
   private static class BaseTransportState extends Http2ClientStreamTransportState {
+    private int onReadyThreshold;
+
+    public BaseTransportState(TransportTracer transportTracer, CallOptions options) {
+      super(DEFAULT_MAX_MESSAGE_SIZE, StatsTraceContext.NOOP, transportTracer, options);
+    }
     public BaseTransportState(TransportTracer transportTracer) {
-      super(DEFAULT_MAX_MESSAGE_SIZE, StatsTraceContext.NOOP, transportTracer, CallOptions.DEFAULT);
+      this(transportTracer, CallOptions.DEFAULT);
     }
 
     @Override
@@ -368,5 +380,11 @@ public void bytesRead(int processedBytes) {}
     public void runOnTransportThread(Runnable r) {
       r.run();
     }
+
+    @Override
+    void setOnReadyThreshold(int numBytes) {
+      onReadyThreshold = numBytes;
+      super.setOnReadyThreshold(numBytes);
+    }
   }
 }