OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/spdy/spdy_websocket_stream.h" | 5 #include "net/spdy/spdy_websocket_stream.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
(...skipping 18 matching lines...) Expand all Loading... |
29 EVENT_SENT_HEADERS, | 29 EVENT_SENT_HEADERS, |
30 EVENT_RECEIVED_HEADER, | 30 EVENT_RECEIVED_HEADER, |
31 EVENT_SENT_DATA, | 31 EVENT_SENT_DATA, |
32 EVENT_RECEIVED_DATA, | 32 EVENT_RECEIVED_DATA, |
33 EVENT_CLOSE, | 33 EVENT_CLOSE, |
34 }; | 34 }; |
35 SpdyWebSocketStreamEvent(EventType type, | 35 SpdyWebSocketStreamEvent(EventType type, |
36 const SpdyHeaderBlock& headers, | 36 const SpdyHeaderBlock& headers, |
37 int result, | 37 int result, |
38 const std::string& data) | 38 const std::string& data) |
39 : event_type(type), | 39 : event_type(type), headers(headers), result(result), data(data) {} |
40 headers(headers), | |
41 result(result), | |
42 data(data) {} | |
43 | 40 |
44 EventType event_type; | 41 EventType event_type; |
45 SpdyHeaderBlock headers; | 42 SpdyHeaderBlock headers; |
46 int result; | 43 int result; |
47 std::string data; | 44 std::string data; |
48 }; | 45 }; |
49 | 46 |
50 class SpdyWebSocketStreamEventRecorder : public SpdyWebSocketStream::Delegate { | 47 class SpdyWebSocketStreamEventRecorder : public SpdyWebSocketStream::Delegate { |
51 public: | 48 public: |
52 explicit SpdyWebSocketStreamEventRecorder(const CompletionCallback& callback) | 49 explicit SpdyWebSocketStreamEventRecorder(const CompletionCallback& callback) |
(...skipping 10 matching lines...) Expand all Loading... |
63 } | 60 } |
64 void SetOnReceivedHeader(const StreamEventCallback& callback) { | 61 void SetOnReceivedHeader(const StreamEventCallback& callback) { |
65 on_received_header_ = callback; | 62 on_received_header_ = callback; |
66 } | 63 } |
67 void SetOnSentData(const StreamEventCallback& callback) { | 64 void SetOnSentData(const StreamEventCallback& callback) { |
68 on_sent_data_ = callback; | 65 on_sent_data_ = callback; |
69 } | 66 } |
70 void SetOnReceivedData(const StreamEventCallback& callback) { | 67 void SetOnReceivedData(const StreamEventCallback& callback) { |
71 on_received_data_ = callback; | 68 on_received_data_ = callback; |
72 } | 69 } |
73 void SetOnClose(const StreamEventCallback& callback) { | 70 void SetOnClose(const StreamEventCallback& callback) { on_close_ = callback; } |
74 on_close_ = callback; | |
75 } | |
76 | 71 |
77 virtual void OnCreatedSpdyStream(int result) OVERRIDE { | 72 virtual void OnCreatedSpdyStream(int result) OVERRIDE { |
78 events_.push_back( | 73 events_.push_back( |
79 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_CREATED, | 74 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_CREATED, |
80 SpdyHeaderBlock(), | 75 SpdyHeaderBlock(), |
81 result, | 76 result, |
82 std::string())); | 77 std::string())); |
83 if (!on_created_.is_null()) | 78 if (!on_created_.is_null()) |
84 on_created_.Run(&events_.back()); | 79 on_created_.Run(&events_.back()); |
85 } | 80 } |
86 virtual void OnSentSpdyHeaders() OVERRIDE { | 81 virtual void OnSentSpdyHeaders() OVERRIDE { |
87 events_.push_back( | 82 events_.push_back( |
88 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 83 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, |
89 SpdyHeaderBlock(), | 84 SpdyHeaderBlock(), |
90 OK, | 85 OK, |
91 std::string())); | 86 std::string())); |
92 if (!on_sent_data_.is_null()) | 87 if (!on_sent_data_.is_null()) |
93 on_sent_data_.Run(&events_.back()); | 88 on_sent_data_.Run(&events_.back()); |
94 } | 89 } |
95 virtual void OnSpdyResponseHeadersUpdated( | 90 virtual void OnSpdyResponseHeadersUpdated( |
96 const SpdyHeaderBlock& response_headers) OVERRIDE { | 91 const SpdyHeaderBlock& response_headers) OVERRIDE { |
97 events_.push_back( | 92 events_.push_back(SpdyWebSocketStreamEvent( |
98 SpdyWebSocketStreamEvent( | 93 SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
99 SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 94 response_headers, |
100 response_headers, | 95 OK, |
101 OK, | 96 std::string())); |
102 std::string())); | |
103 if (!on_received_header_.is_null()) | 97 if (!on_received_header_.is_null()) |
104 on_received_header_.Run(&events_.back()); | 98 on_received_header_.Run(&events_.back()); |
105 } | 99 } |
106 virtual void OnSentSpdyData(size_t bytes_sent) OVERRIDE { | 100 virtual void OnSentSpdyData(size_t bytes_sent) OVERRIDE { |
107 events_.push_back( | 101 events_.push_back( |
108 SpdyWebSocketStreamEvent( | 102 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, |
109 SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 103 SpdyHeaderBlock(), |
110 SpdyHeaderBlock(), | 104 static_cast<int>(bytes_sent), |
111 static_cast<int>(bytes_sent), | 105 std::string())); |
112 std::string())); | |
113 if (!on_sent_data_.is_null()) | 106 if (!on_sent_data_.is_null()) |
114 on_sent_data_.Run(&events_.back()); | 107 on_sent_data_.Run(&events_.back()); |
115 } | 108 } |
116 virtual void OnReceivedSpdyData(scoped_ptr<SpdyBuffer> buffer) OVERRIDE { | 109 virtual void OnReceivedSpdyData(scoped_ptr<SpdyBuffer> buffer) OVERRIDE { |
117 std::string buffer_data; | 110 std::string buffer_data; |
118 size_t buffer_len = 0; | 111 size_t buffer_len = 0; |
119 if (buffer) { | 112 if (buffer) { |
120 buffer_len = buffer->GetRemainingSize(); | 113 buffer_len = buffer->GetRemainingSize(); |
121 buffer_data.append(buffer->GetRemainingData(), buffer_len); | 114 buffer_data.append(buffer->GetRemainingData(), buffer_len); |
122 } | 115 } |
123 events_.push_back( | 116 events_.push_back( |
124 SpdyWebSocketStreamEvent( | 117 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
125 SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 118 SpdyHeaderBlock(), |
126 SpdyHeaderBlock(), | 119 buffer_len, |
127 buffer_len, | 120 buffer_data)); |
128 buffer_data)); | |
129 if (!on_received_data_.is_null()) | 121 if (!on_received_data_.is_null()) |
130 on_received_data_.Run(&events_.back()); | 122 on_received_data_.Run(&events_.back()); |
131 } | 123 } |
132 virtual void OnCloseSpdyStream() OVERRIDE { | 124 virtual void OnCloseSpdyStream() OVERRIDE { |
133 events_.push_back( | 125 events_.push_back( |
134 SpdyWebSocketStreamEvent( | 126 SpdyWebSocketStreamEvent(SpdyWebSocketStreamEvent::EVENT_CLOSE, |
135 SpdyWebSocketStreamEvent::EVENT_CLOSE, | 127 SpdyHeaderBlock(), |
136 SpdyHeaderBlock(), | 128 OK, |
137 OK, | 129 std::string())); |
138 std::string())); | |
139 if (!on_close_.is_null()) | 130 if (!on_close_.is_null()) |
140 on_close_.Run(&events_.back()); | 131 on_close_.Run(&events_.back()); |
141 if (!callback_.is_null()) | 132 if (!callback_.is_null()) |
142 callback_.Run(OK); | 133 callback_.Run(OK); |
143 } | 134 } |
144 | 135 |
145 const std::vector<SpdyWebSocketStreamEvent>& GetSeenEvents() const { | 136 const std::vector<SpdyWebSocketStreamEvent>& GetSeenEvents() const { |
146 return events_; | 137 return events_; |
147 } | 138 } |
148 | 139 |
(...skipping 21 matching lines...) Expand all Loading... |
170 void DoSendHelloFrame(SpdyWebSocketStreamEvent* event) { | 161 void DoSendHelloFrame(SpdyWebSocketStreamEvent* event) { |
171 // Record the actual stream_id. | 162 // Record the actual stream_id. |
172 created_stream_id_ = websocket_stream_->stream_->stream_id(); | 163 created_stream_id_ = websocket_stream_->stream_->stream_id(); |
173 websocket_stream_->SendData(kMessageFrame, kMessageFrameLength); | 164 websocket_stream_->SendData(kMessageFrame, kMessageFrameLength); |
174 } | 165 } |
175 | 166 |
176 void DoSendClosingFrame(SpdyWebSocketStreamEvent* event) { | 167 void DoSendClosingFrame(SpdyWebSocketStreamEvent* event) { |
177 websocket_stream_->SendData(kClosingFrame, kClosingFrameLength); | 168 websocket_stream_->SendData(kClosingFrame, kClosingFrameLength); |
178 } | 169 } |
179 | 170 |
180 void DoClose(SpdyWebSocketStreamEvent* event) { | 171 void DoClose(SpdyWebSocketStreamEvent* event) { websocket_stream_->Close(); } |
181 websocket_stream_->Close(); | |
182 } | |
183 | 172 |
184 void DoSync(SpdyWebSocketStreamEvent* event) { | 173 void DoSync(SpdyWebSocketStreamEvent* event) { |
185 sync_callback_.callback().Run(OK); | 174 sync_callback_.callback().Run(OK); |
186 } | 175 } |
187 | 176 |
188 protected: | 177 protected: |
189 SpdyWebSocketStreamTest() | 178 SpdyWebSocketStreamTest() |
190 : spdy_util_(GetParam()), | 179 : spdy_util_(GetParam()), |
191 spdy_settings_id_to_set_(SETTINGS_MAX_CONCURRENT_STREAMS), | 180 spdy_settings_id_to_set_(SETTINGS_MAX_CONCURRENT_STREAMS), |
192 spdy_settings_flags_to_set_(SETTINGS_FLAG_PLEASE_PERSIST), | 181 spdy_settings_flags_to_set_(SETTINGS_FLAG_PLEASE_PERSIST), |
193 spdy_settings_value_to_set_(1), | 182 spdy_settings_value_to_set_(1), |
194 session_deps_(GetParam()), | 183 session_deps_(GetParam()), |
195 stream_id_(0), | 184 stream_id_(0), |
196 created_stream_id_(0) {} | 185 created_stream_id_(0) {} |
197 virtual ~SpdyWebSocketStreamTest() {} | 186 virtual ~SpdyWebSocketStreamTest() {} |
198 | 187 |
199 virtual void SetUp() { | 188 virtual void SetUp() { |
200 host_port_pair_.set_host("example.com"); | 189 host_port_pair_.set_host("example.com"); |
201 host_port_pair_.set_port(80); | 190 host_port_pair_.set_port(80); |
202 spdy_session_key_ = SpdySessionKey(host_port_pair_, | 191 spdy_session_key_ = SpdySessionKey( |
203 ProxyServer::Direct(), | 192 host_port_pair_, ProxyServer::Direct(), PRIVACY_MODE_DISABLED); |
204 PRIVACY_MODE_DISABLED); | |
205 | 193 |
206 spdy_settings_to_send_[spdy_settings_id_to_set_] = | 194 spdy_settings_to_send_[spdy_settings_id_to_set_] = SettingsFlagsAndValue( |
207 SettingsFlagsAndValue( | 195 SETTINGS_FLAG_PERSISTED, spdy_settings_value_to_set_); |
208 SETTINGS_FLAG_PERSISTED, spdy_settings_value_to_set_); | |
209 } | 196 } |
210 | 197 |
211 virtual void TearDown() { | 198 virtual void TearDown() { base::MessageLoop::current()->RunUntilIdle(); } |
212 base::MessageLoop::current()->RunUntilIdle(); | |
213 } | |
214 | 199 |
215 void Prepare(SpdyStreamId stream_id) { | 200 void Prepare(SpdyStreamId stream_id) { |
216 stream_id_ = stream_id; | 201 stream_id_ = stream_id; |
217 | 202 |
218 request_frame_.reset(spdy_util_.ConstructSpdyWebSocketSynStream( | 203 request_frame_.reset(spdy_util_.ConstructSpdyWebSocketSynStream( |
219 stream_id_, | 204 stream_id_, "/echo", "example.com", "http://example.com/wsdemo")); |
220 "/echo", | |
221 "example.com", | |
222 "http://example.com/wsdemo")); | |
223 | 205 |
224 response_frame_.reset( | 206 response_frame_.reset( |
225 spdy_util_.ConstructSpdyWebSocketSynReply(stream_id_)); | 207 spdy_util_.ConstructSpdyWebSocketSynReply(stream_id_)); |
226 | 208 |
227 message_frame_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( | 209 message_frame_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( |
228 kMessageFrame, | 210 kMessageFrame, kMessageFrameLength, stream_id_, false)); |
229 kMessageFrameLength, | |
230 stream_id_, | |
231 false)); | |
232 | 211 |
233 closing_frame_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( | 212 closing_frame_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( |
234 kClosingFrame, | 213 kClosingFrame, kClosingFrameLength, stream_id_, false)); |
235 kClosingFrameLength, | |
236 stream_id_, | |
237 false)); | |
238 | 214 |
239 closing_frame_fin_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( | 215 closing_frame_fin_.reset(spdy_util_.ConstructSpdyWebSocketDataFrame( |
240 kClosingFrame, | 216 kClosingFrame, kClosingFrameLength, stream_id_, true)); |
241 kClosingFrameLength, | |
242 stream_id_, | |
243 true)); | |
244 } | 217 } |
245 | 218 |
246 void InitSession(MockRead* reads, size_t reads_count, | 219 void InitSession(MockRead* reads, |
247 MockWrite* writes, size_t writes_count) { | 220 size_t reads_count, |
248 data_.reset(new OrderedSocketData(reads, reads_count, | 221 MockWrite* writes, |
249 writes, writes_count)); | 222 size_t writes_count) { |
| 223 data_.reset( |
| 224 new OrderedSocketData(reads, reads_count, writes, writes_count)); |
250 session_deps_.socket_factory->AddSocketDataProvider(data_.get()); | 225 session_deps_.socket_factory->AddSocketDataProvider(data_.get()); |
251 http_session_ = SpdySessionDependencies::SpdyCreateSession(&session_deps_); | 226 http_session_ = SpdySessionDependencies::SpdyCreateSession(&session_deps_); |
252 session_ = CreateInsecureSpdySession( | 227 session_ = CreateInsecureSpdySession( |
253 http_session_, spdy_session_key_, BoundNetLog()); | 228 http_session_, spdy_session_key_, BoundNetLog()); |
254 } | 229 } |
255 | 230 |
256 void SendRequest() { | 231 void SendRequest() { |
257 scoped_ptr<SpdyHeaderBlock> headers(new SpdyHeaderBlock); | 232 scoped_ptr<SpdyHeaderBlock> headers(new SpdyHeaderBlock); |
258 spdy_util_.SetHeader("path", "/echo", headers.get()); | 233 spdy_util_.SetHeader("path", "/echo", headers.get()); |
259 spdy_util_.SetHeader("host", "example.com", headers.get()); | 234 spdy_util_.SetHeader("host", "example.com", headers.get()); |
(...skipping 24 matching lines...) Expand all Loading... |
284 SpdySessionKey spdy_session_key_; | 259 SpdySessionKey spdy_session_key_; |
285 TestCompletionCallback completion_callback_; | 260 TestCompletionCallback completion_callback_; |
286 TestCompletionCallback sync_callback_; | 261 TestCompletionCallback sync_callback_; |
287 | 262 |
288 static const char kMessageFrame[]; | 263 static const char kMessageFrame[]; |
289 static const char kClosingFrame[]; | 264 static const char kClosingFrame[]; |
290 static const size_t kMessageFrameLength; | 265 static const size_t kMessageFrameLength; |
291 static const size_t kClosingFrameLength; | 266 static const size_t kClosingFrameLength; |
292 }; | 267 }; |
293 | 268 |
294 INSTANTIATE_TEST_CASE_P( | 269 INSTANTIATE_TEST_CASE_P(NextProto, |
295 NextProto, | 270 SpdyWebSocketStreamTest, |
296 SpdyWebSocketStreamTest, | 271 testing::Values(kProtoDeprecatedSPDY2, |
297 testing::Values(kProtoDeprecatedSPDY2, | 272 kProtoSPDY3, |
298 kProtoSPDY3, kProtoSPDY31, kProtoSPDY4)); | 273 kProtoSPDY31, |
| 274 kProtoSPDY4)); |
299 | 275 |
300 // TODO(toyoshim): Replace old framing data to new one, then use HEADERS and | 276 // TODO(toyoshim): Replace old framing data to new one, then use HEADERS and |
301 // data frames. | 277 // data frames. |
302 const char SpdyWebSocketStreamTest::kMessageFrame[] = "\x81\x05hello"; | 278 const char SpdyWebSocketStreamTest::kMessageFrame[] = "\x81\x05hello"; |
303 const char SpdyWebSocketStreamTest::kClosingFrame[] = "\x88\0"; | 279 const char SpdyWebSocketStreamTest::kClosingFrame[] = "\x88\0"; |
304 const size_t SpdyWebSocketStreamTest::kMessageFrameLength = | 280 const size_t SpdyWebSocketStreamTest::kMessageFrameLength = |
305 arraysize(SpdyWebSocketStreamTest::kMessageFrame) - 1; | 281 arraysize(SpdyWebSocketStreamTest::kMessageFrame) - 1; |
306 const size_t SpdyWebSocketStreamTest::kClosingFrameLength = | 282 const size_t SpdyWebSocketStreamTest::kClosingFrameLength = |
307 arraysize(SpdyWebSocketStreamTest::kClosingFrame) - 1; | 283 arraysize(SpdyWebSocketStreamTest::kClosingFrame) - 1; |
308 | 284 |
309 TEST_P(SpdyWebSocketStreamTest, Basic) { | 285 TEST_P(SpdyWebSocketStreamTest, Basic) { |
310 Prepare(1); | 286 Prepare(1); |
311 MockWrite writes[] = { | 287 MockWrite writes[] = {CreateMockWrite(*request_frame_.get(), 1), |
312 CreateMockWrite(*request_frame_.get(), 1), | 288 CreateMockWrite(*message_frame_.get(), 3), |
313 CreateMockWrite(*message_frame_.get(), 3), | 289 CreateMockWrite(*closing_frame_.get(), 5)}; |
314 CreateMockWrite(*closing_frame_.get(), 5) | |
315 }; | |
316 | 290 |
317 MockRead reads[] = { | 291 MockRead reads[] = { |
318 CreateMockRead(*response_frame_.get(), 2), | 292 CreateMockRead(*response_frame_.get(), 2), |
319 CreateMockRead(*message_frame_.get(), 4), | 293 CreateMockRead(*message_frame_.get(), 4), |
320 // Skip sequence 6 to notify closing has been sent. | 294 // Skip sequence 6 to notify closing has been sent. |
321 CreateMockRead(*closing_frame_.get(), 7), | 295 CreateMockRead(*closing_frame_.get(), 7), |
322 MockRead(SYNCHRONOUS, 0, 8) // EOF cause OnCloseSpdyStream event. | 296 MockRead(SYNCHRONOUS, 0, 8) // EOF cause OnCloseSpdyStream event. |
323 }; | 297 }; |
324 | 298 |
325 InitSession(reads, arraysize(reads), writes, arraysize(writes)); | 299 InitSession(reads, arraysize(reads), writes, arraysize(writes)); |
326 | 300 |
327 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); | 301 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); |
328 delegate.SetOnReceivedHeader( | 302 delegate.SetOnReceivedHeader(base::Bind( |
329 base::Bind(&SpdyWebSocketStreamTest::DoSendHelloFrame, | 303 &SpdyWebSocketStreamTest::DoSendHelloFrame, base::Unretained(this))); |
330 base::Unretained(this))); | 304 delegate.SetOnReceivedData(base::Bind( |
331 delegate.SetOnReceivedData( | 305 &SpdyWebSocketStreamTest::DoSendClosingFrame, base::Unretained(this))); |
332 base::Bind(&SpdyWebSocketStreamTest::DoSendClosingFrame, | |
333 base::Unretained(this))); | |
334 | 306 |
335 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); | 307 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); |
336 | 308 |
337 BoundNetLog net_log; | 309 BoundNetLog net_log; |
338 GURL url("ws://example.com/echo"); | 310 GURL url("ws://example.com/echo"); |
339 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); | 311 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); |
340 | 312 |
341 ASSERT_TRUE(websocket_stream_->stream_.get()); | 313 ASSERT_TRUE(websocket_stream_->stream_.get()); |
342 | 314 |
343 SendRequest(); | 315 SendRequest(); |
344 | 316 |
345 completion_callback_.WaitForResult(); | 317 completion_callback_.WaitForResult(); |
346 | 318 |
347 EXPECT_EQ(stream_id_, created_stream_id_); | 319 EXPECT_EQ(stream_id_, created_stream_id_); |
348 | 320 |
349 websocket_stream_.reset(); | 321 websocket_stream_.reset(); |
350 | 322 |
351 const std::vector<SpdyWebSocketStreamEvent>& events = | 323 const std::vector<SpdyWebSocketStreamEvent>& events = |
352 delegate.GetSeenEvents(); | 324 delegate.GetSeenEvents(); |
353 ASSERT_EQ(7U, events.size()); | 325 ASSERT_EQ(7U, events.size()); |
354 | 326 |
355 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 327 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, events[0].event_type); |
356 events[0].event_type); | |
357 EXPECT_EQ(OK, events[0].result); | 328 EXPECT_EQ(OK, events[0].result); |
358 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 329 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
359 events[1].event_type); | 330 events[1].event_type); |
360 EXPECT_EQ(OK, events[1].result); | 331 EXPECT_EQ(OK, events[1].result); |
361 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 332 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[2].event_type); |
362 events[2].event_type); | |
363 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); | 333 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); |
364 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 334 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
365 events[3].event_type); | 335 events[3].event_type); |
366 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); | 336 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); |
367 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 337 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[4].event_type); |
368 events[4].event_type); | |
369 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[4].result); | 338 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[4].result); |
370 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 339 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
371 events[5].event_type); | 340 events[5].event_type); |
372 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[5].result); | 341 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[5].result); |
373 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, | 342 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, events[6].event_type); |
374 events[6].event_type); | |
375 EXPECT_EQ(OK, events[6].result); | 343 EXPECT_EQ(OK, events[6].result); |
376 | 344 |
377 // EOF close SPDY session. | 345 // EOF close SPDY session. |
378 EXPECT_FALSE( | 346 EXPECT_FALSE( |
379 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); | 347 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); |
380 EXPECT_TRUE(data()->at_read_eof()); | 348 EXPECT_TRUE(data()->at_read_eof()); |
381 EXPECT_TRUE(data()->at_write_eof()); | 349 EXPECT_TRUE(data()->at_write_eof()); |
382 } | 350 } |
383 | 351 |
384 // A SPDY websocket may still send it's close frame after | 352 // A SPDY websocket may still send it's close frame after |
385 // recieving a close with SPDY stream FIN. | 353 // recieving a close with SPDY stream FIN. |
386 TEST_P(SpdyWebSocketStreamTest, RemoteCloseWithFin) { | 354 TEST_P(SpdyWebSocketStreamTest, RemoteCloseWithFin) { |
387 Prepare(1); | 355 Prepare(1); |
388 MockWrite writes[] = { | 356 MockWrite writes[] = { |
389 CreateMockWrite(*request_frame_.get(), 1), | 357 CreateMockWrite(*request_frame_.get(), 1), |
390 CreateMockWrite(*closing_frame_.get(), 4), | 358 CreateMockWrite(*closing_frame_.get(), 4), |
391 }; | 359 }; |
392 MockRead reads[] = { | 360 MockRead reads[] = { |
393 CreateMockRead(*response_frame_.get(), 2), | 361 CreateMockRead(*response_frame_.get(), 2), |
394 CreateMockRead(*closing_frame_fin_.get(), 3), | 362 CreateMockRead(*closing_frame_fin_.get(), 3), |
395 MockRead(SYNCHRONOUS, 0, 5) // EOF cause OnCloseSpdyStream event. | 363 MockRead(SYNCHRONOUS, 0, 5) // EOF cause OnCloseSpdyStream event. |
396 }; | 364 }; |
397 InitSession(reads, arraysize(reads), writes, arraysize(writes)); | 365 InitSession(reads, arraysize(reads), writes, arraysize(writes)); |
398 | 366 |
399 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); | 367 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); |
400 delegate.SetOnReceivedData( | 368 delegate.SetOnReceivedData(base::Bind( |
401 base::Bind(&SpdyWebSocketStreamTest::DoSendClosingFrame, | 369 &SpdyWebSocketStreamTest::DoSendClosingFrame, base::Unretained(this))); |
402 base::Unretained(this))); | |
403 | 370 |
404 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); | 371 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); |
405 BoundNetLog net_log; | 372 BoundNetLog net_log; |
406 GURL url("ws://example.com/echo"); | 373 GURL url("ws://example.com/echo"); |
407 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); | 374 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); |
408 | 375 |
409 SendRequest(); | 376 SendRequest(); |
410 completion_callback_.WaitForResult(); | 377 completion_callback_.WaitForResult(); |
411 websocket_stream_.reset(); | 378 websocket_stream_.reset(); |
412 | 379 |
413 const std::vector<SpdyWebSocketStreamEvent>& events = | 380 const std::vector<SpdyWebSocketStreamEvent>& events = |
414 delegate.GetSeenEvents(); | 381 delegate.GetSeenEvents(); |
415 EXPECT_EQ(5U, events.size()); | 382 EXPECT_EQ(5U, events.size()); |
416 | 383 |
417 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 384 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, events[0].event_type); |
418 events[0].event_type); | |
419 EXPECT_EQ(OK, events[0].result); | 385 EXPECT_EQ(OK, events[0].result); |
420 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 386 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
421 events[1].event_type); | 387 events[1].event_type); |
422 EXPECT_EQ(OK, events[1].result); | 388 EXPECT_EQ(OK, events[1].result); |
423 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 389 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
424 events[2].event_type); | 390 events[2].event_type); |
425 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[2].result); | 391 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[2].result); |
426 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 392 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[3].event_type); |
427 events[3].event_type); | |
428 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[3].result); | 393 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[3].result); |
429 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, | 394 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, events[4].event_type); |
430 events[4].event_type); | |
431 EXPECT_EQ(OK, events[4].result); | 395 EXPECT_EQ(OK, events[4].result); |
432 | 396 |
433 // EOF closes SPDY session. | 397 // EOF closes SPDY session. |
434 EXPECT_FALSE( | 398 EXPECT_FALSE( |
435 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); | 399 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); |
436 EXPECT_TRUE(data()->at_read_eof()); | 400 EXPECT_TRUE(data()->at_read_eof()); |
437 EXPECT_TRUE(data()->at_write_eof()); | 401 EXPECT_TRUE(data()->at_write_eof()); |
438 } | 402 } |
439 | 403 |
440 TEST_P(SpdyWebSocketStreamTest, DestructionBeforeClose) { | 404 TEST_P(SpdyWebSocketStreamTest, DestructionBeforeClose) { |
441 Prepare(1); | 405 Prepare(1); |
442 MockWrite writes[] = { | 406 MockWrite writes[] = {CreateMockWrite(*request_frame_.get(), 1), |
443 CreateMockWrite(*request_frame_.get(), 1), | 407 CreateMockWrite(*message_frame_.get(), 3)}; |
444 CreateMockWrite(*message_frame_.get(), 3) | |
445 }; | |
446 | 408 |
447 MockRead reads[] = { | 409 MockRead reads[] = {CreateMockRead(*response_frame_.get(), 2), |
448 CreateMockRead(*response_frame_.get(), 2), | 410 CreateMockRead(*message_frame_.get(), 4), |
449 CreateMockRead(*message_frame_.get(), 4), | 411 MockRead(ASYNC, ERR_IO_PENDING, 5)}; |
450 MockRead(ASYNC, ERR_IO_PENDING, 5) | |
451 }; | |
452 | 412 |
453 InitSession(reads, arraysize(reads), writes, arraysize(writes)); | 413 InitSession(reads, arraysize(reads), writes, arraysize(writes)); |
454 | 414 |
455 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); | 415 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); |
456 delegate.SetOnReceivedHeader( | 416 delegate.SetOnReceivedHeader(base::Bind( |
457 base::Bind(&SpdyWebSocketStreamTest::DoSendHelloFrame, | 417 &SpdyWebSocketStreamTest::DoSendHelloFrame, base::Unretained(this))); |
458 base::Unretained(this))); | |
459 delegate.SetOnReceivedData( | 418 delegate.SetOnReceivedData( |
460 base::Bind(&SpdyWebSocketStreamTest::DoSync, | 419 base::Bind(&SpdyWebSocketStreamTest::DoSync, base::Unretained(this))); |
461 base::Unretained(this))); | |
462 | 420 |
463 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); | 421 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); |
464 | 422 |
465 BoundNetLog net_log; | 423 BoundNetLog net_log; |
466 GURL url("ws://example.com/echo"); | 424 GURL url("ws://example.com/echo"); |
467 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); | 425 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); |
468 | 426 |
469 SendRequest(); | 427 SendRequest(); |
470 | 428 |
471 sync_callback_.WaitForResult(); | 429 sync_callback_.WaitForResult(); |
472 | 430 |
473 // WebSocketStream destruction remove its SPDY stream from the session. | 431 // WebSocketStream destruction remove its SPDY stream from the session. |
474 EXPECT_TRUE(session_->IsStreamActive(stream_id_)); | 432 EXPECT_TRUE(session_->IsStreamActive(stream_id_)); |
475 websocket_stream_.reset(); | 433 websocket_stream_.reset(); |
476 EXPECT_FALSE(session_->IsStreamActive(stream_id_)); | 434 EXPECT_FALSE(session_->IsStreamActive(stream_id_)); |
477 | 435 |
478 const std::vector<SpdyWebSocketStreamEvent>& events = | 436 const std::vector<SpdyWebSocketStreamEvent>& events = |
479 delegate.GetSeenEvents(); | 437 delegate.GetSeenEvents(); |
480 ASSERT_GE(4U, events.size()); | 438 ASSERT_GE(4U, events.size()); |
481 | 439 |
482 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 440 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, events[0].event_type); |
483 events[0].event_type); | |
484 EXPECT_EQ(OK, events[0].result); | 441 EXPECT_EQ(OK, events[0].result); |
485 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 442 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
486 events[1].event_type); | 443 events[1].event_type); |
487 EXPECT_EQ(OK, events[1].result); | 444 EXPECT_EQ(OK, events[1].result); |
488 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 445 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[2].event_type); |
489 events[2].event_type); | |
490 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); | 446 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); |
491 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 447 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
492 events[3].event_type); | 448 events[3].event_type); |
493 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); | 449 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); |
494 | 450 |
495 EXPECT_TRUE( | 451 EXPECT_TRUE( |
496 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); | 452 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); |
497 EXPECT_TRUE(data()->at_read_eof()); | 453 EXPECT_TRUE(data()->at_read_eof()); |
498 EXPECT_TRUE(data()->at_write_eof()); | 454 EXPECT_TRUE(data()->at_write_eof()); |
499 } | 455 } |
500 | 456 |
501 TEST_P(SpdyWebSocketStreamTest, DestructionAfterExplicitClose) { | 457 TEST_P(SpdyWebSocketStreamTest, DestructionAfterExplicitClose) { |
502 Prepare(1); | 458 Prepare(1); |
503 MockWrite writes[] = { | 459 MockWrite writes[] = {CreateMockWrite(*request_frame_.get(), 1), |
504 CreateMockWrite(*request_frame_.get(), 1), | 460 CreateMockWrite(*message_frame_.get(), 3), |
505 CreateMockWrite(*message_frame_.get(), 3), | 461 CreateMockWrite(*closing_frame_.get(), 5)}; |
506 CreateMockWrite(*closing_frame_.get(), 5) | |
507 }; | |
508 | 462 |
509 MockRead reads[] = { | 463 MockRead reads[] = {CreateMockRead(*response_frame_.get(), 2), |
510 CreateMockRead(*response_frame_.get(), 2), | 464 CreateMockRead(*message_frame_.get(), 4), |
511 CreateMockRead(*message_frame_.get(), 4), | 465 MockRead(ASYNC, ERR_IO_PENDING, 6)}; |
512 MockRead(ASYNC, ERR_IO_PENDING, 6) | |
513 }; | |
514 | 466 |
515 InitSession(reads, arraysize(reads), writes, arraysize(writes)); | 467 InitSession(reads, arraysize(reads), writes, arraysize(writes)); |
516 | 468 |
517 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); | 469 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); |
518 delegate.SetOnReceivedHeader( | 470 delegate.SetOnReceivedHeader(base::Bind( |
519 base::Bind(&SpdyWebSocketStreamTest::DoSendHelloFrame, | 471 &SpdyWebSocketStreamTest::DoSendHelloFrame, base::Unretained(this))); |
520 base::Unretained(this))); | |
521 delegate.SetOnReceivedData( | 472 delegate.SetOnReceivedData( |
522 base::Bind(&SpdyWebSocketStreamTest::DoClose, | 473 base::Bind(&SpdyWebSocketStreamTest::DoClose, base::Unretained(this))); |
523 base::Unretained(this))); | |
524 | 474 |
525 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); | 475 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); |
526 | 476 |
527 BoundNetLog net_log; | 477 BoundNetLog net_log; |
528 GURL url("ws://example.com/echo"); | 478 GURL url("ws://example.com/echo"); |
529 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); | 479 ASSERT_EQ(OK, websocket_stream_->InitializeStream(url, HIGHEST, net_log)); |
530 | 480 |
531 SendRequest(); | 481 SendRequest(); |
532 | 482 |
533 completion_callback_.WaitForResult(); | 483 completion_callback_.WaitForResult(); |
534 | 484 |
535 // SPDY stream has already been removed from the session by Close(). | 485 // SPDY stream has already been removed from the session by Close(). |
536 EXPECT_FALSE(session_->IsStreamActive(stream_id_)); | 486 EXPECT_FALSE(session_->IsStreamActive(stream_id_)); |
537 websocket_stream_.reset(); | 487 websocket_stream_.reset(); |
538 | 488 |
539 const std::vector<SpdyWebSocketStreamEvent>& events = | 489 const std::vector<SpdyWebSocketStreamEvent>& events = |
540 delegate.GetSeenEvents(); | 490 delegate.GetSeenEvents(); |
541 ASSERT_EQ(5U, events.size()); | 491 ASSERT_EQ(5U, events.size()); |
542 | 492 |
543 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 493 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, events[0].event_type); |
544 events[0].event_type); | |
545 EXPECT_EQ(OK, events[0].result); | 494 EXPECT_EQ(OK, events[0].result); |
546 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 495 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
547 events[1].event_type); | 496 events[1].event_type); |
548 EXPECT_EQ(OK, events[1].result); | 497 EXPECT_EQ(OK, events[1].result); |
549 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 498 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[2].event_type); |
550 events[2].event_type); | |
551 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); | 499 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[2].result); |
552 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 500 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
553 events[3].event_type); | 501 events[3].event_type); |
554 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); | 502 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); |
555 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, events[4].event_type); | 503 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, events[4].event_type); |
556 | 504 |
557 EXPECT_TRUE( | 505 EXPECT_TRUE( |
558 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); | 506 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); |
559 } | 507 } |
560 | 508 |
561 TEST_P(SpdyWebSocketStreamTest, IOPending) { | 509 TEST_P(SpdyWebSocketStreamTest, IOPending) { |
562 Prepare(1); | 510 Prepare(1); |
563 scoped_ptr<SpdyFrame> settings_frame( | 511 scoped_ptr<SpdyFrame> settings_frame( |
564 spdy_util_.ConstructSpdySettings(spdy_settings_to_send_)); | 512 spdy_util_.ConstructSpdySettings(spdy_settings_to_send_)); |
565 MockWrite writes[] = { | 513 MockWrite writes[] = {CreateMockWrite(*request_frame_.get(), 1), |
566 CreateMockWrite(*request_frame_.get(), 1), | 514 CreateMockWrite(*message_frame_.get(), 3), |
567 CreateMockWrite(*message_frame_.get(), 3), | 515 CreateMockWrite(*closing_frame_.get(), 5)}; |
568 CreateMockWrite(*closing_frame_.get(), 5) | 516 |
| 517 MockRead reads[] = { |
| 518 CreateMockRead(*settings_frame.get(), 0), |
| 519 CreateMockRead(*response_frame_.get(), 2), |
| 520 CreateMockRead(*message_frame_.get(), 4), |
| 521 CreateMockRead(*closing_frame_.get(), 6), |
| 522 MockRead(SYNCHRONOUS, 0, 7) // EOF cause OnCloseSpdyStream event. |
569 }; | 523 }; |
570 | 524 |
571 MockRead reads[] = { | 525 DeterministicSocketData data( |
572 CreateMockRead(*settings_frame.get(), 0), | 526 reads, arraysize(reads), writes, arraysize(writes)); |
573 CreateMockRead(*response_frame_.get(), 2), | |
574 CreateMockRead(*message_frame_.get(), 4), | |
575 CreateMockRead(*closing_frame_.get(), 6), | |
576 MockRead(SYNCHRONOUS, 0, 7) // EOF cause OnCloseSpdyStream event. | |
577 }; | |
578 | |
579 DeterministicSocketData data(reads, arraysize(reads), | |
580 writes, arraysize(writes)); | |
581 session_deps_.deterministic_socket_factory->AddSocketDataProvider(&data); | 527 session_deps_.deterministic_socket_factory->AddSocketDataProvider(&data); |
582 http_session_ = | 528 http_session_ = |
583 SpdySessionDependencies::SpdyCreateSessionDeterministic(&session_deps_); | 529 SpdySessionDependencies::SpdyCreateSessionDeterministic(&session_deps_); |
584 | 530 |
585 session_ = CreateInsecureSpdySession( | 531 session_ = CreateInsecureSpdySession( |
586 http_session_, spdy_session_key_, BoundNetLog()); | 532 http_session_, spdy_session_key_, BoundNetLog()); |
587 | 533 |
588 // Create a dummy WebSocketStream which cause ERR_IO_PENDING to another | 534 // Create a dummy WebSocketStream which cause ERR_IO_PENDING to another |
589 // WebSocketStream under test. | 535 // WebSocketStream under test. |
590 SpdyWebSocketStreamEventRecorder block_delegate((CompletionCallback())); | 536 SpdyWebSocketStreamEventRecorder block_delegate((CompletionCallback())); |
591 | 537 |
592 scoped_ptr<SpdyWebSocketStream> block_stream( | 538 scoped_ptr<SpdyWebSocketStream> block_stream( |
593 new SpdyWebSocketStream(session_, &block_delegate)); | 539 new SpdyWebSocketStream(session_, &block_delegate)); |
594 BoundNetLog block_net_log; | 540 BoundNetLog block_net_log; |
595 GURL block_url("ws://example.com/block"); | 541 GURL block_url("ws://example.com/block"); |
596 ASSERT_EQ(OK, | 542 ASSERT_EQ(OK, |
597 block_stream->InitializeStream(block_url, HIGHEST, block_net_log)); | 543 block_stream->InitializeStream(block_url, HIGHEST, block_net_log)); |
598 | 544 |
599 data.RunFor(1); | 545 data.RunFor(1); |
600 | 546 |
601 // Create a WebSocketStream under test. | 547 // Create a WebSocketStream under test. |
602 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); | 548 SpdyWebSocketStreamEventRecorder delegate(completion_callback_.callback()); |
603 delegate.SetOnCreated( | 549 delegate.SetOnCreated( |
604 base::Bind(&SpdyWebSocketStreamTest::DoSync, | 550 base::Bind(&SpdyWebSocketStreamTest::DoSync, base::Unretained(this))); |
605 base::Unretained(this))); | 551 delegate.SetOnReceivedHeader(base::Bind( |
606 delegate.SetOnReceivedHeader( | 552 &SpdyWebSocketStreamTest::DoSendHelloFrame, base::Unretained(this))); |
607 base::Bind(&SpdyWebSocketStreamTest::DoSendHelloFrame, | 553 delegate.SetOnReceivedData(base::Bind( |
608 base::Unretained(this))); | 554 &SpdyWebSocketStreamTest::DoSendClosingFrame, base::Unretained(this))); |
609 delegate.SetOnReceivedData( | |
610 base::Bind(&SpdyWebSocketStreamTest::DoSendClosingFrame, | |
611 base::Unretained(this))); | |
612 | 555 |
613 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); | 556 websocket_stream_.reset(new SpdyWebSocketStream(session_, &delegate)); |
614 BoundNetLog net_log; | 557 BoundNetLog net_log; |
615 GURL url("ws://example.com/echo"); | 558 GURL url("ws://example.com/echo"); |
616 ASSERT_EQ(ERR_IO_PENDING, websocket_stream_->InitializeStream( | 559 ASSERT_EQ(ERR_IO_PENDING, |
617 url, HIGHEST, net_log)); | 560 websocket_stream_->InitializeStream(url, HIGHEST, net_log)); |
618 | 561 |
619 // Delete the fist stream to allow create the second stream. | 562 // Delete the fist stream to allow create the second stream. |
620 block_stream.reset(); | 563 block_stream.reset(); |
621 ASSERT_EQ(OK, sync_callback_.WaitForResult()); | 564 ASSERT_EQ(OK, sync_callback_.WaitForResult()); |
622 | 565 |
623 SendRequest(); | 566 SendRequest(); |
624 | 567 |
625 data.RunFor(7); | 568 data.RunFor(7); |
626 completion_callback_.WaitForResult(); | 569 completion_callback_.WaitForResult(); |
627 | 570 |
628 websocket_stream_.reset(); | 571 websocket_stream_.reset(); |
629 | 572 |
630 const std::vector<SpdyWebSocketStreamEvent>& block_events = | 573 const std::vector<SpdyWebSocketStreamEvent>& block_events = |
631 block_delegate.GetSeenEvents(); | 574 block_delegate.GetSeenEvents(); |
632 ASSERT_EQ(0U, block_events.size()); | 575 ASSERT_EQ(0U, block_events.size()); |
633 | 576 |
634 const std::vector<SpdyWebSocketStreamEvent>& events = | 577 const std::vector<SpdyWebSocketStreamEvent>& events = |
635 delegate.GetSeenEvents(); | 578 delegate.GetSeenEvents(); |
636 ASSERT_EQ(8U, events.size()); | 579 ASSERT_EQ(8U, events.size()); |
637 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CREATED, | 580 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CREATED, events[0].event_type); |
638 events[0].event_type); | |
639 EXPECT_EQ(0, events[0].result); | 581 EXPECT_EQ(0, events[0].result); |
640 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, | 582 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_HEADERS, events[1].event_type); |
641 events[1].event_type); | |
642 EXPECT_EQ(OK, events[1].result); | 583 EXPECT_EQ(OK, events[1].result); |
643 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, | 584 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_HEADER, |
644 events[2].event_type); | 585 events[2].event_type); |
645 EXPECT_EQ(OK, events[2].result); | 586 EXPECT_EQ(OK, events[2].result); |
646 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 587 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[3].event_type); |
647 events[3].event_type); | |
648 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); | 588 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[3].result); |
649 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 589 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
650 events[4].event_type); | 590 events[4].event_type); |
651 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[4].result); | 591 EXPECT_EQ(static_cast<int>(kMessageFrameLength), events[4].result); |
652 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, | 592 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_SENT_DATA, events[5].event_type); |
653 events[5].event_type); | |
654 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[5].result); | 593 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[5].result); |
655 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, | 594 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_RECEIVED_DATA, |
656 events[6].event_type); | 595 events[6].event_type); |
657 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[6].result); | 596 EXPECT_EQ(static_cast<int>(kClosingFrameLength), events[6].result); |
658 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, | 597 EXPECT_EQ(SpdyWebSocketStreamEvent::EVENT_CLOSE, events[7].event_type); |
659 events[7].event_type); | |
660 EXPECT_EQ(OK, events[7].result); | 598 EXPECT_EQ(OK, events[7].result); |
661 | 599 |
662 // EOF close SPDY session. | 600 // EOF close SPDY session. |
663 EXPECT_FALSE( | 601 EXPECT_FALSE( |
664 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); | 602 HasSpdySession(http_session_->spdy_session_pool(), spdy_session_key_)); |
665 EXPECT_TRUE(data.at_read_eof()); | 603 EXPECT_TRUE(data.at_read_eof()); |
666 EXPECT_TRUE(data.at_write_eof()); | 604 EXPECT_TRUE(data.at_write_eof()); |
667 } | 605 } |
668 | 606 |
669 } // namespace net | 607 } // namespace net |
OLD | NEW |