OLD | NEW |
| (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/test/embedded_test_server/http_request.h" | |
6 | |
7 #include "base/memory/scoped_ptr.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 namespace net { | |
11 namespace test_server { | |
12 | |
13 TEST(HttpRequestTest, ParseRequest) { | |
14 HttpRequestParser parser; | |
15 | |
16 // Process request in chunks to check if the parser deals with border cases. | |
17 // Also, check multi-line headers as well as multiple requests in the same | |
18 // chunk. This basically should cover all the simplest border cases. | |
19 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n"); | |
20 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
21 parser.ProcessChunk("Host: localhost:1234\r\n"); | |
22 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
23 parser.ProcessChunk("Multi-line-header: abcd\r\n"); | |
24 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
25 parser.ProcessChunk(" efgh\r\n"); | |
26 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
27 parser.ProcessChunk(" ijkl\r\n"); | |
28 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
29 parser.ProcessChunk("Content-Length: 10\r\n\r\n"); | |
30 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
31 // Content data and another request in the same chunk (possible in http/1.1). | |
32 parser.ProcessChunk("1234567890GET /another.html HTTP/1.1\r\n\r\n"); | |
33 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
34 | |
35 // Fetch the first request and validate it. | |
36 { | |
37 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
38 EXPECT_EQ("/foobar.html", request->relative_url); | |
39 EXPECT_EQ("POST", request->method_string); | |
40 EXPECT_EQ(METHOD_POST, request->method); | |
41 EXPECT_EQ("1234567890", request->content); | |
42 ASSERT_EQ(3u, request->headers.size()); | |
43 | |
44 EXPECT_EQ(1u, request->headers.count("Host")); | |
45 EXPECT_EQ(1u, request->headers.count("Multi-line-header")); | |
46 EXPECT_EQ(1u, request->headers.count("Content-Length")); | |
47 | |
48 const char kExpectedAllHeaders[] = | |
49 "POST /foobar.html HTTP/1.1\r\n" | |
50 "Host: localhost:1234\r\n" | |
51 "Multi-line-header: abcd\r\n" | |
52 " efgh\r\n" | |
53 " ijkl\r\n" | |
54 "Content-Length: 10\r\n"; | |
55 EXPECT_EQ(kExpectedAllHeaders, request->all_headers); | |
56 EXPECT_EQ("localhost:1234", request->headers["Host"]); | |
57 EXPECT_EQ("abcd efgh ijkl", request->headers["Multi-line-header"]); | |
58 EXPECT_EQ("10", request->headers["Content-Length"]); | |
59 } | |
60 | |
61 // No other request available yet since we do not support multiple requests | |
62 // per connection. | |
63 EXPECT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
64 } | |
65 | |
66 TEST(HttpRequestTest, ParseRequestWithEmptyBody) { | |
67 HttpRequestParser parser; | |
68 | |
69 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n"); | |
70 parser.ProcessChunk("Content-Length: 0\r\n\r\n"); | |
71 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
72 | |
73 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
74 EXPECT_EQ("", request->content); | |
75 EXPECT_TRUE(request->has_content); | |
76 EXPECT_EQ(1u, request->headers.count("Content-Length")); | |
77 EXPECT_EQ("0", request->headers["Content-Length"]); | |
78 } | |
79 | |
80 TEST(HttpRequestTest, ParseRequestWithChunkedBody) { | |
81 HttpRequestParser parser; | |
82 | |
83 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n"); | |
84 parser.ProcessChunk("Transfer-Encoding: chunked\r\n\r\n"); | |
85 parser.ProcessChunk("5\r\nhello\r\n"); | |
86 parser.ProcessChunk("1\r\n \r\n"); | |
87 parser.ProcessChunk("5\r\nworld\r\n"); | |
88 parser.ProcessChunk("0\r\n\r\n"); | |
89 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
90 | |
91 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
92 EXPECT_EQ("hello world", request->content); | |
93 EXPECT_TRUE(request->has_content); | |
94 EXPECT_EQ(1u, request->headers.count("Transfer-Encoding")); | |
95 EXPECT_EQ("chunked", request->headers["Transfer-Encoding"]); | |
96 } | |
97 | |
98 TEST(HttpRequestTest, ParseRequestWithChunkedBodySlow) { | |
99 HttpRequestParser parser; | |
100 | |
101 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n"); | |
102 parser.ProcessChunk("Transfer-Encoding: chunked\r\n\r\n"); | |
103 std::string chunked_body = "5\r\nhello\r\n0\r\n\r\n"; | |
104 | |
105 // Send one character at a time, and make the parser parse the request. | |
106 for (size_t i = 0; i < chunked_body.size(); i++) { | |
107 parser.ProcessChunk(chunked_body.substr(i, 1)); | |
108 // Except for the last pass, ParseRequest() should give WAITING. | |
109 if (i != chunked_body.size() - 1) { | |
110 ASSERT_EQ(HttpRequestParser::WAITING, parser.ParseRequest()); | |
111 } | |
112 } | |
113 // All chunked data has been sent, the last ParseRequest should give ACCEPTED. | |
114 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
115 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
116 EXPECT_EQ("hello", request->content); | |
117 EXPECT_TRUE(request->has_content); | |
118 EXPECT_EQ(1u, request->headers.count("Transfer-Encoding")); | |
119 EXPECT_EQ("chunked", request->headers["Transfer-Encoding"]); | |
120 } | |
121 | |
122 TEST(HttpRequestTest, ParseRequestWithoutBody) { | |
123 HttpRequestParser parser; | |
124 | |
125 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n\r\n"); | |
126 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
127 | |
128 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
129 EXPECT_EQ("", request->content); | |
130 EXPECT_FALSE(request->has_content); | |
131 } | |
132 | |
133 TEST(HttpRequestTest, ParseGet) { | |
134 HttpRequestParser parser; | |
135 | |
136 parser.ProcessChunk("GET /foobar.html HTTP/1.1\r\n\r\n"); | |
137 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | |
138 | |
139 scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
140 EXPECT_EQ("/foobar.html", request->relative_url); | |
141 EXPECT_EQ("GET", request->method_string); | |
142 EXPECT_EQ(METHOD_GET, request->method); | |
143 EXPECT_EQ("", request->content); | |
144 EXPECT_FALSE(request->has_content); | |
145 } | |
146 | |
147 } // namespace test_server | |
148 } // namespace net | |
OLD | NEW |