From 2e3bdf55f82dc683ca1c9907a7bd1a915ee41d71 Mon Sep 17 00:00:00 2001
From: iamqizhao <toqizhao@gmail.com>
Date: Fri, 29 Jul 2016 17:52:30 -0700
Subject: [PATCH] make it faster
---
transport/transport_test.go | 26 ++++++++++++++------------
1 file changed, 14 insertions(+), 12 deletions(-)
diff --git a/transport/transport_test.go b/transport/transport_test.go
index 80a1b15f..7fecbeaf 100644
--- a/transport/transport_test.go
+++ b/transport/transport_test.go
@@ -111,20 +111,22 @@ func (h *testStreamHandler) handleStreamMisbehave(t *testing.T, s *Stream) {
if !ok {
t.Fatalf("Failed to convert %v to *http2Server", s.ServerTransport())
}
- var end int
- if s.Method() == "foo.Connection" {
- // Violate connection level flow control window of client but do not
- // violate any stream level windows.
- end = initialWindowSize
- } else {
- // Violate stream level flow control window of client.
- end = initialWindowSize + 1
- }
- // Violate the client side stream/connection flow control window.
var sent int
- p := make([]byte, 1)
- for sent < end {
+ p := make([]byte, http2MaxFrameLen)
+ for sent < initialWindowSize {
<-conn.writableChan
+ n := initialWindowSize - sent
+ // The last message may be smaller than http2MaxFrameLen
+ if n < http2MaxFrameLen {
+ if s.Method() == "foo.Connection" {
+ // Violate connection level flow control window of client but do not
+ // violate any stream level windows.
+ p = make([]byte, n)
+ } else {
+ // Violate stream level flow control window of client.
+ p = make([]byte, n+1)
+ }
+ }
if err := conn.framer.writeData(true, s.id, false, p); err != nil {
conn.writableChan <- 0
break