| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/test/embedded_test_server/http_request.h" | 5 #include "net/test/embedded_test_server/http_request.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 8 | 9 |
| 9 namespace net { | 10 namespace net { |
| 10 namespace test_server { | 11 namespace test_server { |
| 11 | 12 |
| 12 TEST(HttpRequestTest, ParseRequest) { | 13 TEST(HttpRequestTest, ParseRequest) { |
| 13 HttpRequestParser parser; | 14 HttpRequestParser parser; |
| 14 | 15 |
| 15 // Process request in chunks to check if the parser deals with border cases. | 16 // Process request in chunks to check if the parser deals with border cases. |
| 16 // Also, check multi-line headers as well as multiple requests in the same | 17 // Also, check multi-line headers as well as multiple requests in the same |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 69 parser.ProcessChunk("Content-Length: 0\r\n\r\n"); | 70 parser.ProcessChunk("Content-Length: 0\r\n\r\n"); |
| 70 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | 71 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); |
| 71 | 72 |
| 72 scoped_ptr<HttpRequest> request = parser.GetRequest(); | 73 scoped_ptr<HttpRequest> request = parser.GetRequest(); |
| 73 EXPECT_EQ("", request->content); | 74 EXPECT_EQ("", request->content); |
| 74 EXPECT_TRUE(request->has_content); | 75 EXPECT_TRUE(request->has_content); |
| 75 EXPECT_EQ(1u, request->headers.count("Content-Length")); | 76 EXPECT_EQ(1u, request->headers.count("Content-Length")); |
| 76 EXPECT_EQ("0", request->headers["Content-Length"]); | 77 EXPECT_EQ("0", request->headers["Content-Length"]); |
| 77 } | 78 } |
| 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 |
| 79 TEST(HttpRequestTest, ParseRequestWithoutBody) { | 122 TEST(HttpRequestTest, ParseRequestWithoutBody) { |
| 80 HttpRequestParser parser; | 123 HttpRequestParser parser; |
| 81 | 124 |
| 82 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n\r\n"); | 125 parser.ProcessChunk("POST /foobar.html HTTP/1.1\r\n\r\n"); |
| 83 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | 126 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); |
| 84 | 127 |
| 85 scoped_ptr<HttpRequest> request = parser.GetRequest(); | 128 scoped_ptr<HttpRequest> request = parser.GetRequest(); |
| 86 EXPECT_EQ("", request->content); | 129 EXPECT_EQ("", request->content); |
| 87 EXPECT_FALSE(request->has_content); | 130 EXPECT_FALSE(request->has_content); |
| 88 } | 131 } |
| 89 | 132 |
| 90 TEST(HttpRequestTest, ParseGet) { | 133 TEST(HttpRequestTest, ParseGet) { |
| 91 HttpRequestParser parser; | 134 HttpRequestParser parser; |
| 92 | 135 |
| 93 parser.ProcessChunk("GET /foobar.html HTTP/1.1\r\n\r\n"); | 136 parser.ProcessChunk("GET /foobar.html HTTP/1.1\r\n\r\n"); |
| 94 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); | 137 ASSERT_EQ(HttpRequestParser::ACCEPTED, parser.ParseRequest()); |
| 95 | 138 |
| 96 scoped_ptr<HttpRequest> request = parser.GetRequest(); | 139 scoped_ptr<HttpRequest> request = parser.GetRequest(); |
| 97 EXPECT_EQ("/foobar.html", request->relative_url); | 140 EXPECT_EQ("/foobar.html", request->relative_url); |
| 98 EXPECT_EQ("GET", request->method_string); | 141 EXPECT_EQ("GET", request->method_string); |
| 99 EXPECT_EQ(METHOD_GET, request->method); | 142 EXPECT_EQ(METHOD_GET, request->method); |
| 100 EXPECT_EQ("", request->content); | 143 EXPECT_EQ("", request->content); |
| 101 EXPECT_FALSE(request->has_content); | 144 EXPECT_FALSE(request->has_content); |
| 102 } | 145 } |
| 103 | 146 |
| 104 } // namespace test_server | 147 } // namespace test_server |
| 105 } // namespace net | 148 } // namespace net |
| OLD | NEW |