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_handshake_stream_create_helper.h" | 5 #include "net/websockets/websocket_handshake_stream_create_helper.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <utility> | 8 #include <utility> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 // The created socket expects |expect_written| to be written to the socket, | 42 // The created socket expects |expect_written| to be written to the socket, |
43 // and will respond with |return_to_read|. The test will fail if the expected | 43 // and will respond with |return_to_read|. The test will fail if the expected |
44 // text is not written, or if all the bytes are not read. | 44 // text is not written, or if all the bytes are not read. |
45 std::unique_ptr<ClientSocketHandle> CreateClientSocketHandle( | 45 std::unique_ptr<ClientSocketHandle> CreateClientSocketHandle( |
46 const std::string& expect_written, | 46 const std::string& expect_written, |
47 const std::string& return_to_read) { | 47 const std::string& return_to_read) { |
48 socket_factory_maker_.SetExpectations(expect_written, return_to_read); | 48 socket_factory_maker_.SetExpectations(expect_written, return_to_read); |
49 std::unique_ptr<ClientSocketHandle> socket_handle(new ClientSocketHandle); | 49 std::unique_ptr<ClientSocketHandle> socket_handle(new ClientSocketHandle); |
50 socket_handle->Init("a", scoped_refptr<MockTransportSocketParams>(), MEDIUM, | 50 socket_handle->Init("a", scoped_refptr<MockTransportSocketParams>(), MEDIUM, |
51 ClientSocketPool::RespectLimits::ENABLED, | 51 ClientSocketPool::RespectLimits::ENABLED, |
52 CompletionCallback(), &pool_, BoundNetLog()); | 52 CompletionCallback(), &pool_, NetLogWithSource()); |
53 return socket_handle; | 53 return socket_handle; |
54 } | 54 } |
55 | 55 |
56 private: | 56 private: |
57 WebSocketMockClientSocketFactoryMaker socket_factory_maker_; | 57 WebSocketMockClientSocketFactoryMaker socket_factory_maker_; |
58 MockTransportClientSocketPool pool_; | 58 MockTransportClientSocketPool pool_; |
59 | 59 |
60 DISALLOW_COPY_AND_ASSIGN(MockClientSocketHandleFactory); | 60 DISALLOW_COPY_AND_ASSIGN(MockClientSocketHandleFactory); |
61 }; | 61 }; |
62 | 62 |
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 // If in future the implementation type returned by CreateBasicStream() | 112 // If in future the implementation type returned by CreateBasicStream() |
113 // changes, this static_cast will be wrong. However, in that case the test | 113 // changes, this static_cast will be wrong. However, in that case the test |
114 // will fail and AddressSanitizer should identify the issue. | 114 // will fail and AddressSanitizer should identify the issue. |
115 static_cast<WebSocketBasicHandshakeStream*>(handshake.get()) | 115 static_cast<WebSocketBasicHandshakeStream*>(handshake.get()) |
116 ->SetWebSocketKeyForTesting("dGhlIHNhbXBsZSBub25jZQ=="); | 116 ->SetWebSocketKeyForTesting("dGhlIHNhbXBsZSBub25jZQ=="); |
117 | 117 |
118 HttpRequestInfo request_info; | 118 HttpRequestInfo request_info; |
119 request_info.url = GURL("ws://localhost/"); | 119 request_info.url = GURL("ws://localhost/"); |
120 request_info.method = "GET"; | 120 request_info.method = "GET"; |
121 request_info.load_flags = LOAD_DISABLE_CACHE; | 121 request_info.load_flags = LOAD_DISABLE_CACHE; |
122 int rv = handshake->InitializeStream( | 122 int rv = |
123 &request_info, DEFAULT_PRIORITY, BoundNetLog(), CompletionCallback()); | 123 handshake->InitializeStream(&request_info, DEFAULT_PRIORITY, |
| 124 NetLogWithSource(), CompletionCallback()); |
124 EXPECT_THAT(rv, IsOk()); | 125 EXPECT_THAT(rv, IsOk()); |
125 | 126 |
126 HttpRequestHeaders headers; | 127 HttpRequestHeaders headers; |
127 headers.SetHeader("Host", "localhost"); | 128 headers.SetHeader("Host", "localhost"); |
128 headers.SetHeader("Connection", "Upgrade"); | 129 headers.SetHeader("Connection", "Upgrade"); |
129 headers.SetHeader("Pragma", "no-cache"); | 130 headers.SetHeader("Pragma", "no-cache"); |
130 headers.SetHeader("Cache-Control", "no-cache"); | 131 headers.SetHeader("Cache-Control", "no-cache"); |
131 headers.SetHeader("Upgrade", "websocket"); | 132 headers.SetHeader("Upgrade", "websocket"); |
132 headers.SetHeader("Origin", kOrigin); | 133 headers.SetHeader("Origin", kOrigin); |
133 headers.SetHeader("Sec-WebSocket-Version", "13"); | 134 headers.SetHeader("Sec-WebSocket-Version", "13"); |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
194 | 195 |
195 EXPECT_EQ( | 196 EXPECT_EQ( |
196 "permessage-deflate;" | 197 "permessage-deflate;" |
197 " client_max_window_bits=14; server_max_window_bits=14;" | 198 " client_max_window_bits=14; server_max_window_bits=14;" |
198 " server_no_context_takeover; client_no_context_takeover", | 199 " server_no_context_takeover; client_no_context_takeover", |
199 stream->GetExtensions()); | 200 stream->GetExtensions()); |
200 } | 201 } |
201 | 202 |
202 } // namespace | 203 } // namespace |
203 } // namespace net | 204 } // namespace net |
OLD | NEW |