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

Issues Search
    My Issues | Starred     Open | Closed | All

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

Issue 23614009: Rename websockets/*_unittest.cc to *_test.cc (Closed) Base URL: http://git.chromium.org/chromium/src.git@master
Patch Set: Created 7 years, 3 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
« no previous file with comments | « net/websockets/websocket_handshake_handler_spdy_test.cc ('k') | net/websockets/websocket_handshake_handler_test.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 "net/socket/next_proto.h"
10 #include "net/spdy/spdy_header_block.h"
11 #include "net/spdy/spdy_websocket_test_util.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13 #include "url/gurl.h"
14
15 namespace net {
16
17 namespace {
18
19 class WebSocketHandshakeHandlerSpdyTest
20 : public ::testing::Test,
21 public ::testing::WithParamInterface<NextProto> {
22 protected:
23 WebSocketHandshakeHandlerSpdyTest() : spdy_util_(GetParam()) {}
24
25 SpdyWebSocketTestUtil spdy_util_;
26 };
27
28 INSTANTIATE_TEST_CASE_P(
29 NextProto,
30 WebSocketHandshakeHandlerSpdyTest,
31 testing::Values(kProtoSPDY2, kProtoSPDY3, kProtoSPDY31, kProtoSPDY4a2,
32 kProtoHTTP2Draft04));
33
34 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponse) {
35 WebSocketHandshakeRequestHandler request_handler;
36
37 static const char kHandshakeRequestMessage[] =
38 "GET /demo HTTP/1.1\r\n"
39 "Host: example.com\r\n"
40 "Upgrade: websocket\r\n"
41 "Connection: Upgrade\r\n"
42 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
43 "Origin: http://example.com\r\n"
44 "Sec-WebSocket-Protocol: sample\r\n"
45 "Sec-WebSocket-Extensions: foo\r\n"
46 "Sec-WebSocket-Version: 13\r\n"
47 "\r\n";
48
49 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
50 strlen(kHandshakeRequestMessage)));
51 EXPECT_EQ(13, request_handler.protocol_version());
52
53 GURL url("ws://example.com/demo");
54 std::string challenge;
55 SpdyHeaderBlock headers;
56 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
57 &headers,
58 &challenge,
59 spdy_util_.spdy_version()));
60
61 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path"));
62 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty());
63 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty());
64 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty());
65 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty());
66 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty());
67 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty());
68 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty());
69 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty());
70 EXPECT_EQ("example.com", spdy_util_.GetHeader(headers, "host"));
71 EXPECT_EQ("http://example.com", spdy_util_.GetHeader(headers, "origin"));
72 EXPECT_EQ("sample", spdy_util_.GetHeader(headers, "sec-websocket-protocol"));
73 EXPECT_EQ("foo", spdy_util_.GetHeader(headers, "sec-websocket-extensions"));
74 EXPECT_EQ("ws", spdy_util_.GetHeader(headers, "scheme"));
75 EXPECT_EQ("WebSocket/13", spdy_util_.GetHeader(headers, "version"));
76
77 static const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
78
79 EXPECT_EQ(expected_challenge, challenge);
80
81 headers.clear();
82
83 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers);
84 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers);
85 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers);
86
87 WebSocketHandshakeResponseHandler response_handler;
88 response_handler.set_protocol_version(13);
89 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(
90 headers, challenge, spdy_util_.spdy_version()));
91 EXPECT_TRUE(response_handler.HasResponse());
92
93 // Note that order of sec-websocket-* is sensitive with hash_map order.
94 static const char kHandshakeResponseExpectedMessage[] =
95 "HTTP/1.1 101 Switching Protocols\r\n"
96 "Upgrade: websocket\r\n"
97 "Connection: Upgrade\r\n"
98 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
99 "sec-websocket-extensions: foo\r\n"
100 "sec-websocket-protocol: sample\r\n"
101 "\r\n";
102
103 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
104 }
105
106 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponseWithCookies) {
107 WebSocketHandshakeRequestHandler request_handler;
108
109 // Note that websocket won't use multiple headers in request now.
110 static const char kHandshakeRequestMessage[] =
111 "GET /demo HTTP/1.1\r\n"
112 "Host: example.com\r\n"
113 "Upgrade: websocket\r\n"
114 "Connection: Upgrade\r\n"
115 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
116 "Origin: http://example.com\r\n"
117 "Sec-WebSocket-Protocol: sample\r\n"
118 "Sec-WebSocket-Extensions: foo\r\n"
119 "Sec-WebSocket-Version: 13\r\n"
120 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n"
121 "\r\n";
122
123 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
124 strlen(kHandshakeRequestMessage)));
125 EXPECT_EQ(13, request_handler.protocol_version());
126
127 GURL url("ws://example.com/demo");
128 std::string challenge;
129 SpdyHeaderBlock headers;
130 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
131 &headers,
132 &challenge,
133 spdy_util_.spdy_version()));
134
135 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path"));
136 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty());
137 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty());
138 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty());
139 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty());
140 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty());
141 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty());
142 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty());
143 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty());
144 EXPECT_EQ("example.com", spdy_util_.GetHeader(headers, "host"));
145 EXPECT_EQ("http://example.com", spdy_util_.GetHeader(headers, "origin"));
146 EXPECT_EQ("sample", spdy_util_.GetHeader(headers, "sec-websocket-protocol"));
147 EXPECT_EQ("foo", spdy_util_.GetHeader(headers, "sec-websocket-extensions"));
148 EXPECT_EQ("ws", spdy_util_.GetHeader(headers, "scheme"));
149 EXPECT_EQ("WebSocket/13", spdy_util_.GetHeader(headers, "version"));
150 EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1",
151 headers["cookie"]);
152
153 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
154
155 EXPECT_EQ(expected_challenge, challenge);
156
157 headers.clear();
158
159 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers);
160 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers);
161 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers);
162 std::string cookie = "WK-websocket-test=1";
163 cookie.append(1, '\0');
164 cookie += "WK-websocket-test-httponly=1; HttpOnly";
165 headers["set-cookie"] = cookie;
166
167
168 WebSocketHandshakeResponseHandler response_handler;
169 response_handler.set_protocol_version(13);
170 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(
171 headers, challenge, spdy_util_.spdy_version()));
172 EXPECT_TRUE(response_handler.HasResponse());
173
174 // Note that order of sec-websocket-* is sensitive with hash_map order.
175 static const char kHandshakeResponseExpectedMessage[] =
176 "HTTP/1.1 101 Switching Protocols\r\n"
177 "Upgrade: websocket\r\n"
178 "Connection: Upgrade\r\n"
179 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
180 "sec-websocket-extensions: foo\r\n"
181 "sec-websocket-protocol: sample\r\n"
182 "set-cookie: WK-websocket-test=1\r\n"
183 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n"
184 "\r\n";
185
186 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
187 }
188
189 } // namespace
190
191 } // namespace net
OLDNEW
« no previous file with comments | « net/websockets/websocket_handshake_handler_spdy_test.cc ('k') | net/websockets/websocket_handshake_handler_test.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698