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 // Tests for WebSocketBasicStream. Note that we do not attempt to verify that | 5 // Tests for WebSocketBasicStream. Note that we do not attempt to verify that |
6 // frame parsing itself functions correctly, as that is covered by the | 6 // frame parsing itself functions correctly, as that is covered by the |
7 // WebSocketFrameParser tests. | 7 // WebSocketFrameParser tests. |
8 | 8 |
9 #include "net/websockets/websocket_basic_stream.h" | 9 #include "net/websockets/websocket_basic_stream.h" |
10 | 10 |
(...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
103 | 103 |
104 // A fixture for tests which only perform normal socket operations. | 104 // A fixture for tests which only perform normal socket operations. |
105 class WebSocketBasicStreamSocketTest : public WebSocketBasicStreamTest { | 105 class WebSocketBasicStreamSocketTest : public WebSocketBasicStreamTest { |
106 protected: | 106 protected: |
107 WebSocketBasicStreamSocketTest() | 107 WebSocketBasicStreamSocketTest() |
108 : histograms_("a"), | 108 : histograms_("a"), |
109 pool_(1, 1, &histograms_, &factory_), | 109 pool_(1, 1, &histograms_, &factory_), |
110 generator_(&GenerateNulMaskingKey), | 110 generator_(&GenerateNulMaskingKey), |
111 expect_all_io_to_complete_(true) {} | 111 expect_all_io_to_complete_(true) {} |
112 | 112 |
113 virtual ~WebSocketBasicStreamSocketTest() { | 113 ~WebSocketBasicStreamSocketTest() override { |
114 // stream_ has a reference to socket_data_ (via MockTCPClientSocket) and so | 114 // stream_ has a reference to socket_data_ (via MockTCPClientSocket) and so |
115 // should be destroyed first. | 115 // should be destroyed first. |
116 stream_.reset(); | 116 stream_.reset(); |
117 } | 117 } |
118 | 118 |
119 scoped_ptr<ClientSocketHandle> MakeTransportSocket(MockRead reads[], | 119 scoped_ptr<ClientSocketHandle> MakeTransportSocket(MockRead reads[], |
120 size_t reads_count, | 120 size_t reads_count, |
121 MockWrite writes[], | 121 MockWrite writes[], |
122 size_t writes_count) { | 122 size_t writes_count) { |
123 socket_data_.reset(new StrictStaticSocketDataProvider( | 123 socket_data_.reset(new StrictStaticSocketDataProvider( |
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
230 | 230 |
231 scoped_ptr<MockRead[]> reads_; | 231 scoped_ptr<MockRead[]> reads_; |
232 }; | 232 }; |
233 | 233 |
234 // Test fixture for write tests. | 234 // Test fixture for write tests. |
235 class WebSocketBasicStreamSocketWriteTest | 235 class WebSocketBasicStreamSocketWriteTest |
236 : public WebSocketBasicStreamSocketTest { | 236 : public WebSocketBasicStreamSocketTest { |
237 protected: | 237 protected: |
238 // All write tests use the same frame, so it is easiest to create it during | 238 // All write tests use the same frame, so it is easiest to create it during |
239 // test creation. | 239 // test creation. |
240 virtual void SetUp() override { PrepareWriteFrame(); } | 240 void SetUp() override { PrepareWriteFrame(); } |
241 | 241 |
242 // Creates a WebSocketFrame with a wire format matching kWriteFrame and adds | 242 // Creates a WebSocketFrame with a wire format matching kWriteFrame and adds |
243 // it to |frames_|. | 243 // it to |frames_|. |
244 void PrepareWriteFrame() { | 244 void PrepareWriteFrame() { |
245 scoped_ptr<WebSocketFrame> frame( | 245 scoped_ptr<WebSocketFrame> frame( |
246 new WebSocketFrame(WebSocketFrameHeader::kOpCodeText)); | 246 new WebSocketFrame(WebSocketFrameHeader::kOpCodeText)); |
247 const size_t payload_size = | 247 const size_t payload_size = |
248 kWriteFrameSize - (WebSocketFrameHeader::kBaseHeaderSize + | 248 kWriteFrameSize - (WebSocketFrameHeader::kBaseHeaderSize + |
249 WebSocketFrameHeader::kMaskingKeyLength); | 249 WebSocketFrameHeader::kMaskingKeyLength); |
250 frame->data = new IOBuffer(payload_size); | 250 frame->data = new IOBuffer(payload_size); |
(...skipping 685 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
936 | 936 |
937 TEST_F(WebSocketBasicStreamSocketTest, GetSubProtocolWorks) { | 937 TEST_F(WebSocketBasicStreamSocketTest, GetSubProtocolWorks) { |
938 sub_protocol_ = "cyberchat"; | 938 sub_protocol_ = "cyberchat"; |
939 CreateNullStream(); | 939 CreateNullStream(); |
940 | 940 |
941 EXPECT_EQ("cyberchat", stream_->GetSubProtocol()); | 941 EXPECT_EQ("cyberchat", stream_->GetSubProtocol()); |
942 } | 942 } |
943 | 943 |
944 } // namespace | 944 } // namespace |
945 } // namespace net | 945 } // namespace net |
OLD | NEW |