| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_reliable_client_stream.h" | 5 #include "net/quic/quic_chromium_client_stream.h" |
| 6 | 6 |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "net/base/net_errors.h" | 8 #include "net/base/net_errors.h" |
| 9 #include "net/base/test_completion_callback.h" | 9 #include "net/base/test_completion_callback.h" |
| 10 #include "net/quic/quic_chromium_client_session.h" | 10 #include "net/quic/quic_chromium_client_session.h" |
| 11 #include "net/quic/quic_utils.h" | 11 #include "net/quic/quic_utils.h" |
| 12 #include "net/quic/spdy_utils.h" | 12 #include "net/quic/spdy_utils.h" |
| 13 #include "net/quic/test_tools/crypto_test_utils.h" | 13 #include "net/quic/test_tools/crypto_test_utils.h" |
| 14 #include "net/quic/test_tools/quic_test_utils.h" | 14 #include "net/quic/test_tools/quic_test_utils.h" |
| 15 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 16 #include "testing/gmock_mutant.h" | 16 #include "testing/gmock_mutant.h" |
| 17 | 17 |
| 18 using testing::AnyNumber; | 18 using testing::AnyNumber; |
| 19 using testing::CreateFunctor; | 19 using testing::CreateFunctor; |
| 20 using testing::Invoke; | 20 using testing::Invoke; |
| 21 using testing::Return; | 21 using testing::Return; |
| 22 using testing::StrEq; | 22 using testing::StrEq; |
| 23 using testing::_; | 23 using testing::_; |
| 24 | 24 |
| 25 namespace net { | 25 namespace net { |
| 26 namespace test { | 26 namespace test { |
| 27 namespace { | 27 namespace { |
| 28 | 28 |
| 29 const QuicStreamId kTestStreamId = 5u; | 29 const QuicStreamId kTestStreamId = 5u; |
| 30 | 30 |
| 31 class MockDelegate : public QuicReliableClientStream::Delegate { | 31 class MockDelegate : public QuicChromiumClientStream::Delegate { |
| 32 public: | 32 public: |
| 33 MockDelegate() {} | 33 MockDelegate() {} |
| 34 | 34 |
| 35 MOCK_METHOD0(OnSendData, int()); | 35 MOCK_METHOD0(OnSendData, int()); |
| 36 MOCK_METHOD2(OnSendDataComplete, int(int, bool*)); | 36 MOCK_METHOD2(OnSendDataComplete, int(int, bool*)); |
| 37 MOCK_METHOD2(OnHeadersAvailable, void(const SpdyHeaderBlock&, size_t)); | 37 MOCK_METHOD2(OnHeadersAvailable, void(const SpdyHeaderBlock&, size_t)); |
| 38 MOCK_METHOD2(OnDataReceived, int(const char*, int)); | 38 MOCK_METHOD2(OnDataReceived, int(const char*, int)); |
| 39 MOCK_METHOD0(OnDataAvailable, void()); | 39 MOCK_METHOD0(OnDataAvailable, void()); |
| 40 MOCK_METHOD1(OnClose, void(QuicErrorCode)); | 40 MOCK_METHOD1(OnClose, void(QuicErrorCode)); |
| 41 MOCK_METHOD1(OnError, void(int)); | 41 MOCK_METHOD1(OnError, void(int)); |
| 42 MOCK_METHOD0(HasSendHeadersComplete, bool()); | 42 MOCK_METHOD0(HasSendHeadersComplete, bool()); |
| 43 | 43 |
| 44 private: | 44 private: |
| 45 DISALLOW_COPY_AND_ASSIGN(MockDelegate); | 45 DISALLOW_COPY_AND_ASSIGN(MockDelegate); |
| 46 }; | 46 }; |
| 47 | 47 |
| 48 class QuicReliableClientStreamTest | 48 class QuicChromiumClientStreamTest |
| 49 : public ::testing::TestWithParam<QuicVersion> { | 49 : public ::testing::TestWithParam<QuicVersion> { |
| 50 public: | 50 public: |
| 51 QuicReliableClientStreamTest() | 51 QuicChromiumClientStreamTest() |
| 52 : crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), | 52 : crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), |
| 53 session_(new MockConnection(&helper_, | 53 session_(new MockConnection(&helper_, |
| 54 Perspective::IS_CLIENT, | 54 Perspective::IS_CLIENT, |
| 55 SupportedVersions(GetParam()))) { | 55 SupportedVersions(GetParam()))) { |
| 56 stream_ = | 56 stream_ = |
| 57 new QuicReliableClientStream(kTestStreamId, &session_, BoundNetLog()); | 57 new QuicChromiumClientStream(kTestStreamId, &session_, BoundNetLog()); |
| 58 session_.ActivateStream(stream_); | 58 session_.ActivateStream(stream_); |
| 59 stream_->SetDelegate(&delegate_); | 59 stream_->SetDelegate(&delegate_); |
| 60 } | 60 } |
| 61 | 61 |
| 62 void InitializeHeaders() { | 62 void InitializeHeaders() { |
| 63 headers_[":host"] = "www.google.com"; | 63 headers_[":host"] = "www.google.com"; |
| 64 headers_[":path"] = "/index.hml"; | 64 headers_[":path"] = "/index.hml"; |
| 65 headers_[":scheme"] = "https"; | 65 headers_[":scheme"] = "https"; |
| 66 headers_["cookie"] = | 66 headers_["cookie"] = |
| 67 "__utma=208381060.1228362404.1372200928.1372200928.1372200928.1; " | 67 "__utma=208381060.1228362404.1372200928.1372200928.1372200928.1; " |
| (...skipping 26 matching lines...) Expand all Loading... |
| 94 EXPECT_EQ(static_cast<int>(expected_data.length()), | 94 EXPECT_EQ(static_cast<int>(expected_data.length()), |
| 95 stream_->Read(buffer.get(), expected_data.length() + 1)); | 95 stream_->Read(buffer.get(), expected_data.length() + 1)); |
| 96 EXPECT_EQ(expected_data, | 96 EXPECT_EQ(expected_data, |
| 97 StringPiece(buffer->data(), expected_data.length())); | 97 StringPiece(buffer->data(), expected_data.length())); |
| 98 } | 98 } |
| 99 | 99 |
| 100 QuicCryptoClientConfig crypto_config_; | 100 QuicCryptoClientConfig crypto_config_; |
| 101 testing::StrictMock<MockDelegate> delegate_; | 101 testing::StrictMock<MockDelegate> delegate_; |
| 102 MockConnectionHelper helper_; | 102 MockConnectionHelper helper_; |
| 103 MockQuicSpdySession session_; | 103 MockQuicSpdySession session_; |
| 104 QuicReliableClientStream* stream_; | 104 QuicChromiumClientStream* stream_; |
| 105 SpdyHeaderBlock headers_; | 105 SpdyHeaderBlock headers_; |
| 106 }; | 106 }; |
| 107 | 107 |
| 108 INSTANTIATE_TEST_CASE_P(Version, | 108 INSTANTIATE_TEST_CASE_P(Version, |
| 109 QuicReliableClientStreamTest, | 109 QuicChromiumClientStreamTest, |
| 110 ::testing::ValuesIn(QuicSupportedVersions())); | 110 ::testing::ValuesIn(QuicSupportedVersions())); |
| 111 | 111 |
| 112 TEST_P(QuicReliableClientStreamTest, OnFinRead) { | 112 TEST_P(QuicChromiumClientStreamTest, OnFinRead) { |
| 113 InitializeHeaders(); | 113 InitializeHeaders(); |
| 114 std::string uncompressed_headers = | 114 std::string uncompressed_headers = |
| 115 SpdyUtils::SerializeUncompressedHeaders(headers_); | 115 SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 116 QuicStreamOffset offset = 0; | 116 QuicStreamOffset offset = 0; |
| 117 stream_->OnStreamHeaders(uncompressed_headers); | 117 stream_->OnStreamHeaders(uncompressed_headers); |
| 118 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 118 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
| 119 | 119 |
| 120 EXPECT_CALL(delegate_, | 120 EXPECT_CALL(delegate_, |
| 121 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 121 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
| 122 base::MessageLoop::current()->RunUntilIdle(); | 122 base::MessageLoop::current()->RunUntilIdle(); |
| 123 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 123 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
| 124 | 124 |
| 125 QuicStreamFrame frame2(kTestStreamId, true, offset, StringPiece()); | 125 QuicStreamFrame frame2(kTestStreamId, true, offset, StringPiece()); |
| 126 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 126 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 127 stream_->OnStreamFrame(frame2); | 127 stream_->OnStreamFrame(frame2); |
| 128 } | 128 } |
| 129 | 129 |
| 130 TEST_P(QuicReliableClientStreamTest, OnDataAvailableBeforeHeaders) { | 130 TEST_P(QuicChromiumClientStreamTest, OnDataAvailableBeforeHeaders) { |
| 131 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 131 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 132 | 132 |
| 133 EXPECT_CALL(delegate_, OnDataAvailable()).Times(0); | 133 EXPECT_CALL(delegate_, OnDataAvailable()).Times(0); |
| 134 stream_->OnDataAvailable(); | 134 stream_->OnDataAvailable(); |
| 135 } | 135 } |
| 136 | 136 |
| 137 TEST_P(QuicReliableClientStreamTest, OnDataAvailable) { | 137 TEST_P(QuicChromiumClientStreamTest, OnDataAvailable) { |
| 138 InitializeHeaders(); | 138 InitializeHeaders(); |
| 139 std::string uncompressed_headers = | 139 std::string uncompressed_headers = |
| 140 SpdyUtils::SerializeUncompressedHeaders(headers_); | 140 SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 141 stream_->OnStreamHeaders(uncompressed_headers); | 141 stream_->OnStreamHeaders(uncompressed_headers); |
| 142 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 142 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
| 143 | 143 |
| 144 EXPECT_CALL(delegate_, | 144 EXPECT_CALL(delegate_, |
| 145 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 145 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
| 146 base::MessageLoop::current()->RunUntilIdle(); | 146 base::MessageLoop::current()->RunUntilIdle(); |
| 147 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 147 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
| 148 | 148 |
| 149 const char data[] = "hello world!"; | 149 const char data[] = "hello world!"; |
| 150 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, | 150 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, |
| 151 /*offset=*/0, data)); | 151 /*offset=*/0, data)); |
| 152 | 152 |
| 153 EXPECT_CALL(delegate_, OnDataAvailable()) | 153 EXPECT_CALL(delegate_, OnDataAvailable()) |
| 154 .WillOnce(testing::Invoke( | 154 .WillOnce(testing::Invoke( |
| 155 CreateFunctor(this, &QuicReliableClientStreamTest::ReadData, | 155 CreateFunctor(this, &QuicChromiumClientStreamTest::ReadData, |
| 156 StringPiece(data, arraysize(data) - 1)))); | 156 StringPiece(data, arraysize(data) - 1)))); |
| 157 base::MessageLoop::current()->RunUntilIdle(); | 157 base::MessageLoop::current()->RunUntilIdle(); |
| 158 | 158 |
| 159 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 159 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 160 } | 160 } |
| 161 | 161 |
| 162 TEST_P(QuicReliableClientStreamTest, ProcessHeadersWithError) { | 162 TEST_P(QuicChromiumClientStreamTest, ProcessHeadersWithError) { |
| 163 std::string bad_headers = "..."; | 163 std::string bad_headers = "..."; |
| 164 stream_->OnStreamHeaders(StringPiece(bad_headers)); | 164 stream_->OnStreamHeaders(StringPiece(bad_headers)); |
| 165 stream_->OnStreamHeadersComplete(false, bad_headers.length()); | 165 stream_->OnStreamHeadersComplete(false, bad_headers.length()); |
| 166 | 166 |
| 167 EXPECT_CALL(session_, | 167 EXPECT_CALL(session_, |
| 168 SendRstStream(kTestStreamId, QUIC_BAD_APPLICATION_PAYLOAD, 0)); | 168 SendRstStream(kTestStreamId, QUIC_BAD_APPLICATION_PAYLOAD, 0)); |
| 169 base::MessageLoop::current()->RunUntilIdle(); | 169 base::MessageLoop::current()->RunUntilIdle(); |
| 170 | 170 |
| 171 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 171 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 172 } | 172 } |
| 173 | 173 |
| 174 TEST_P(QuicReliableClientStreamTest, OnDataAvailableWithError) { | 174 TEST_P(QuicChromiumClientStreamTest, OnDataAvailableWithError) { |
| 175 InitializeHeaders(); | 175 InitializeHeaders(); |
| 176 std::string uncompressed_headers = | 176 std::string uncompressed_headers = |
| 177 SpdyUtils::SerializeUncompressedHeaders(headers_); | 177 SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 178 stream_->OnStreamHeaders(uncompressed_headers); | 178 stream_->OnStreamHeaders(uncompressed_headers); |
| 179 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 179 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
| 180 | 180 |
| 181 EXPECT_CALL(delegate_, | 181 EXPECT_CALL(delegate_, |
| 182 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 182 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
| 183 base::MessageLoop::current()->RunUntilIdle(); | 183 base::MessageLoop::current()->RunUntilIdle(); |
| 184 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 184 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
| 185 | 185 |
| 186 const char data[] = "hello world!"; | 186 const char data[] = "hello world!"; |
| 187 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, | 187 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, |
| 188 /*offset=*/0, data)); | 188 /*offset=*/0, data)); |
| 189 EXPECT_CALL(delegate_, OnDataAvailable()) | 189 EXPECT_CALL(delegate_, OnDataAvailable()) |
| 190 .WillOnce(testing::Invoke(CreateFunctor( | 190 .WillOnce(testing::Invoke(CreateFunctor( |
| 191 stream_, &QuicReliableClientStream::Reset, QUIC_STREAM_CANCELLED))); | 191 stream_, &QuicChromiumClientStream::Reset, QUIC_STREAM_CANCELLED))); |
| 192 base::MessageLoop::current()->RunUntilIdle(); | 192 base::MessageLoop::current()->RunUntilIdle(); |
| 193 | 193 |
| 194 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 194 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 195 } | 195 } |
| 196 | 196 |
| 197 TEST_P(QuicReliableClientStreamTest, OnError) { | 197 TEST_P(QuicChromiumClientStreamTest, OnError) { |
| 198 EXPECT_CALL(delegate_, OnError(ERR_INTERNET_DISCONNECTED)); | 198 EXPECT_CALL(delegate_, OnError(ERR_INTERNET_DISCONNECTED)); |
| 199 | 199 |
| 200 stream_->OnError(ERR_INTERNET_DISCONNECTED); | 200 stream_->OnError(ERR_INTERNET_DISCONNECTED); |
| 201 EXPECT_FALSE(stream_->GetDelegate()); | 201 EXPECT_FALSE(stream_->GetDelegate()); |
| 202 } | 202 } |
| 203 | 203 |
| 204 TEST_P(QuicReliableClientStreamTest, WriteStreamData) { | 204 TEST_P(QuicChromiumClientStreamTest, WriteStreamData) { |
| 205 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 205 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 206 | 206 |
| 207 const char kData1[] = "hello world"; | 207 const char kData1[] = "hello world"; |
| 208 const size_t kDataLen = arraysize(kData1); | 208 const size_t kDataLen = arraysize(kData1); |
| 209 | 209 |
| 210 // All data written. | 210 // All data written. |
| 211 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 211 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
| 212 .WillOnce(Return(QuicConsumedData(kDataLen, true))); | 212 .WillOnce(Return(QuicConsumedData(kDataLen, true))); |
| 213 TestCompletionCallback callback; | 213 TestCompletionCallback callback; |
| 214 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), | 214 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), |
| 215 true, callback.callback())); | 215 true, callback.callback())); |
| 216 } | 216 } |
| 217 | 217 |
| 218 TEST_P(QuicReliableClientStreamTest, WriteStreamDataAsync) { | 218 TEST_P(QuicChromiumClientStreamTest, WriteStreamDataAsync) { |
| 219 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); | 219 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); |
| 220 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 220 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
| 221 | 221 |
| 222 const char kData1[] = "hello world"; | 222 const char kData1[] = "hello world"; |
| 223 const size_t kDataLen = arraysize(kData1); | 223 const size_t kDataLen = arraysize(kData1); |
| 224 | 224 |
| 225 // No data written. | 225 // No data written. |
| 226 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 226 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
| 227 .WillOnce(Return(QuicConsumedData(0, false))); | 227 .WillOnce(Return(QuicConsumedData(0, false))); |
| 228 TestCompletionCallback callback; | 228 TestCompletionCallback callback; |
| 229 EXPECT_EQ(ERR_IO_PENDING, | 229 EXPECT_EQ(ERR_IO_PENDING, |
| 230 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), true, | 230 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), true, |
| 231 callback.callback())); | 231 callback.callback())); |
| 232 ASSERT_FALSE(callback.have_result()); | 232 ASSERT_FALSE(callback.have_result()); |
| 233 | 233 |
| 234 // All data written. | 234 // All data written. |
| 235 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 235 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
| 236 .WillOnce(Return(QuicConsumedData(kDataLen, true))); | 236 .WillOnce(Return(QuicConsumedData(kDataLen, true))); |
| 237 stream_->OnCanWrite(); | 237 stream_->OnCanWrite(); |
| 238 ASSERT_TRUE(callback.have_result()); | 238 ASSERT_TRUE(callback.have_result()); |
| 239 EXPECT_EQ(OK, callback.WaitForResult()); | 239 EXPECT_EQ(OK, callback.WaitForResult()); |
| 240 } | 240 } |
| 241 | 241 |
| 242 } // namespace | 242 } // namespace |
| 243 } // namespace test | 243 } // namespace test |
| 244 } // namespace net | 244 } // namespace net |
| OLD | NEW |