Upstream version 11.40.277.0
[platform/framework/web/crosswalk.git] / src / net / websockets / websocket_test_util.cc
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
3 // found in the LICENSE file.
4
5 #include "net/websockets/websocket_test_util.h"
6
7 #include <algorithm>
8 #include <vector>
9
10 #include "base/basictypes.h"
11 #include "base/memory/scoped_vector.h"
12 #include "base/stl_util.h"
13 #include "base/strings/stringprintf.h"
14 #include "net/socket/socket_test_util.h"
15
16 namespace net {
17
18 namespace {
19 const uint64 kA =
20     (static_cast<uint64>(0x5851f42d) << 32) + static_cast<uint64>(0x4c957f2d);
21 const uint64 kC = 12345;
22 const uint64 kM = static_cast<uint64>(1) << 48;
23
24 }  // namespace
25
26 LinearCongruentialGenerator::LinearCongruentialGenerator(uint32 seed)
27     : current_(seed) {}
28
29 uint32 LinearCongruentialGenerator::Generate() {
30   uint64 result = current_;
31   current_ = (current_ * kA + kC) % kM;
32   return static_cast<uint32>(result >> 16);
33 }
34
35 std::string WebSocketStandardRequest(const std::string& path,
36                                      const std::string& origin,
37                                      const std::string& extra_headers) {
38   // Unrelated changes in net/http may change the order and default-values of
39   // HTTP headers, causing WebSocket tests to fail. It is safe to update this
40   // string in that case.
41   return base::StringPrintf(
42       "GET %s HTTP/1.1\r\n"
43       "Host: localhost\r\n"
44       "Connection: Upgrade\r\n"
45       "Pragma: no-cache\r\n"
46       "Cache-Control: no-cache\r\n"
47       "Upgrade: websocket\r\n"
48       "Origin: %s\r\n"
49       "Sec-WebSocket-Version: 13\r\n"
50       "User-Agent:\r\n"
51       "Accept-Encoding: gzip, deflate\r\n"
52       "Accept-Language: en-us,fr\r\n"
53       "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
54       "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits\r\n"
55       "%s\r\n",
56       path.c_str(),
57       origin.c_str(),
58       extra_headers.c_str());
59 }
60
61 std::string WebSocketStandardResponse(const std::string& extra_headers) {
62   return base::StringPrintf(
63       "HTTP/1.1 101 Switching Protocols\r\n"
64       "Upgrade: websocket\r\n"
65       "Connection: Upgrade\r\n"
66       "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
67       "%s\r\n",
68       extra_headers.c_str());
69 }
70
71 struct WebSocketDeterministicMockClientSocketFactoryMaker::Detail {
72   std::string expect_written;
73   std::string return_to_read;
74   std::vector<MockRead> reads;
75   MockWrite write;
76   ScopedVector<DeterministicSocketData> socket_data_vector;
77   ScopedVector<SSLSocketDataProvider> ssl_socket_data_vector;
78   DeterministicMockClientSocketFactory factory;
79 };
80
81 WebSocketDeterministicMockClientSocketFactoryMaker::
82     WebSocketDeterministicMockClientSocketFactoryMaker()
83     : detail_(new Detail) {}
84
85 WebSocketDeterministicMockClientSocketFactoryMaker::
86     ~WebSocketDeterministicMockClientSocketFactoryMaker() {}
87
88 DeterministicMockClientSocketFactory*
89 WebSocketDeterministicMockClientSocketFactoryMaker::factory() {
90   return &detail_->factory;
91 }
92
93 void WebSocketDeterministicMockClientSocketFactoryMaker::SetExpectations(
94     const std::string& expect_written,
95     const std::string& return_to_read) {
96   const size_t kHttpStreamParserBufferSize = 4096;
97   // We need to extend the lifetime of these strings.
98   detail_->expect_written = expect_written;
99   detail_->return_to_read = return_to_read;
100   int sequence = 0;
101   detail_->write = MockWrite(SYNCHRONOUS,
102                              detail_->expect_written.data(),
103                              detail_->expect_written.size(),
104                              sequence++);
105   // HttpStreamParser reads 4KB at a time. We need to take this implementation
106   // detail into account if |return_to_read| is big enough.
107   for (size_t place = 0; place < detail_->return_to_read.size();
108        place += kHttpStreamParserBufferSize) {
109     detail_->reads.push_back(
110         MockRead(SYNCHRONOUS, detail_->return_to_read.data() + place,
111                  std::min(detail_->return_to_read.size() - place,
112                           kHttpStreamParserBufferSize),
113                  sequence++));
114   }
115   scoped_ptr<DeterministicSocketData> socket_data(
116       new DeterministicSocketData(vector_as_array(&detail_->reads),
117                                   detail_->reads.size(),
118                                   &detail_->write,
119                                   1));
120   socket_data->set_connect_data(MockConnect(SYNCHRONOUS, OK));
121   socket_data->SetStop(sequence);
122   AddRawExpectations(socket_data.Pass());
123 }
124
125 void WebSocketDeterministicMockClientSocketFactoryMaker::AddRawExpectations(
126     scoped_ptr<DeterministicSocketData> socket_data) {
127   detail_->factory.AddSocketDataProvider(socket_data.get());
128   detail_->socket_data_vector.push_back(socket_data.release());
129 }
130
131 void
132 WebSocketDeterministicMockClientSocketFactoryMaker::AddSSLSocketDataProvider(
133     scoped_ptr<SSLSocketDataProvider> ssl_socket_data) {
134   detail_->factory.AddSSLSocketDataProvider(ssl_socket_data.get());
135   detail_->ssl_socket_data_vector.push_back(ssl_socket_data.release());
136 }
137
138 WebSocketTestURLRequestContextHost::WebSocketTestURLRequestContextHost()
139     : url_request_context_(true), url_request_context_initialized_(false) {
140   url_request_context_.set_client_socket_factory(maker_.factory());
141 }
142
143 WebSocketTestURLRequestContextHost::~WebSocketTestURLRequestContextHost() {}
144
145 void WebSocketTestURLRequestContextHost::AddRawExpectations(
146     scoped_ptr<DeterministicSocketData> socket_data) {
147   maker_.AddRawExpectations(socket_data.Pass());
148 }
149
150 void WebSocketTestURLRequestContextHost::AddSSLSocketDataProvider(
151     scoped_ptr<SSLSocketDataProvider> ssl_socket_data) {
152   maker_.AddSSLSocketDataProvider(ssl_socket_data.Pass());
153 }
154
155 TestURLRequestContext*
156 WebSocketTestURLRequestContextHost::GetURLRequestContext() {
157   if (!url_request_context_initialized_) {
158     url_request_context_.Init();
159     // A Network Delegate is required to make the URLRequest::Delegate work.
160     url_request_context_.set_network_delegate(&network_delegate_);
161     url_request_context_initialized_ = true;
162   }
163   return &url_request_context_;
164 }
165
166 }  // namespace net