Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(573)

Issues Search
    My Issues | Starred     Open | Closed | All

Side by Side Diff: net/websockets/websocket_handshake_handler_spdy3_unittest.cc

Issue 17570014: [SPDY] Unfork websocket_handshake_handler_spdy{2,3}_unittest.cc (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Address comments Created 7 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « net/websockets/websocket_handshake_handler_spdy2_unittest.cc ('k') | net/websockets/websocket_handshake_handler_spdy_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright (c) 2012 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_handshake_handler.h"
6
7 #include <string>
8
9 #include "googleurl/src/gurl.h"
10 #include "net/http/http_util.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace net {
14
15 namespace {
16
17 TEST(WebSocketHandshakeHandlerSpdy3Test, RequestResponse) {
18 WebSocketHandshakeRequestHandler request_handler;
19
20 static const char kHandshakeRequestMessage[] =
21 "GET /demo HTTP/1.1\r\n"
22 "Host: example.com\r\n"
23 "Upgrade: websocket\r\n"
24 "Connection: Upgrade\r\n"
25 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
26 "Origin: http://example.com\r\n"
27 "Sec-WebSocket-Protocol: sample\r\n"
28 "Sec-WebSocket-Extensions: foo\r\n"
29 "Sec-WebSocket-Version: 13\r\n"
30 "X-Foo: foo\r\n"
31 "\r\n";
32
33 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
34 strlen(kHandshakeRequestMessage)));
35 EXPECT_EQ(13, request_handler.protocol_version());
36
37 GURL url("ws://example.com/demo");
38 std::string challenge;
39 SpdyHeaderBlock headers;
40 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
41 &headers,
42 &challenge,
43 3));
44
45 EXPECT_EQ(url.path(), headers[":path"]);
46 EXPECT_TRUE(headers.find(":upgrade") == headers.end());
47 EXPECT_TRUE(headers.find(":Upgrade") == headers.end());
48 EXPECT_TRUE(headers.find(":connection") == headers.end());
49 EXPECT_TRUE(headers.find(":Connection") == headers.end());
50 EXPECT_TRUE(headers.find(":Sec-WebSocket-Key") == headers.end());
51 EXPECT_TRUE(headers.find(":sec-websocket-key") == headers.end());
52 EXPECT_TRUE(headers.find(":Sec-WebSocket-Version") == headers.end());
53 EXPECT_TRUE(headers.find(":sec-webSocket-version") == headers.end());
54 EXPECT_TRUE(headers.find(":x-foo") == headers.end());
55 EXPECT_EQ("example.com", headers[":host"]);
56 EXPECT_EQ("http://example.com", headers[":origin"]);
57 EXPECT_EQ("sample", headers[":sec-websocket-protocol"]);
58 EXPECT_EQ("foo", headers[":sec-websocket-extensions"]);
59 EXPECT_EQ("ws", headers[":scheme"]);
60 EXPECT_EQ("WebSocket/13", headers[":version"]);
61 EXPECT_EQ("foo", headers["x-foo"]);
62
63 static const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
64
65 EXPECT_EQ(expected_challenge, challenge);
66
67 headers.clear();
68
69 headers[":status"] = "101 Switching Protocols";
70 headers[":sec-websocket-protocol"] = "sample";
71 headers[":sec-websocket-extensions"] = "foo";
72 headers["x-bar"] = "bar";
73
74 WebSocketHandshakeResponseHandler response_handler;
75 response_handler.set_protocol_version(13);
76 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
77 challenge,
78 3));
79 EXPECT_TRUE(response_handler.HasResponse());
80
81 // Note that order of sec-websocket-* is sensitive with hash_map order.
82 static const char kHandshakeResponseExpectedMessage[] =
83 "HTTP/1.1 101 Switching Protocols\r\n"
84 "Upgrade: websocket\r\n"
85 "Connection: Upgrade\r\n"
86 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
87 "sec-websocket-extensions: foo\r\n"
88 "sec-websocket-protocol: sample\r\n"
89 "x-bar: bar\r\n"
90 "\r\n";
91
92 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
93 }
94
95 TEST(WebSocketHandshakeHandlerSpdy3Test, RequestResponseWithCookies) {
96 WebSocketHandshakeRequestHandler request_handler;
97
98 // Note that websocket won't use multiple headers in request now.
99 static const char kHandshakeRequestMessage[] =
100 "GET /demo HTTP/1.1\r\n"
101 "Host: example.com\r\n"
102 "Upgrade: websocket\r\n"
103 "Connection: Upgrade\r\n"
104 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
105 "Origin: http://example.com\r\n"
106 "Sec-WebSocket-Protocol: sample\r\n"
107 "Sec-WebSocket-Extensions: foo\r\n"
108 "Sec-WebSocket-Version: 13\r\n"
109 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n"
110 "\r\n";
111
112 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
113 strlen(kHandshakeRequestMessage)));
114 EXPECT_EQ(13, request_handler.protocol_version());
115
116 GURL url("ws://example.com/demo");
117 std::string challenge;
118 SpdyHeaderBlock headers;
119 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
120 &headers,
121 &challenge,
122 3));
123
124 EXPECT_EQ(url.path(), headers[":path"]);
125 EXPECT_TRUE(headers.find(":upgrade") == headers.end());
126 EXPECT_TRUE(headers.find(":Upgrade") == headers.end());
127 EXPECT_TRUE(headers.find(":connection") == headers.end());
128 EXPECT_TRUE(headers.find(":Connection") == headers.end());
129 EXPECT_TRUE(headers.find(":Sec-WebSocket-Key") == headers.end());
130 EXPECT_TRUE(headers.find(":sec-websocket-key") == headers.end());
131 EXPECT_TRUE(headers.find(":Sec-WebSocket-Version") == headers.end());
132 EXPECT_TRUE(headers.find(":sec-webSocket-version") == headers.end());
133 EXPECT_TRUE(headers.find(":Cookie") == headers.end());
134 EXPECT_TRUE(headers.find(":cookie") == headers.end());
135 EXPECT_EQ("example.com", headers[":host"]);
136 EXPECT_EQ("http://example.com", headers[":origin"]);
137 EXPECT_EQ("sample", headers[":sec-websocket-protocol"]);
138 EXPECT_EQ("foo", headers[":sec-websocket-extensions"]);
139 EXPECT_EQ("ws", headers[":scheme"]);
140 EXPECT_EQ("WebSocket/13", headers[":version"]);
141 EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1",
142 headers["cookie"]);
143
144 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
145
146 EXPECT_EQ(expected_challenge, challenge);
147
148 headers.clear();
149
150 headers[":status"] = "101 Switching Protocols";
151 headers[":sec-websocket-protocol"] = "sample";
152 headers[":sec-websocket-extensions"] = "foo";
153 std::string cookie = "WK-websocket-test=1";
154 cookie.append(1, '\0');
155 cookie += "WK-websocket-test-httponly=1; HttpOnly";
156 headers["set-cookie"] = cookie;
157
158
159 WebSocketHandshakeResponseHandler response_handler;
160 response_handler.set_protocol_version(13);
161 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
162 challenge,
163 3));
164 EXPECT_TRUE(response_handler.HasResponse());
165
166 // Note that order of sec-websocket-* is sensitive with hash_map order.
167 static const char kHandshakeResponseExpectedMessage[] =
168 "HTTP/1.1 101 Switching Protocols\r\n"
169 "Upgrade: websocket\r\n"
170 "Connection: Upgrade\r\n"
171 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
172 "sec-websocket-extensions: foo\r\n"
173 "sec-websocket-protocol: sample\r\n"
174 "set-cookie: WK-websocket-test=1\r\n"
175 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n"
176 "\r\n";
177
178 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
179 }
180
181 } // namespace
182
183 } // namespace net
OLDNEW
« no previous file with comments | « net/websockets/websocket_handshake_handler_spdy2_unittest.cc ('k') | net/websockets/websocket_handshake_handler_spdy_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698