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

Issues Search
    My Issues | Starred     Open | Closed | All

Side by Side Diff: net/websockets/websocket_handshake_handler_spdy2_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/spdy/spdy_websocket_test_util.cc ('k') | net/websockets/websocket_handshake_handler_spdy3_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/spdy/spdy_header_block.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace net {
14
15 namespace {
16
17 TEST(WebSocketHandshakeHandlerSpdy2Test, 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 "\r\n";
31
32 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
33 strlen(kHandshakeRequestMessage)));
34 EXPECT_EQ(13, request_handler.protocol_version());
35
36 GURL url("ws://example.com/demo");
37 std::string challenge;
38 SpdyHeaderBlock headers;
39 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
40 &headers,
41 &challenge,
42 2));
43
44 EXPECT_EQ(url.path(), headers["path"]);
45 EXPECT_TRUE(headers.find("upgrade") == headers.end());
46 EXPECT_TRUE(headers.find("Upgrade") == headers.end());
47 EXPECT_TRUE(headers.find("connection") == headers.end());
48 EXPECT_TRUE(headers.find("Connection") == headers.end());
49 EXPECT_TRUE(headers.find("Sec-WebSocket-Key") == headers.end());
50 EXPECT_TRUE(headers.find("sec-websocket-key") == headers.end());
51 EXPECT_TRUE(headers.find("Sec-WebSocket-Version") == headers.end());
52 EXPECT_TRUE(headers.find("sec-webSocket-version") == headers.end());
53 EXPECT_EQ("example.com", headers["host"]);
54 EXPECT_EQ("http://example.com", headers["origin"]);
55 EXPECT_EQ("sample", headers["sec-websocket-protocol"]);
56 EXPECT_EQ("foo", headers["sec-websocket-extensions"]);
57 EXPECT_EQ("ws", headers["scheme"]);
58 EXPECT_EQ("WebSocket/13", headers["version"]);
59
60 static const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
61
62 EXPECT_EQ(expected_challenge, challenge);
63
64 headers.clear();
65
66 headers["status"] = "101 Switching Protocols";
67 headers["sec-websocket-protocol"] = "sample";
68 headers["sec-websocket-extensions"] = "foo";
69
70 WebSocketHandshakeResponseHandler response_handler;
71 response_handler.set_protocol_version(13);
72 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
73 challenge,
74 2));
75 EXPECT_TRUE(response_handler.HasResponse());
76
77 // Note that order of sec-websocket-* is sensitive with hash_map order.
78 static const char kHandshakeResponseExpectedMessage[] =
79 "HTTP/1.1 101 Switching Protocols\r\n"
80 "Upgrade: websocket\r\n"
81 "Connection: Upgrade\r\n"
82 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
83 "sec-websocket-extensions: foo\r\n"
84 "sec-websocket-protocol: sample\r\n"
85 "\r\n";
86
87 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
88 }
89
90 TEST(WebSocketHandshakeHandlerSpdy2Test, RequestResponseWithCookies) {
91 WebSocketHandshakeRequestHandler request_handler;
92
93 // Note that websocket won't use multiple headers in request now.
94 static const char kHandshakeRequestMessage[] =
95 "GET /demo HTTP/1.1\r\n"
96 "Host: example.com\r\n"
97 "Upgrade: websocket\r\n"
98 "Connection: Upgrade\r\n"
99 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
100 "Origin: http://example.com\r\n"
101 "Sec-WebSocket-Protocol: sample\r\n"
102 "Sec-WebSocket-Extensions: foo\r\n"
103 "Sec-WebSocket-Version: 13\r\n"
104 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n"
105 "\r\n";
106
107 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
108 strlen(kHandshakeRequestMessage)));
109 EXPECT_EQ(13, request_handler.protocol_version());
110
111 GURL url("ws://example.com/demo");
112 std::string challenge;
113 SpdyHeaderBlock headers;
114 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
115 &headers,
116 &challenge,
117 2));
118
119 EXPECT_EQ(url.path(), headers["path"]);
120 EXPECT_TRUE(headers.find("upgrade") == headers.end());
121 EXPECT_TRUE(headers.find("Upgrade") == headers.end());
122 EXPECT_TRUE(headers.find("connection") == headers.end());
123 EXPECT_TRUE(headers.find("Connection") == headers.end());
124 EXPECT_TRUE(headers.find("Sec-WebSocket-Key") == headers.end());
125 EXPECT_TRUE(headers.find("sec-websocket-key") == headers.end());
126 EXPECT_TRUE(headers.find("Sec-WebSocket-Version") == headers.end());
127 EXPECT_TRUE(headers.find("sec-webSocket-version") == headers.end());
128 EXPECT_EQ("example.com", headers["host"]);
129 EXPECT_EQ("http://example.com", headers["origin"]);
130 EXPECT_EQ("sample", headers["sec-websocket-protocol"]);
131 EXPECT_EQ("foo", headers["sec-websocket-extensions"]);
132 EXPECT_EQ("ws", headers["scheme"]);
133 EXPECT_EQ("WebSocket/13", headers["version"]);
134 EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1",
135 headers["cookie"]);
136
137 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
138
139 EXPECT_EQ(expected_challenge, challenge);
140
141 headers.clear();
142
143 headers["status"] = "101 Switching Protocols";
144 headers["sec-websocket-protocol"] = "sample";
145 headers["sec-websocket-extensions"] = "foo";
146 std::string cookie = "WK-websocket-test=1";
147 cookie.append(1, '\0');
148 cookie += "WK-websocket-test-httponly=1; HttpOnly";
149 headers["set-cookie"] = cookie;
150
151
152 WebSocketHandshakeResponseHandler response_handler;
153 response_handler.set_protocol_version(13);
154 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
155 challenge,
156 2));
157 EXPECT_TRUE(response_handler.HasResponse());
158
159 // Note that order of sec-websocket-* is sensitive with hash_map order.
160 static const char kHandshakeResponseExpectedMessage[] =
161 "HTTP/1.1 101 Switching Protocols\r\n"
162 "Upgrade: websocket\r\n"
163 "Connection: Upgrade\r\n"
164 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
165 "sec-websocket-extensions: foo\r\n"
166 "sec-websocket-protocol: sample\r\n"
167 "set-cookie: WK-websocket-test=1\r\n"
168 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n"
169 "\r\n";
170
171 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
172 }
173
174 } // namespace
175
176 } // namespace net
OLDNEW
« no previous file with comments | « net/spdy/spdy_websocket_test_util.cc ('k') | net/websockets/websocket_handshake_handler_spdy3_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698