| 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 |