1 /*
2 * Copyright 2022 Google LLC
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "fcp/client/http/curl/curl_header_parser.h"
18
19 #include "fcp/client/http/http_client_util.h"
20 #include "fcp/testing/testing.h"
21
22 namespace fcp::client::http::curl {
23 namespace {
24
25 using ::testing::ElementsAre;
26 using ::testing::Pair;
27
TEST(CurlHeaderParserTest,Parse_HTTP_1_1_StatusCode)28 TEST(CurlHeaderParserTest, Parse_HTTP_1_1_StatusCode) {
29 CurlHeaderParser parser;
30 EXPECT_THAT(parser.GetStatusCode(), -1);
31 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
32
33 auto line1 = "HTTP/1.1 200 OK\r\n";
34 parser.ParseHeader(line1);
35
36 EXPECT_THAT(parser.GetStatusCode(), 200);
37 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
38
39 auto line2 = "HTTP/1.1 500 Internal Error\r\n";
40 parser.ParseHeader(line2);
41
42 EXPECT_THAT(parser.GetStatusCode(), 500);
43 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
44
45 auto line3 = "HTTP/1.1 123 Some:Custom Message\r\n";
46 parser.ParseHeader(line3);
47
48 EXPECT_THAT(parser.GetStatusCode(), 123);
49 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
50 }
51
TEST(CurlHeaderParserTest,Parse_HTTP_2_StatusCode)52 TEST(CurlHeaderParserTest, Parse_HTTP_2_StatusCode) {
53 CurlHeaderParser parser;
54 EXPECT_THAT(parser.GetStatusCode(), -1);
55 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
56
57 auto line1 = "HTTP/2 200\r\n";
58 parser.ParseHeader(line1);
59
60 EXPECT_THAT(parser.GetStatusCode(), 200);
61 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
62
63 auto line2 = "HTTP/2 500\r\n";
64 parser.ParseHeader(line2);
65
66 EXPECT_THAT(parser.GetStatusCode(), 500);
67 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
68 }
69
TEST(CurlHeaderParserTest,ParseHeaders)70 TEST(CurlHeaderParserTest, ParseHeaders) {
71 CurlHeaderParser parser;
72 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
73 EXPECT_THAT(parser.IsLastHeader(), false);
74
75 auto line = "HTTP/2 301\r\n"; // Redirect
76 parser.ParseHeader(line);
77 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
78 EXPECT_THAT(parser.IsLastHeader(), false);
79
80 line = "Content-Encoding: gzip\r\n";
81 parser.ParseHeader(line);
82 EXPECT_THAT(parser.GetHeaderList(),
83 ElementsAre(Pair(kContentEncodingHdr, "gzip")));
84 EXPECT_THAT(parser.IsLastHeader(), false);
85
86 line = "\r\n";
87 parser.ParseHeader(line);
88 EXPECT_THAT(parser.GetHeaderList(),
89 ElementsAre(Pair(kContentEncodingHdr, "gzip")));
90 EXPECT_THAT(parser.IsLastHeader(), false);
91
92 line = "HTTP/2 200\r\n"; // OK
93 parser.ParseHeader(line);
94 EXPECT_THAT(parser.GetHeaderList(), ElementsAre());
95 EXPECT_THAT(parser.IsLastHeader(), false);
96
97 line = "Content-Type: application/octet-stream\r\n";
98 parser.ParseHeader(line);
99 EXPECT_THAT(parser.GetHeaderList(),
100 ElementsAre(Pair("Content-Type", "application/octet-stream")));
101 EXPECT_THAT(parser.IsLastHeader(), false);
102
103 line = "Content-Length: 150\r\n";
104 parser.ParseHeader(line);
105 EXPECT_THAT(parser.GetHeaderList(),
106 ElementsAre(Pair("Content-Type", "application/octet-stream"),
107 Pair(kContentLengthHdr, "150")));
108 EXPECT_THAT(parser.IsLastHeader(), false);
109
110 line = "\r\n";
111 parser.ParseHeader(line);
112 EXPECT_THAT(parser.GetHeaderList(),
113 ElementsAre(Pair("Content-Type", "application/octet-stream"),
114 Pair(kContentLengthHdr, "150")));
115 EXPECT_THAT(parser.IsLastHeader(), true);
116 }
117 } // namespace
118 } // namespace fcp::client::http::curl
119