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/quic_client_promised_info.h" | 5 #include "net/quic/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 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
65 bool authorized_; | 65 bool authorized_; |
66 | 66 |
67 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); | 67 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); |
68 }; | 68 }; |
69 | 69 |
70 class QuicClientPromisedInfoTest : public ::testing::Test { | 70 class QuicClientPromisedInfoTest : public ::testing::Test { |
71 public: | 71 public: |
72 class StreamVisitor; | 72 class StreamVisitor; |
73 | 73 |
74 QuicClientPromisedInfoTest() | 74 QuicClientPromisedInfoTest() |
75 : connection_( | 75 : connection_(new StrictMock<MockConnection>(&helper_, |
76 new StrictMock<MockConnection>(&helper_, Perspective::IS_CLIENT)), | 76 &alarm_factory_, |
| 77 Perspective::IS_CLIENT)), |
77 session_(connection_, &push_promise_index_), | 78 session_(connection_, &push_promise_index_), |
78 body_("hello world"), | 79 body_("hello world"), |
79 promise_id_(gfe_quic::test::kServerDataStreamId1) { | 80 promise_id_(gfe_quic::test::kServerDataStreamId1) { |
80 FLAGS_quic_supports_push_promise = true; | 81 FLAGS_quic_supports_push_promise = true; |
81 | 82 |
82 session_.Initialize(); | 83 session_.Initialize(); |
83 | 84 |
84 headers_.SetResponseFirstline("HTTP/1.1", 200, "Ok"); | 85 headers_.SetResponseFirstline("HTTP/1.1", 200, "Ok"); |
85 headers_.ReplaceOrAppendHeader("content-length", "11"); | 86 headers_.ReplaceOrAppendHeader("content-length", "11"); |
86 headers_string_ = SpdyBalsaUtils::SerializeResponseHeaders(headers_); | 87 headers_string_ = SpdyBalsaUtils::SerializeResponseHeaders(headers_); |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
137 QuicSpdyClientStream* rendezvous_stream_; | 138 QuicSpdyClientStream* rendezvous_stream_; |
138 }; | 139 }; |
139 | 140 |
140 void ReceivePromise(QuicStreamId id) { | 141 void ReceivePromise(QuicStreamId id) { |
141 stream_->OnStreamHeaders(serialized_push_promise_); | 142 stream_->OnStreamHeaders(serialized_push_promise_); |
142 | 143 |
143 stream_->OnPromiseHeadersComplete(id, serialized_push_promise_.size()); | 144 stream_->OnPromiseHeadersComplete(id, serialized_push_promise_.size()); |
144 } | 145 } |
145 | 146 |
146 MockConnectionHelper helper_; | 147 MockConnectionHelper helper_; |
| 148 MockAlarmFactory alarm_factory_; |
147 StrictMock<MockConnection>* connection_; | 149 StrictMock<MockConnection>* connection_; |
148 QuicClientPushPromiseIndex push_promise_index_; | 150 QuicClientPushPromiseIndex push_promise_index_; |
149 | 151 |
150 MockQuicClientSession session_; | 152 MockQuicClientSession session_; |
151 std::unique_ptr<QuicSpdyClientStream> stream_; | 153 std::unique_ptr<QuicSpdyClientStream> stream_; |
152 std::unique_ptr<StreamVisitor> stream_visitor_; | 154 std::unique_ptr<StreamVisitor> stream_visitor_; |
153 std::unique_ptr<QuicSpdyClientStream> promised_stream_; | 155 std::unique_ptr<QuicSpdyClientStream> promised_stream_; |
154 BalsaHeaders headers_; | 156 BalsaHeaders headers_; |
155 string headers_string_; | 157 string headers_string_; |
156 string body_; | 158 string body_; |
(...skipping 14 matching lines...) Expand all Loading... |
171 TEST_F(QuicClientPromisedInfoTest, PushPromiseCleanupAlarm) { | 173 TEST_F(QuicClientPromisedInfoTest, PushPromiseCleanupAlarm) { |
172 ReceivePromise(promise_id_); | 174 ReceivePromise(promise_id_); |
173 | 175 |
174 // Verify that the promise is in the unclaimed streams map. | 176 // Verify that the promise is in the unclaimed streams map. |
175 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); | 177 QuicClientPromisedInfo* promised = session_.GetPromisedById(promise_id_); |
176 ASSERT_NE(promised, nullptr); | 178 ASSERT_NE(promised, nullptr); |
177 | 179 |
178 // Fire the alarm that will cancel the promised stream. | 180 // Fire the alarm that will cancel the promised stream. |
179 EXPECT_CALL(*connection_, | 181 EXPECT_CALL(*connection_, |
180 SendRstStream(promise_id_, QUIC_STREAM_CANCELLED, 0)); | 182 SendRstStream(promise_id_, QUIC_STREAM_CANCELLED, 0)); |
181 helper_.FireAlarm(QuicClientPromisedInfoPeer::GetAlarm(promised)); | 183 alarm_factory_.FireAlarm(QuicClientPromisedInfoPeer::GetAlarm(promised)); |
182 | 184 |
183 // Verify that the promise is gone after the alarm fires. | 185 // Verify that the promise is gone after the alarm fires. |
184 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); | 186 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); |
185 EXPECT_EQ(session_.GetPromisedByUrl(promise_url_), nullptr); | 187 EXPECT_EQ(session_.GetPromisedByUrl(promise_url_), nullptr); |
186 } | 188 } |
187 | 189 |
188 TEST_F(QuicClientPromisedInfoTest, PushPromiseInvalidMethod) { | 190 TEST_F(QuicClientPromisedInfoTest, PushPromiseInvalidMethod) { |
189 // Promise with an unsafe method | 191 // Promise with an unsafe method |
190 push_promise_[":method"] = "PUT"; | 192 push_promise_[":method"] = "PUT"; |
191 serialized_push_promise_ = | 193 serialized_push_promise_ = |
(...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
369 EXPECT_FALSE(delegate.rendezvous_fired()); | 371 EXPECT_FALSE(delegate.rendezvous_fired()); |
370 EXPECT_EQ(delegate.rendezvous_stream(), nullptr); | 372 EXPECT_EQ(delegate.rendezvous_stream(), nullptr); |
371 | 373 |
372 // Promise is gone | 374 // Promise is gone |
373 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); | 375 EXPECT_EQ(session_.GetPromisedById(promise_id_), nullptr); |
374 } | 376 } |
375 | 377 |
376 } // namespace | 378 } // namespace |
377 } // namespace test | 379 } // namespace test |
378 } // namespace net | 380 } // namespace net |
OLD | NEW |