diff options
Diffstat (limited to 'net/quic/quic_spdy_stream_test.cc')
| -rw-r--r-- | net/quic/quic_spdy_stream_test.cc | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/net/quic/quic_spdy_stream_test.cc b/net/quic/quic_spdy_stream_test.cc index 4f71cb7..1e7abc6 100644 --- a/net/quic/quic_spdy_stream_test.cc +++ b/net/quic/quic_spdy_stream_test.cc @@ -631,6 +631,7 @@ TEST_P(QuicSpdyStreamTest, ReceivingTrailers) { trailers_block["key1"] = "value1"; trailers_block["key2"] = "value2"; trailers_block["key3"] = "value3"; + trailers_block[kFinalOffsetHeaderKey] = "0"; string trailers = SpdyUtils::SerializeUncompressedHeaders(trailers_block); stream_->OnStreamHeaders(trailers); stream_->OnStreamHeadersComplete(/*fin=*/true, trailers.size()); @@ -709,6 +710,44 @@ TEST_P(QuicSpdyStreamTest, ReceivingTrailersAfterBodyWithFin) { stream_->OnStreamHeadersComplete(/*fin=*/true, trailers.size()); } +TEST_P(QuicSpdyStreamTest, ReceivingTrailersWithOffset) { + // Test that when receiving trailing headers with an offset before response + // body, stream is closed at the right offset. + Initialize(kShouldProcessData); + + // Receive initial headers. + string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); + stream_->OnStreamHeaders(headers); + stream_->OnStreamHeadersComplete(false, headers.size()); + stream_->MarkHeadersConsumed(stream_->decompressed_headers().size()); + + const string body = "this is the body"; + // Receive trailing headers. + SpdyHeaderBlock trailers_block; + trailers_block["key1"] = "value1"; + trailers_block["key2"] = "value2"; + trailers_block["key3"] = "value3"; + trailers_block[kFinalOffsetHeaderKey] = base::IntToString(body.size()); + string trailers = SpdyUtils::SerializeUncompressedHeaders(trailers_block); + stream_->OnStreamHeaders(trailers); + stream_->OnStreamHeadersComplete(/*fin=*/true, trailers.size()); + + // The trailers should be decompressed, and readable from the stream. + EXPECT_TRUE(stream_->trailers_decompressed()); + const string decompressed_trailers = stream_->decompressed_trailers(); + EXPECT_EQ(trailers, decompressed_trailers); + // Consuming the trailers erases them from the stream. + stream_->MarkTrailersConsumed(decompressed_trailers.size()); + EXPECT_EQ("", stream_->decompressed_trailers()); + + EXPECT_FALSE(stream_->IsDoneReading()); + // Receive and consume body. + QuicStreamFrame frame(kClientDataStreamId1, /*fin=*/false, 0, body); + stream_->OnStreamFrame(frame); + EXPECT_EQ(body, stream_->data()); + EXPECT_TRUE(stream_->IsDoneReading()); +} + TEST_P(QuicSpdyStreamTest, ClosingStreamWithNoTrailers) { // Verify that a stream receiving headers, body, and no trailers is correctly // marked as done reading on consumption of headers and body. |