aria2/test/RequestGroupTest.cc
Tatsuhiro Tsujikawa fca7b9d7e4 2007-12-12 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
If several protocols are available for a mirror, aria2 now use 
one of
	them. --metalink-preferred-protocol option was added to specify 
the
	preference of protocol.
	* src/AbstractCommand.cc
	* src/OptionHandlerFactory.cc
	* src/ServerHost.{h, cc}
	* src/Metalink2RequestGroup.cc
	* src/RequestGroup.{h, cc}
	* test/RequestGroupTest.cc
	* src/option_processing.cc
	* src/prefs.h
	* src/HttpResponseCommand.cc
	* src/MetalinkResource.{h, cc}
	* src/FtpNegotiationCommand.cc
	* src/MetalinkEntry.{h, cc}
	* src/MetalinkEntryTest.cc
2007-12-12 13:53:33 +00:00

71 lines
1.8 KiB
C++

#include "RequestGroup.h"
#include "ServerHost.h"
#include "Option.h"
#include <cppunit/extensions/HelperMacros.h>
using namespace std;
class RequestGroupTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(RequestGroupTest);
CPPUNIT_TEST(testRegisterSearchRemove);
CPPUNIT_TEST(testRemoveURIWhoseHostnameIs);
CPPUNIT_TEST_SUITE_END();
private:
public:
void setUp() {}
void testRegisterSearchRemove();
void testRemoveURIWhoseHostnameIs();
};
CPPUNIT_TEST_SUITE_REGISTRATION( RequestGroupTest );
void RequestGroupTest::testRegisterSearchRemove()
{
Option op;
RequestGroup rg(&op, Strings());
ServerHostHandle sv1 = new ServerHost(1, "localhost1");
ServerHostHandle sv2 = new ServerHost(2, "localhost2");
ServerHostHandle sv3 = new ServerHost(3, "localhost3");
rg.registerServerHost(sv3);
rg.registerServerHost(sv1);
rg.registerServerHost(sv2);
CPPUNIT_ASSERT(rg.searchServerHost(0).isNull());
{
ServerHostHandle sv = rg.searchServerHost(1);
CPPUNIT_ASSERT(!sv.isNull());
CPPUNIT_ASSERT_EQUAL(string("localhost1"), sv->getHostname());
}
rg.removeServerHost(1);
{
ServerHostHandle sv = rg.searchServerHost(1);
CPPUNIT_ASSERT(sv.isNull());
}
{
ServerHostHandle sv = rg.searchServerHost(2);
CPPUNIT_ASSERT(!sv.isNull());
CPPUNIT_ASSERT_EQUAL(string("localhost2"), sv->getHostname());
}
}
void RequestGroupTest::testRemoveURIWhoseHostnameIs()
{
const char* uris[] = { "http://localhost/aria2.zip",
"ftp://localhost/aria2.zip",
"http://mirror/aria2.zip" };
Option op;
RequestGroup rg(&op, Strings(&uris[0], &uris[3]));
rg.removeURIWhoseHostnameIs("localhost");
CPPUNIT_ASSERT_EQUAL((size_t)1, rg.getRemainingUris().size());
CPPUNIT_ASSERT_EQUAL(string("http://mirror/aria2.zip"),
rg.getRemainingUris()[0]);
}