OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/core/quic_client_promised_info.h" | 5 #include "net/quic/core/quic_client_promised_info.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/scoped_ptr.h" | 10 #include "base/scoped_ptr.h" |
(...skipping 253 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
264 } | 264 } |
265 | 265 |
266 TEST_F(QuicClientPromisedInfoTest, PushPromiseVaryWaits) { | 266 TEST_F(QuicClientPromisedInfoTest, PushPromiseVaryWaits) { |
267 ReceivePromise(promise_id_); | 267 ReceivePromise(promise_id_); |
268 | 268 |
269 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); | 269 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); |
270 ASSERT_NE(promised, nullptr); | 270 ASSERT_NE(promised, nullptr); |
271 | 271 |
272 // Now initiate rendezvous. | 272 // Now initiate rendezvous. |
273 PushPromiseDelegate delegate(/*match=*/true); | 273 PushPromiseDelegate delegate(/*match=*/true); |
274 promised->HandleClientRequest(std::move(client_request_), &delegate); | 274 promised->HandleClientRequest(client_request_, &delegate); |
275 | 275 |
276 // Promise is still there, waiting for response. | 276 // Promise is still there, waiting for response. |
277 EXPECT_NE(session_.GetPromisedById(promise_id_), nullptr); | 277 EXPECT_NE(session_.GetPromisedById(promise_id_), nullptr); |
278 | 278 |
279 // Send Response, should trigger promise validation and complete rendezvous | 279 // Send Response, should trigger promise validation and complete rendezvous |
280 QuicSpdyClientStream* promise_stream = static_cast<QuicSpdyClientStream*>( | 280 QuicSpdyClientStream* promise_stream = static_cast<QuicSpdyClientStream*>( |
281 session_.GetOrCreateStream(promise_id_)); | 281 session_.GetOrCreateStream(promise_id_)); |
282 ASSERT_NE(promise_stream, nullptr); | 282 ASSERT_NE(promise_stream, nullptr); |
283 promise_stream->OnStreamHeaders(headers_string_); | 283 promise_stream->OnStreamHeaders(headers_string_); |
284 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); | 284 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); |
(...skipping 11 matching lines...) Expand all Loading... |
296 QuicSpdyClientStream* promise_stream = static_cast<QuicSpdyClientStream*>( | 296 QuicSpdyClientStream* promise_stream = static_cast<QuicSpdyClientStream*>( |
297 session_.GetOrCreateStream(promise_id_)); | 297 session_.GetOrCreateStream(promise_id_)); |
298 ASSERT_NE(promise_stream, nullptr); | 298 ASSERT_NE(promise_stream, nullptr); |
299 | 299 |
300 // Send Response, should trigger promise validation and complete rendezvous | 300 // Send Response, should trigger promise validation and complete rendezvous |
301 promise_stream->OnStreamHeaders(headers_string_); | 301 promise_stream->OnStreamHeaders(headers_string_); |
302 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); | 302 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); |
303 | 303 |
304 // Now initiate rendezvous. | 304 // Now initiate rendezvous. |
305 PushPromiseDelegate delegate(/*match=*/true); | 305 PushPromiseDelegate delegate(/*match=*/true); |
306 promised->HandleClientRequest(std::move(client_request_), &delegate); | 306 promised->HandleClientRequest(client_request_, &delegate); |
307 | 307 |
308 // Promise is gone | 308 // Promise is gone |
309 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); | 309 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); |
310 // Have a push stream | 310 // Have a push stream |
311 EXPECT_TRUE(delegate.rendezvous_fired()); | 311 EXPECT_TRUE(delegate.rendezvous_fired()); |
312 | 312 |
313 EXPECT_NE(delegate.rendezvous_stream(), nullptr); | 313 EXPECT_NE(delegate.rendezvous_stream(), nullptr); |
314 } | 314 } |
315 | 315 |
316 TEST_F(QuicClientPromisedInfoTest, PushPromiseWaitCancels) { | 316 TEST_F(QuicClientPromisedInfoTest, PushPromiseWaitCancels) { |
317 ReceivePromise(promise_id_); | 317 ReceivePromise(promise_id_); |
318 | 318 |
319 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); | 319 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); |
320 ASSERT_NE(promised, nullptr); | 320 ASSERT_NE(promised, nullptr); |
321 | 321 |
322 // Now initiate rendezvous. | 322 // Now initiate rendezvous. |
323 PushPromiseDelegate delegate(/*match=*/true); | 323 PushPromiseDelegate delegate(/*match=*/true); |
324 promised->HandleClientRequest(std::move(client_request_), &delegate); | 324 promised->HandleClientRequest(client_request_, &delegate); |
325 | 325 |
326 // Promise is still there, waiting for response. | 326 // Promise is still there, waiting for response. |
327 EXPECT_NE(session_.GetPromisedById(promise_id_), nullptr); | 327 EXPECT_NE(session_.GetPromisedById(promise_id_), nullptr); |
328 | 328 |
329 // Create response stream, but no data yet. | 329 // Create response stream, but no data yet. |
330 session_.GetOrCreateStream(promise_id_); | 330 session_.GetOrCreateStream(promise_id_); |
331 | 331 |
332 // Fire the alarm that will cancel the promised stream. | 332 // Fire the alarm that will cancel the promised stream. |
333 EXPECT_CALL(session_, CloseStream(promise_id_)); | 333 EXPECT_CALL(session_, CloseStream(promise_id_)); |
334 EXPECT_CALL(*connection_, | 334 EXPECT_CALL(*connection_, |
(...skipping 18 matching lines...) Expand all Loading... |
353 promise_stream->OnStreamHeaders(headers_string_); | 353 promise_stream->OnStreamHeaders(headers_string_); |
354 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); | 354 promise_stream->OnStreamHeadersComplete(false, headers_string_.size()); |
355 | 355 |
356 EXPECT_CALL(session_, CloseStream(promise_id_)); | 356 EXPECT_CALL(session_, CloseStream(promise_id_)); |
357 EXPECT_CALL(*connection_, | 357 EXPECT_CALL(*connection_, |
358 SendRstStream(promise_id_, QUIC_STREAM_PEER_GOING_AWAY, 0)); | 358 SendRstStream(promise_id_, QUIC_STREAM_PEER_GOING_AWAY, 0)); |
359 session_.SendRstStream(promise_id_, QUIC_STREAM_PEER_GOING_AWAY, 0); | 359 session_.SendRstStream(promise_id_, QUIC_STREAM_PEER_GOING_AWAY, 0); |
360 | 360 |
361 // Now initiate rendezvous. | 361 // Now initiate rendezvous. |
362 PushPromiseDelegate delegate(/*match=*/true); | 362 PushPromiseDelegate delegate(/*match=*/true); |
363 EXPECT_EQ( | 363 EXPECT_EQ(promised->HandleClientRequest(client_request_, &delegate), |
364 promised->HandleClientRequest(std::move(client_request_), &delegate), | 364 QUIC_FAILURE); |
365 QUIC_FAILURE); | |
366 | 365 |
367 // Got an indication of the stream failure, client should retry | 366 // Got an indication of the stream failure, client should retry |
368 // request. | 367 // request. |
369 EXPECT_FALSE(delegate.rendezvous_fired()); | 368 EXPECT_FALSE(delegate.rendezvous_fired()); |
370 EXPECT_EQ(delegate.rendezvous_stream(), nullptr); | 369 EXPECT_EQ(delegate.rendezvous_stream(), nullptr); |
371 | 370 |
372 // Promise is gone | 371 // Promise is gone |
373 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); | 372 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); |
374 } | 373 } |
375 | 374 |
376 } // namespace | 375 } // namespace |
377 } // namespace test | 376 } // namespace test |
378 } // namespace net | 377 } // namespace net |
OLD | NEW |