diff options
Diffstat (limited to 'net/quic/quic_headers_stream_test.cc')
-rw-r--r-- | net/quic/quic_headers_stream_test.cc | 111 |
1 files changed, 53 insertions, 58 deletions
diff --git a/net/quic/quic_headers_stream_test.cc b/net/quic/quic_headers_stream_test.cc index 680bdf2..7e8e118 100644 --- a/net/quic/quic_headers_stream_test.cc +++ b/net/quic/quic_headers_stream_test.cc @@ -37,35 +37,36 @@ const bool kFins[] = {false, true}; class MockVisitor : public SpdyFramerVisitorInterface { public: MOCK_METHOD1(OnError, void(SpdyFramer* framer)); - MOCK_METHOD3(OnDataFrameHeader, void(SpdyStreamId stream_id, - size_t length, - bool fin)); - MOCK_METHOD4(OnStreamFrameData, void(SpdyStreamId stream_id, - const char* data, - size_t len, - bool fin)); + MOCK_METHOD3(OnDataFrameHeader, + void(SpdyStreamId stream_id, size_t length, bool fin)); + MOCK_METHOD4( + OnStreamFrameData, + void(SpdyStreamId stream_id, const char* data, size_t len, bool fin)); MOCK_METHOD2(OnStreamPadding, void(SpdyStreamId stream_id, size_t len)); MOCK_METHOD1(OnHeaderFrameStart, SpdyHeadersHandlerInterface*(SpdyStreamId stream_id)); MOCK_METHOD2(OnHeaderFrameEnd, void(SpdyStreamId stream_id, bool end)); - MOCK_METHOD3(OnControlFrameHeaderData, bool(SpdyStreamId stream_id, - const char* header_data, - size_t len)); - MOCK_METHOD5(OnSynStream, void(SpdyStreamId stream_id, - SpdyStreamId associated_stream_id, - SpdyPriority priority, - bool fin, - bool unidirectional)); + MOCK_METHOD3(OnControlFrameHeaderData, + bool(SpdyStreamId stream_id, + const char* header_data, + size_t len)); + MOCK_METHOD5(OnSynStream, + void(SpdyStreamId stream_id, + SpdyStreamId associated_stream_id, + SpdyPriority priority, + bool fin, + bool unidirectional)); MOCK_METHOD2(OnSynReply, void(SpdyStreamId stream_id, bool fin)); - MOCK_METHOD2(OnRstStream, void(SpdyStreamId stream_id, - SpdyRstStreamStatus status)); + MOCK_METHOD2(OnRstStream, + void(SpdyStreamId stream_id, SpdyRstStreamStatus status)); MOCK_METHOD1(OnSettings, void(bool clear_persisted)); MOCK_METHOD3(OnSetting, void(SpdySettingsIds id, uint8 flags, uint32 value)); MOCK_METHOD0(OnSettingsAck, void()); MOCK_METHOD0(OnSettingsEnd, void()); MOCK_METHOD2(OnPing, void(SpdyPingId unique_id, bool is_ack)); - MOCK_METHOD2(OnGoAway, void(SpdyStreamId last_accepted_stream_id, - SpdyGoAwayStatus status)); + MOCK_METHOD2(OnGoAway, + void(SpdyStreamId last_accepted_stream_id, + SpdyGoAwayStatus status)); MOCK_METHOD7(OnHeaders, void(SpdyStreamId stream_id, bool has_priority, @@ -77,9 +78,10 @@ class MockVisitor : public SpdyFramerVisitorInterface { MOCK_METHOD2(OnWindowUpdate, void(SpdyStreamId stream_id, int delta_window_size)); MOCK_METHOD1(OnBlocked, void(SpdyStreamId stream_id)); - MOCK_METHOD3(OnPushPromise, void(SpdyStreamId stream_id, - SpdyStreamId promised_stream_id, - bool end)); + MOCK_METHOD3(OnPushPromise, + void(SpdyStreamId stream_id, + SpdyStreamId promised_stream_id, + bool end)); MOCK_METHOD2(OnContinuation, void(SpdyStreamId stream_id, bool end)); MOCK_METHOD3(OnAltSvc, void(SpdyStreamId stream_id, @@ -127,7 +129,7 @@ class QuicHeadersStreamTest : public ::testing::TestWithParam<TestParams> { framer_(HTTP2), stream_frame_(kHeadersStreamId, /*fin=*/false, /*offset=*/0, ""), next_promised_stream_id_(2) { - headers_[":version"] = "HTTP/1.1"; + headers_[":version"] = "HTTP/1.1"; headers_[":status"] = "200 Ok"; headers_["content-length"] = "11"; framer_.set_visitor(&visitor_); @@ -140,7 +142,7 @@ class QuicHeadersStreamTest : public ::testing::TestWithParam<TestParams> { QuicConsumedData SaveIov(const QuicIOVector& data) { const iovec* iov = data.iov; int count = data.iov_count; - for (int i = 0 ; i < count; ++i) { + for (int i = 0; i < count; ++i) { saved_data_.append(static_cast<char*>(iov[i].iov_base), iov[i].iov_len); } return QuicConsumedData(saved_data_.length(), false); @@ -161,8 +163,7 @@ class QuicHeadersStreamTest : public ::testing::TestWithParam<TestParams> { WriteHeadersAndCheckData(stream_id, fin, priority, SYN_STREAM); } - void WriteHeadersAndExpectSynReply(QuicStreamId stream_id, - bool fin) { + void WriteHeadersAndExpectSynReply(QuicStreamId stream_id, bool fin) { WriteHeadersAndCheckData(stream_id, fin, 0, SYN_REPLY); } @@ -177,10 +178,10 @@ class QuicHeadersStreamTest : public ::testing::TestWithParam<TestParams> { // Parse the outgoing data and check that it matches was was written. if (type == SYN_STREAM) { - EXPECT_CALL(visitor_, OnHeaders(stream_id, kHasPriority, priority, - /*parent_stream_id=*/0, - /*exclusive=*/false, - fin, kFrameComplete)); + EXPECT_CALL(visitor_, + OnHeaders(stream_id, kHasPriority, priority, + /*parent_stream_id=*/0, + /*exclusive=*/false, fin, kFrameComplete)); } else { EXPECT_CALL(visitor_, OnHeaders(stream_id, !kHasPriority, @@ -220,9 +221,7 @@ class QuicHeadersStreamTest : public ::testing::TestWithParam<TestParams> { return versions; } - void CloseConnection() { - QuicConnectionPeer::CloseConnection(connection_); - } + void CloseConnection() { QuicConnectionPeer::CloseConnection(connection_); } QuicStreamId NextPromisedStreamId() { return next_promised_stream_id_++; } @@ -323,9 +322,8 @@ TEST_P(QuicHeadersStreamTest, ProcessRawData) { frame.reset(framer_.SerializeFrame(headers_frame)); } EXPECT_CALL(session_, OnStreamHeaders(stream_id, _)) - .WillRepeatedly(WithArgs<1>( - Invoke(this, - &QuicHeadersStreamTest::SaveHeaderDataStringPiece))); + .WillRepeatedly(WithArgs<1>(Invoke( + this, &QuicHeadersStreamTest::SaveHeaderDataStringPiece))); EXPECT_CALL(session_, OnStreamHeadersComplete(stream_id, fin, frame->size())); stream_frame_.frame_buffer = frame->data(); @@ -474,8 +472,8 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdyDataFrame) { EXPECT_CALL(*connection_, SendConnectionCloseWithDetails(QUIC_INVALID_HEADERS_STREAM_DATA, "SPDY DATA frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); @@ -485,11 +483,10 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdyRstStreamFrame) { SpdyRstStreamIR data(2, RST_STREAM_PROTOCOL_ERROR); scoped_ptr<SpdySerializedFrame> frame(framer_.SerializeFrame(data)); EXPECT_CALL(*connection_, - SendConnectionCloseWithDetails( - QUIC_INVALID_HEADERS_STREAM_DATA, - "SPDY RST_STREAM frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + SendConnectionCloseWithDetails(QUIC_INVALID_HEADERS_STREAM_DATA, + "SPDY RST_STREAM frame received.")) + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); @@ -500,11 +497,10 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdySettingsFrame) { data.AddSetting(SETTINGS_HEADER_TABLE_SIZE, true, true, 0); scoped_ptr<SpdySerializedFrame> frame(framer_.SerializeFrame(data)); EXPECT_CALL(*connection_, - SendConnectionCloseWithDetails( - QUIC_INVALID_HEADERS_STREAM_DATA, - "SPDY SETTINGS frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + SendConnectionCloseWithDetails(QUIC_INVALID_HEADERS_STREAM_DATA, + "SPDY SETTINGS frame received.")) + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); @@ -516,8 +512,8 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdyPingFrame) { EXPECT_CALL(*connection_, SendConnectionCloseWithDetails(QUIC_INVALID_HEADERS_STREAM_DATA, "SPDY PING frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); @@ -529,8 +525,8 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdyGoAwayFrame) { EXPECT_CALL(*connection_, SendConnectionCloseWithDetails(QUIC_INVALID_HEADERS_STREAM_DATA, "SPDY GOAWAY frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); @@ -539,12 +535,11 @@ TEST_P(QuicHeadersStreamTest, ProcessSpdyGoAwayFrame) { TEST_P(QuicHeadersStreamTest, ProcessSpdyWindowUpdateFrame) { SpdyWindowUpdateIR data(1, 1); scoped_ptr<SpdySerializedFrame> frame(framer_.SerializeFrame(data)); - EXPECT_CALL(*connection_, - SendConnectionCloseWithDetails( - QUIC_INVALID_HEADERS_STREAM_DATA, - "SPDY WINDOW_UPDATE frame received.")) - .WillOnce(InvokeWithoutArgs(this, - &QuicHeadersStreamTest::CloseConnection)); + EXPECT_CALL(*connection_, SendConnectionCloseWithDetails( + QUIC_INVALID_HEADERS_STREAM_DATA, + "SPDY WINDOW_UPDATE frame received.")) + .WillOnce( + InvokeWithoutArgs(this, &QuicHeadersStreamTest::CloseConnection)); stream_frame_.frame_buffer = frame->data(); stream_frame_.frame_length = frame->size(); headers_stream_->OnStreamFrame(stream_frame_); |