OLD | NEW |
| (Empty) |
1 // Copyright (c) 2009 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/ftp/ftp_ctrl_response_buffer.h" | |
6 | |
7 #include <string.h> | |
8 | |
9 #include "net/base/net_errors.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | |
11 | |
12 namespace { | |
13 | |
14 class FtpCtrlResponseBufferTest : public testing::Test { | |
15 public: | |
16 FtpCtrlResponseBufferTest() : buffer_(net::BoundNetLog()) { | |
17 } | |
18 | |
19 protected: | |
20 int PushDataToBuffer(const char* data) { | |
21 return buffer_.ConsumeData(data, strlen(data)); | |
22 } | |
23 | |
24 net::FtpCtrlResponseBuffer buffer_; | |
25 }; | |
26 | |
27 TEST_F(FtpCtrlResponseBufferTest, Basic) { | |
28 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
29 | |
30 EXPECT_EQ(net::OK, PushDataToBuffer("200 Status Text\r\n")); | |
31 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
32 | |
33 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
34 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
35 EXPECT_EQ(200, response.status_code); | |
36 ASSERT_EQ(1U, response.lines.size()); | |
37 EXPECT_EQ("Status Text", response.lines[0]); | |
38 } | |
39 | |
40 TEST_F(FtpCtrlResponseBufferTest, Chunks) { | |
41 EXPECT_EQ(net::OK, PushDataToBuffer("20")); | |
42 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
43 EXPECT_EQ(net::OK, PushDataToBuffer("0 Status")); | |
44 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
45 EXPECT_EQ(net::OK, PushDataToBuffer(" Text")); | |
46 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
47 EXPECT_EQ(net::OK, PushDataToBuffer("\r")); | |
48 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
49 EXPECT_EQ(net::OK, PushDataToBuffer("\n")); | |
50 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
51 | |
52 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
53 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
54 EXPECT_EQ(200, response.status_code); | |
55 ASSERT_EQ(1U, response.lines.size()); | |
56 EXPECT_EQ("Status Text", response.lines[0]); | |
57 } | |
58 | |
59 TEST_F(FtpCtrlResponseBufferTest, Continuation) { | |
60 EXPECT_EQ(net::OK, PushDataToBuffer("230-FirstLine\r\n")); | |
61 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
62 | |
63 EXPECT_EQ(net::OK, PushDataToBuffer("230-SecondLine\r\n")); | |
64 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
65 | |
66 EXPECT_EQ(net::OK, PushDataToBuffer("230 LastLine\r\n")); | |
67 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
68 | |
69 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
70 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
71 EXPECT_EQ(230, response.status_code); | |
72 ASSERT_EQ(3U, response.lines.size()); | |
73 EXPECT_EQ("FirstLine", response.lines[0]); | |
74 EXPECT_EQ("SecondLine", response.lines[1]); | |
75 EXPECT_EQ("LastLine", response.lines[2]); | |
76 } | |
77 | |
78 TEST_F(FtpCtrlResponseBufferTest, MultilineContinuation) { | |
79 EXPECT_EQ(net::OK, PushDataToBuffer("230-FirstLine\r\n")); | |
80 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
81 | |
82 EXPECT_EQ(net::OK, PushDataToBuffer("Continued\r\n")); | |
83 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
84 | |
85 EXPECT_EQ(net::OK, PushDataToBuffer("230-SecondLine\r\n")); | |
86 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
87 | |
88 EXPECT_EQ(net::OK, PushDataToBuffer("215 Continued\r\n")); | |
89 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
90 | |
91 EXPECT_EQ(net::OK, PushDataToBuffer("230 LastLine\r\n")); | |
92 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
93 | |
94 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
95 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
96 EXPECT_EQ(230, response.status_code); | |
97 ASSERT_EQ(3U, response.lines.size()); | |
98 EXPECT_EQ("FirstLineContinued", response.lines[0]); | |
99 EXPECT_EQ("SecondLine215 Continued", response.lines[1]); | |
100 EXPECT_EQ("LastLine", response.lines[2]); | |
101 } | |
102 | |
103 TEST_F(FtpCtrlResponseBufferTest, MultilineContinuationZeroLength) { | |
104 // For the corner case from bug 29322. | |
105 EXPECT_EQ(net::OK, PushDataToBuffer("230-\r\n")); | |
106 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
107 | |
108 EXPECT_EQ(net::OK, PushDataToBuffer("example.com\r\n")); | |
109 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
110 | |
111 EXPECT_EQ(net::OK, PushDataToBuffer("230 LastLine\r\n")); | |
112 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
113 | |
114 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
115 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
116 EXPECT_EQ(230, response.status_code); | |
117 ASSERT_EQ(2U, response.lines.size()); | |
118 EXPECT_EQ("example.com", response.lines[0]); | |
119 EXPECT_EQ("LastLine", response.lines[1]); | |
120 } | |
121 | |
122 TEST_F(FtpCtrlResponseBufferTest, SimilarContinuation) { | |
123 EXPECT_EQ(net::OK, PushDataToBuffer("230-FirstLine\r\n")); | |
124 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
125 | |
126 // Notice the space at the start of the line. It should be recognized | |
127 // as a continuation, and not the last line. | |
128 EXPECT_EQ(net::OK, PushDataToBuffer(" 230 Continued\r\n")); | |
129 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
130 | |
131 EXPECT_EQ(net::OK, PushDataToBuffer("230 TrueLastLine\r\n")); | |
132 EXPECT_TRUE(buffer_.ResponseAvailable()); | |
133 | |
134 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
135 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
136 EXPECT_EQ(230, response.status_code); | |
137 ASSERT_EQ(2U, response.lines.size()); | |
138 EXPECT_EQ("FirstLine 230 Continued", response.lines[0]); | |
139 EXPECT_EQ("TrueLastLine", response.lines[1]); | |
140 } | |
141 | |
142 // The nesting of multi-line responses is not allowed. | |
143 TEST_F(FtpCtrlResponseBufferTest, NoNesting) { | |
144 EXPECT_EQ(net::OK, PushDataToBuffer("230-FirstLine\r\n")); | |
145 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
146 | |
147 EXPECT_EQ(net::OK, PushDataToBuffer("300-Continuation\r\n")); | |
148 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
149 | |
150 EXPECT_EQ(net::OK, PushDataToBuffer("300 Still continuation\r\n")); | |
151 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
152 | |
153 EXPECT_EQ(net::OK, PushDataToBuffer("230 Real End\r\n")); | |
154 ASSERT_TRUE(buffer_.ResponseAvailable()); | |
155 | |
156 net::FtpCtrlResponse response = buffer_.PopResponse(); | |
157 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
158 EXPECT_EQ(230, response.status_code); | |
159 ASSERT_EQ(2U, response.lines.size()); | |
160 EXPECT_EQ("FirstLine300-Continuation300 Still continuation", | |
161 response.lines[0]); | |
162 EXPECT_EQ("Real End", response.lines[1]); | |
163 } | |
164 | |
165 TEST_F(FtpCtrlResponseBufferTest, NonNumericResponse) { | |
166 EXPECT_EQ(net::ERR_INVALID_RESPONSE, PushDataToBuffer("Non-numeric\r\n")); | |
167 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
168 } | |
169 | |
170 TEST_F(FtpCtrlResponseBufferTest, OutOfRangeResponse) { | |
171 EXPECT_EQ(net::ERR_INVALID_RESPONSE, PushDataToBuffer("777 OK?\r\n")); | |
172 EXPECT_FALSE(buffer_.ResponseAvailable()); | |
173 } | |
174 | |
175 } // namespace | |
OLD | NEW |