diff options
| author | Ross Smith II <aria2spam@smithii.com> | 2008-11-22 20:28:50 +0000 |
|---|---|---|
| committer | Ross Smith II <aria2spam@smithii.com> | 2008-11-22 20:28:50 +0000 |
| commit | 67d5d6d55efa708889dc41792bbfeb7983166790 (patch) | |
| tree | a27c58b96bd9d349f12c23e35611e324a326c09e /test/FtpConnectionTest.cc | |
| parent | 5faa91e24c7af82d1643f862295c413528ee8776 (diff) | |
Applied aria2-1.0.0-mingw-r5.patch
Diffstat (limited to 'test/FtpConnectionTest.cc')
| -rw-r--r-- | test/FtpConnectionTest.cc | 40 |
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()); } |
