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/websockets/websocket_channel.h" | 5 #include "net/websockets/websocket_channel.h" |
6 | 6 |
7 #include <limits.h> | 7 #include <limits.h> |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 #include <string.h> | 9 #include <string.h> |
10 | 10 |
(...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
160 | 160 |
161 ChannelState OnDataFrame(bool fin, | 161 ChannelState OnDataFrame(bool fin, |
162 WebSocketMessageType type, | 162 WebSocketMessageType type, |
163 scoped_refptr<IOBuffer> buffer, | 163 scoped_refptr<IOBuffer> buffer, |
164 size_t buffer_size) override { | 164 size_t buffer_size) override { |
165 const char* data = buffer ? buffer->data() : nullptr; | 165 const char* data = buffer ? buffer->data() : nullptr; |
166 return OnDataFrameVector(fin, type, | 166 return OnDataFrameVector(fin, type, |
167 std::vector<char>(data, data + buffer_size)); | 167 std::vector<char>(data, data + buffer_size)); |
168 } | 168 } |
169 | 169 |
| 170 MOCK_METHOD1(OnCreateURLRequest, void(URLRequest*)); |
170 MOCK_METHOD2(OnAddChannelResponse, | 171 MOCK_METHOD2(OnAddChannelResponse, |
171 ChannelState(const std::string&, | 172 ChannelState(const std::string&, |
172 const std::string&)); // NOLINT | 173 const std::string&)); // NOLINT |
173 MOCK_METHOD3(OnDataFrameVector, | 174 MOCK_METHOD3(OnDataFrameVector, |
174 ChannelState(bool, | 175 ChannelState(bool, |
175 WebSocketMessageType, | 176 WebSocketMessageType, |
176 const std::vector<char>&)); // NOLINT | 177 const std::vector<char>&)); // NOLINT |
177 MOCK_METHOD1(OnFlowControl, ChannelState(int64_t)); // NOLINT | 178 MOCK_METHOD1(OnFlowControl, ChannelState(int64_t)); // NOLINT |
178 MOCK_METHOD0(OnClosingHandshake, ChannelState(void)); // NOLINT | 179 MOCK_METHOD0(OnClosingHandshake, ChannelState(void)); // NOLINT |
179 MOCK_METHOD1(OnFailChannel, ChannelState(const std::string&)); // NOLINT | 180 MOCK_METHOD1(OnFailChannel, ChannelState(const std::string&)); // NOLINT |
(...skipping 24 matching lines...) Expand all Loading... |
204 MOCK_METHOD0(OnStartOpeningHandshakeCalled, void()); // NOLINT | 205 MOCK_METHOD0(OnStartOpeningHandshakeCalled, void()); // NOLINT |
205 MOCK_METHOD0(OnFinishOpeningHandshakeCalled, void()); // NOLINT | 206 MOCK_METHOD0(OnFinishOpeningHandshakeCalled, void()); // NOLINT |
206 MOCK_METHOD4( | 207 MOCK_METHOD4( |
207 OnSSLCertificateErrorCalled, | 208 OnSSLCertificateErrorCalled, |
208 void(SSLErrorCallbacks*, const GURL&, const SSLInfo&, bool)); // NOLINT | 209 void(SSLErrorCallbacks*, const GURL&, const SSLInfo&, bool)); // NOLINT |
209 }; | 210 }; |
210 | 211 |
211 // This fake EventInterface is for tests which need a WebSocketEventInterface | 212 // This fake EventInterface is for tests which need a WebSocketEventInterface |
212 // implementation but are not verifying how it is used. | 213 // implementation but are not verifying how it is used. |
213 class FakeWebSocketEventInterface : public WebSocketEventInterface { | 214 class FakeWebSocketEventInterface : public WebSocketEventInterface { |
| 215 void OnCreateURLRequest(URLRequest* request) override {} |
214 ChannelState OnAddChannelResponse(const std::string& selected_protocol, | 216 ChannelState OnAddChannelResponse(const std::string& selected_protocol, |
215 const std::string& extensions) override { | 217 const std::string& extensions) override { |
216 return CHANNEL_ALIVE; | 218 return CHANNEL_ALIVE; |
217 } | 219 } |
218 ChannelState OnDataFrame(bool fin, | 220 ChannelState OnDataFrame(bool fin, |
219 WebSocketMessageType type, | 221 WebSocketMessageType type, |
220 scoped_refptr<IOBuffer> data, | 222 scoped_refptr<IOBuffer> data, |
221 size_t data_size) override { | 223 size_t data_size) override { |
222 return CHANNEL_ALIVE; | 224 return CHANNEL_ALIVE; |
223 } | 225 } |
(...skipping 3301 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3525 channel_->SendFrame( | 3527 channel_->SendFrame( |
3526 true, WebSocketFrameHeader::kOpCodeText, | 3528 true, WebSocketFrameHeader::kOpCodeText, |
3527 AsIOBuffer(std::string(static_cast<size_t>(kMessageSize), 'a')), | 3529 AsIOBuffer(std::string(static_cast<size_t>(kMessageSize), 'a')), |
3528 static_cast<size_t>(kMessageSize)); | 3530 static_cast<size_t>(kMessageSize)); |
3529 int new_send_quota = channel_->current_send_quota(); | 3531 int new_send_quota = channel_->current_send_quota(); |
3530 EXPECT_EQ(kMessageSize, initial_send_quota - new_send_quota); | 3532 EXPECT_EQ(kMessageSize, initial_send_quota - new_send_quota); |
3531 } | 3533 } |
3532 | 3534 |
3533 } // namespace | 3535 } // namespace |
3534 } // namespace net | 3536 } // namespace net |
OLD | NEW |