Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(653)

Side by Side Diff: net/quic/quic_chromium_client_stream_test.cc

Issue 1581283002: Rename chromium-specific QUIC classes. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: add missing Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/quic/quic_chromium_client_stream.cc ('k') | net/quic/quic_chromium_connection_helper.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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
OLDNEW
« no previous file with comments | « net/quic/quic_chromium_client_stream.cc ('k') | net/quic/quic_chromium_connection_helper.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698