From 8c40c3f768773bc6fced1962e7d05967f094c56e Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Tue, 28 May 2019 15:56:02 +0200
Subject: [PATCH] improve logging of STREAMS_BLOCKED frames

---
 internal/wire/log.go      | 7 +++++++
 internal/wire/log_test.go | 9 +++++++++
 2 files changed, 16 insertions(+)

diff --git a/internal/wire/log.go b/internal/wire/log.go
index b6d626a2..790523d7 100644
--- a/internal/wire/log.go
+++ b/internal/wire/log.go
@@ -40,6 +40,13 @@ func LogFrame(logger utils.Logger, frame Frame, sent bool) {
 		case protocol.StreamTypeBidi:
 			logger.Debugf("\t%s &wire.MaxStreamsFrame{Type: bidi, MaxStreams: %d}", dir, f.MaxStreams)
 		}
+	case *StreamsBlockedFrame:
+		switch f.Type {
+		case protocol.StreamTypeUni:
+			logger.Debugf("\t%s &wire.StreamsBlockedFrame{Type: uni, MaxStreams: %d}", dir, f.StreamLimit)
+		case protocol.StreamTypeBidi:
+			logger.Debugf("\t%s &wire.StreamsBlockedFrame{Type: bidi, MaxStreams: %d}", dir, f.StreamLimit)
+		}
 	case *NewConnectionIDFrame:
 		logger.Debugf("\t%s &wire.NewConnectionIDFrame{SequenceNumber: %d, ConnectionID: %s, StatelessResetToken: %#x}", dir, f.SequenceNumber, f.ConnectionID, f.StatelessResetToken)
 	case *NewTokenFrame:
diff --git a/internal/wire/log_test.go b/internal/wire/log_test.go
index 804ab8f4..f611e5f4 100644
--- a/internal/wire/log_test.go
+++ b/internal/wire/log_test.go
@@ -96,6 +96,15 @@ var _ = Describe("Frame logging", func() {
 		Expect(buf.String()).To(ContainSubstring("\t<- &wire.MaxStreamsFrame{Type: bidi, MaxStreams: 42}\n"))
 	})
 
+	It("logs STREAMS_BLOCKED frames", func() {
+		frame := &StreamsBlockedFrame{
+			Type:        protocol.StreamTypeBidi,
+			StreamLimit: 42,
+		}
+		LogFrame(logger, frame, false)
+		Expect(buf.String()).To(ContainSubstring("\t<- &wire.StreamsBlockedFrame{Type: bidi, MaxStreams: 42}\n"))
+	})
+
 	It("logs NEW_CONNECTION_ID frames", func() {
 		LogFrame(logger, &NewConnectionIDFrame{
 			SequenceNumber:      42,