summaryrefslogtreecommitdiff
path: root/test/FtpConnectionTest.cc
diff options
context:
space:
mode:
authorRoss Smith II <aria2spam@smithii.com>2008-11-22 20:28:50 +0000
committerRoss Smith II <aria2spam@smithii.com>2008-11-22 20:28:50 +0000
commit67d5d6d55efa708889dc41792bbfeb7983166790 (patch)
treea27c58b96bd9d349f12c23e35611e324a326c09e /test/FtpConnectionTest.cc
parent5faa91e24c7af82d1643f862295c413528ee8776 (diff)
Applied aria2-1.0.0-mingw-r5.patch
Diffstat (limited to 'test/FtpConnectionTest.cc')
-rw-r--r--test/FtpConnectionTest.cc40
1 files changed, 35 insertions, 5 deletions
diff --git a/test/FtpConnectionTest.cc b/test/FtpConnectionTest.cc
index 6fb9699b..d11158b7 100644
--- a/test/FtpConnectionTest.cc
+++ b/test/FtpConnectionTest.cc
@@ -34,6 +34,7 @@ class FtpConnectionTest:public CppUnit::TestFixture {
private:
SharedHandle<SocketCore> _serverSocket;
uint16_t _listenPort;
+ SharedHandle<SocketCore> _clientSocket;
SharedHandle<FtpConnection> _ftp;
SharedHandle<Option> _option;
SharedHandle<AuthConfigFactory> _authConfigFactory;
@@ -54,14 +55,14 @@ public:
SharedHandle<Request> req(new Request());
req->setUrl("ftp://localhost/dir/file.img");
- SharedHandle<SocketCore> clientSocket(new SocketCore());
- clientSocket->establishConnection("127.0.0.1", _listenPort);
+ _clientSocket.reset(new SocketCore());
+ _clientSocket->establishConnection("127.0.0.1", _listenPort);
- while(!clientSocket->isWritable(0));
- clientSocket->setBlockingMode();
+ while(!_clientSocket->isWritable(0));
+ _clientSocket->setBlockingMode();
_serverSocket.reset(listenSocket->acceptConnection());
- _ftp.reset(new FtpConnection(1, clientSocket, req,
+ _ftp.reset(new FtpConnection(1, _clientSocket, req,
_authConfigFactory->createAuthConfig(req),
_option.get()));
}
@@ -83,35 +84,50 @@ public:
CPPUNIT_TEST_SUITE_REGISTRATION(FtpConnectionTest);
+static void waitRead(const SharedHandle<SocketCore>& socket)
+{
+ while(!socket->isReadable(0));
+}
+
void FtpConnectionTest::testReceiveResponse()
{
_serverSocket->writeData("100");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData(" single line response");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData("\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)100, _ftp->receiveResponse());
// 2 responses in the buffer
_serverSocket->writeData("101 single1\r\n"
"102 single2\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)101, _ftp->receiveResponse());
CPPUNIT_ASSERT_EQUAL((unsigned int)102, _ftp->receiveResponse());
_serverSocket->writeData("103-multi line response\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData("103-line2\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData("103");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData(" ");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
_serverSocket->writeData("last\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)103, _ftp->receiveResponse());
_serverSocket->writeData("104-multi\r\n"
"104 \r\n"
"105-multi\r\n"
"105 \r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)104, _ftp->receiveResponse());
CPPUNIT_ASSERT_EQUAL((unsigned int)105, _ftp->receiveResponse());
}
@@ -132,8 +148,10 @@ void FtpConnectionTest::testReceiveMdtmResponse()
{
Time t;
_serverSocket->writeData("213 20080908124312");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveMdtmResponse(t));
_serverSocket->writeData("\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)213, _ftp->receiveMdtmResponse(t));
CPPUNIT_ASSERT_EQUAL((time_t)1220877792, t.getTime());
}
@@ -141,6 +159,7 @@ void FtpConnectionTest::testReceiveMdtmResponse()
// see milli second part is ignored
Time t;
_serverSocket->writeData("213 20080908124312.014\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)213, _ftp->receiveMdtmResponse(t));
CPPUNIT_ASSERT_EQUAL((time_t)1220877792, t.getTime());
}
@@ -148,19 +167,24 @@ void FtpConnectionTest::testReceiveMdtmResponse()
// hhmmss part is missing
Time t;
_serverSocket->writeData("213 20080908\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)213, _ftp->receiveMdtmResponse(t));
CPPUNIT_ASSERT(t.bad());
}
{
+#ifdef HAVE_STRPTIME
// invalid month: 19
Time t;
_serverSocket->writeData("213 20081908124312\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)213, _ftp->receiveMdtmResponse(t));
CPPUNIT_ASSERT(t.bad());
+#endif
}
{
Time t;
_serverSocket->writeData("550 File Not Found\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)550, _ftp->receiveMdtmResponse(t));
}
}
@@ -172,9 +196,11 @@ void FtpConnectionTest::testReceiveResponse_overflow()
memcpy(data, "213 ", 4);
for(int i = 0; i < 4; ++i) {
_serverSocket->writeData(data, sizeof(data));
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receiveResponse());
}
_serverSocket->writeData(data, sizeof(data));
+ waitRead(_clientSocket);
try {
_ftp->receiveResponse();
CPPUNIT_FAIL("exception must be thrown.");
@@ -198,9 +224,11 @@ void FtpConnectionTest::testReceivePwdResponse()
{
std::string pwd;
_serverSocket->writeData("257 ");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)0, _ftp->receivePwdResponse(pwd));
CPPUNIT_ASSERT(pwd.empty());
_serverSocket->writeData("\"/dir/to\" is your directory.\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)257, _ftp->receivePwdResponse(pwd));
CPPUNIT_ASSERT_EQUAL(std::string("/dir/to"), pwd);
}
@@ -209,6 +237,7 @@ void FtpConnectionTest::testReceivePwdResponse_unquotedResponse()
{
std::string pwd;
_serverSocket->writeData("257 /dir/to\r\n");
+ waitRead(_clientSocket);
try {
_ftp->receivePwdResponse(pwd);
CPPUNIT_FAIL("exception must be thrown.");
@@ -221,6 +250,7 @@ void FtpConnectionTest::testReceivePwdResponse_badStatus()
{
std::string pwd;
_serverSocket->writeData("500 failed\r\n");
+ waitRead(_clientSocket);
CPPUNIT_ASSERT_EQUAL((unsigned int)500, _ftp->receivePwdResponse(pwd));
CPPUNIT_ASSERT(pwd.empty());
}