| 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 |