mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 13:37:40 +03:00
01fdb2aaeb
Try all available addresses returned by DNS until it gets connected in HTTP(S)/FTP download * src/AbstractCommand.cc * src/AbstractCommand.h * src/AbstractProxyRequestCommand.cc * src/AbstractProxyRequestCommand.h * src/DNSCache.h * src/DownloadEngine.cc * src/DownloadEngine.h * src/FtpInitiateConnectionCommand.cc * src/FtpInitiateConnectionCommand.h * src/FtpNegotiationCommand.cc * src/FtpNegotiationCommand.h * src/HttpInitiateConnectionCommand.cc * src/HttpInitiateConnectionCommand.h * src/HttpRequestCommand.cc * src/HttpRequestCommand.h * src/InitiateConnectionCommand.cc * src/InitiateConnectionCommand.h * test/DNSCacheTest.cc * test/Makefile.am * test/SimpleDNSCacheTest.cc
64 lines
1.6 KiB
C++
64 lines
1.6 KiB
C++
#include "DNSCache.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class DNSCacheTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(DNSCacheTest);
|
|
CPPUNIT_TEST(testFind);
|
|
CPPUNIT_TEST(testMarkBad);
|
|
CPPUNIT_TEST(testPutBadAddr);
|
|
CPPUNIT_TEST(testRemove);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
DNSCache _cache;
|
|
public:
|
|
void setUp()
|
|
{
|
|
_cache = DNSCache();
|
|
_cache.put("www", "192.168.0.1", 80);
|
|
_cache.put("www", "::1", 80);
|
|
_cache.put("ftp", "192.168.0.1", 21);
|
|
_cache.put("proxy", "192.168.1.2", 8080);
|
|
}
|
|
|
|
void testFind();
|
|
void testMarkBad();
|
|
void testPutBadAddr();
|
|
void testRemove();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(DNSCacheTest);
|
|
|
|
void DNSCacheTest::testFind()
|
|
{
|
|
CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), _cache.find("www", 80));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), _cache.find("ftp", 21));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("192.168.1.2"), _cache.find("proxy", 8080));
|
|
CPPUNIT_ASSERT_EQUAL(std::string(""), _cache.find("www", 8080));
|
|
CPPUNIT_ASSERT_EQUAL(std::string(""), _cache.find("another", 80));
|
|
}
|
|
|
|
void DNSCacheTest::testMarkBad()
|
|
{
|
|
_cache.markBad("www", "192.168.0.1", 80);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("::1"), _cache.find("www", 80));
|
|
}
|
|
|
|
void DNSCacheTest::testPutBadAddr()
|
|
{
|
|
_cache.markBad("www", "192.168.0.1", 80);
|
|
_cache.put("www", "192.168.0.1", 80);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("::1"), _cache.find("www", 80));
|
|
}
|
|
|
|
void DNSCacheTest::testRemove()
|
|
{
|
|
_cache.remove("www", 80);
|
|
CPPUNIT_ASSERT_EQUAL(std::string(""), _cache.find("www", 80));
|
|
}
|
|
|
|
} // namespace aria2
|