flow control fixes

1 files changed, 11 insertions, 0 deletions

diff --git a/src/core/ext/transport/chttp2/transport/flow_control.h b/src/core/ext/transport/chttp2/transport/flow_control.h
index 13a3341122..33542a31e4 100644
--- a/src/core/ext/transport/chttp2/transport/flow_control.h
+++ b/src/core/ext/transport/chttp2/transport/flow_control.h
@@ -201,6 +201,11 @@ class TransportFlowControl {
 
   FlowControlAction PeriodicUpdate(grpc_exec_ctx* exec_ctx);
 
+  void TestOnlyForceHugeWindow() {
+    announced_window_ = 1024 * 1024 * 1024;
+    remote_window_ = 1024 * 1024 * 1024;
+  }
+
  private:
   double SmoothLogBdp(grpc_exec_ctx* exec_ctx, double value);
   FlowControlAction::Urgency DeltaUrgency(int32_t value,
@@ -289,6 +294,12 @@ class StreamFlowControl {
 
   const grpc_chttp2_stream* stream() const { return s_; }
 
+  void TestOnlyForceHugeWindow() {
+    announced_window_delta_ = 1024 * 1024 * 1024;
+    local_window_delta_ = 1024 * 1024 * 1024;
+    remote_window_delta_ = 1024 * 1024 * 1024;
+  }
+
  private:
   TransportFlowControl* const tfc_;
   const grpc_chttp2_stream* const s_;