Download-Tools/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-01-04 09:03:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
c5b83fed36
aria2/test/RequestGroupManTest.cc
Tatsuhiro Tsujikawa b1132d6b10 make clang-format using clang-format-3.6
2015-12-27 18:40:08 +09:00

284 lines
11 KiB
C++
Raw Blame History

#include "RequestGroupMan.h"
#include <fstream>
#include <cppunit/extensions/HelperMacros.h>
#include "TestUtil.h"
#include "prefs.h"
#include "DownloadContext.h"
#include "RequestGroup.h"
#include "Option.h"
#include "DownloadResult.h"
#include "FileEntry.h"
#include "ServerStatMan.h"
#include "ServerStat.h"
#include "File.h"
#include "array_fun.h"
#include "RecoverableException.h"
#include "util.h"
#include "DownloadEngine.h"
#include "SelectEventPoll.h"
#include "UriListParser.h"
namespace aria2 {
class RequestGroupManTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(RequestGroupManTest);
CPPUNIT_TEST(testIsSameFileBeingDownloaded);
CPPUNIT_TEST(testGetInitialCommands);
CPPUNIT_TEST(testLoadServerStat);
CPPUNIT_TEST(testSaveServerStat);
CPPUNIT_TEST(testChangeReservedGroupPosition);
CPPUNIT_TEST(testFillRequestGroupFromReserver);
CPPUNIT_TEST(testFillRequestGroupFromReserver_uriParser);
CPPUNIT_TEST(testInsertReservedGroup);
CPPUNIT_TEST(testAddDownloadResult);
CPPUNIT_TEST_SUITE_END();
private:
std::unique_ptr<DownloadEngine> e_;
std::shared_ptr<Option> option_;
RequestGroupMan* rgman_;
public:
void setUp()
{
option_ = std::make_shared<Option>();
option_->put(PREF_PIECE_LENGTH, "1048576");
// To enable paused RequestGroup
option_->put(PREF_ENABLE_RPC, A2_V_TRUE);
File(option_->get(PREF_DIR)).mkdirs();
e_ = make_unique<DownloadEngine>(make_unique<SelectEventPoll>());
e_->setOption(option_.get());
auto rgman = make_unique<RequestGroupMan>(
std::vector<std::shared_ptr<RequestGroup>>{}, 3, option_.get());
rgman_ = rgman.get();
e_->setRequestGroupMan(std::move(rgman));
}
void testIsSameFileBeingDownloaded();
void testGetInitialCommands();
void testLoadServerStat();
void testSaveServerStat();
void testChangeReservedGroupPosition();
void testFillRequestGroupFromReserver();
void testFillRequestGroupFromReserver_uriParser();
void testInsertReservedGroup();
void testAddDownloadResult();
};
CPPUNIT_TEST_SUITE_REGISTRATION(RequestGroupManTest);
void RequestGroupManTest::testIsSameFileBeingDownloaded()
{
std::shared_ptr<RequestGroup> rg1(
new RequestGroup(GroupId::create(), util::copy(option_)));
std::shared_ptr<RequestGroup> rg2(
new RequestGroup(GroupId::create(), util::copy(option_)));
std::shared_ptr<DownloadContext> dctx1(
new DownloadContext(0, 0, "aria2.tar.bz2"));
std::shared_ptr<DownloadContext> dctx2(
new DownloadContext(0, 0, "aria2.tar.bz2"));
rg1->setDownloadContext(dctx1);
rg2->setDownloadContext(dctx2);
RequestGroupMan gm(std::vector<std::shared_ptr<RequestGroup>>(), 1,
option_.get());
gm.addRequestGroup(rg1);
gm.addRequestGroup(rg2);
CPPUNIT_ASSERT(gm.isSameFileBeingDownloaded(rg1.get()));
dctx2->getFirstFileEntry()->setPath("aria2.tar.gz");
CPPUNIT_ASSERT(!gm.isSameFileBeingDownloaded(rg1.get()));
}
void RequestGroupManTest::testGetInitialCommands()
{
// TODO implement later
}
void RequestGroupManTest::testSaveServerStat()
{
RequestGroupMan rm(std::vector<std::shared_ptr<RequestGroup>>(), 0,
option_.get());
std::shared_ptr<ServerStat> ss_localhost(new ServerStat("localhost", "http"));
rm.addServerStat(ss_localhost);
File f(A2_TEST_OUT_DIR "/aria2_RequestGroupManTest_testSaveServerStat");
if (f.exists()) {
f.remove();
}
CPPUNIT_ASSERT(rm.saveServerStat(f.getPath()));
CPPUNIT_ASSERT(f.isFile());
f.remove();
CPPUNIT_ASSERT(f.mkdirs());
CPPUNIT_ASSERT(!rm.saveServerStat(f.getPath()));
}
void RequestGroupManTest::testLoadServerStat()
{
File f(A2_TEST_OUT_DIR "/aria2_RequestGroupManTest_testLoadServerStat");
std::ofstream o(f.getPath().c_str(), std::ios::binary);
o << "host=localhost, protocol=http, dl_speed=0, last_updated=1219505257,"
<< "status=OK";
o.close();
RequestGroupMan rm(std::vector<std::shared_ptr<RequestGroup>>(), 0,
option_.get());
std::cerr << "testLoadServerStat" << std::endl;
CPPUNIT_ASSERT(rm.loadServerStat(f.getPath()));
std::shared_ptr<ServerStat> ss_localhost =
rm.findServerStat("localhost", "http");
CPPUNIT_ASSERT(ss_localhost);
CPPUNIT_ASSERT_EQUAL(std::string("localhost"), ss_localhost->getHostname());
}
void RequestGroupManTest::testChangeReservedGroupPosition()
{
std::vector<std::shared_ptr<RequestGroup>> gs{
std::make_shared<RequestGroup>(GroupId::create(), util::copy(option_)),
std::make_shared<RequestGroup>(GroupId::create(), util::copy(option_)),
std::make_shared<RequestGroup>(GroupId::create(), util::copy(option_)),
std::make_shared<RequestGroup>(GroupId::create(), util::copy(option_))};
RequestGroupMan rm(gs, 0, option_.get());
CPPUNIT_ASSERT_EQUAL((size_t)0, rm.changeReservedGroupPosition(
gs[0]->getGID(), 0, OFFSET_MODE_SET));
CPPUNIT_ASSERT_EQUAL((size_t)1, rm.changeReservedGroupPosition(
gs[0]->getGID(), 1, OFFSET_MODE_SET));
CPPUNIT_ASSERT_EQUAL((size_t)3, rm.changeReservedGroupPosition(
gs[0]->getGID(), 10, OFFSET_MODE_SET));
CPPUNIT_ASSERT_EQUAL((size_t)0, rm.changeReservedGroupPosition(
gs[0]->getGID(), -10, OFFSET_MODE_SET));
CPPUNIT_ASSERT_EQUAL((size_t)1, rm.changeReservedGroupPosition(
gs[1]->getGID(), 0, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)2, rm.changeReservedGroupPosition(
gs[1]->getGID(), 1, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)1, rm.changeReservedGroupPosition(
gs[1]->getGID(), -1, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)0, rm.changeReservedGroupPosition(
gs[1]->getGID(), -10, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)1, rm.changeReservedGroupPosition(
gs[1]->getGID(), 1, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)3, rm.changeReservedGroupPosition(
gs[1]->getGID(), 10, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)1, rm.changeReservedGroupPosition(
gs[1]->getGID(), -2, OFFSET_MODE_CUR));
CPPUNIT_ASSERT_EQUAL((size_t)3, rm.changeReservedGroupPosition(
gs[3]->getGID(), 0, OFFSET_MODE_END));
CPPUNIT_ASSERT_EQUAL((size_t)2, rm.changeReservedGroupPosition(
gs[3]->getGID(), -1, OFFSET_MODE_END));
CPPUNIT_ASSERT_EQUAL((size_t)0, rm.changeReservedGroupPosition(
gs[3]->getGID(), -10, OFFSET_MODE_END));
CPPUNIT_ASSERT_EQUAL((size_t)3, rm.changeReservedGroupPosition(
gs[3]->getGID(), 10, OFFSET_MODE_END));
CPPUNIT_ASSERT_EQUAL((size_t)4, rm.getReservedGroups().size());
try {
rm.changeReservedGroupPosition(GroupId::create()->getNumericId(), 0,
OFFSET_MODE_CUR);
CPPUNIT_FAIL("exception must be thrown.");
}
catch (RecoverableException& e) {
// success
}
}
void RequestGroupManTest::testFillRequestGroupFromReserver()
{
std::shared_ptr<RequestGroup> rgs[] = {
createRequestGroup(0, 0, "foo1", "http://host/foo1", util::copy(option_)),
createRequestGroup(0, 0, "foo2", "http://host/foo2", util::copy(option_)),
createRequestGroup(0, 0, "foo3", "http://host/foo3", util::copy(option_)),
// Intentionally same path/URI for first RequestGroup and set
// length explicitly to do duplicate filename check.
createRequestGroup(0, 10, "foo1", "http://host/foo1",
util::copy(option_)),
createRequestGroup(0, 0, "foo4", "http://host/foo4", util::copy(option_)),
createRequestGroup(0, 0, "foo5", "http://host/foo5",
util::copy(option_))};
rgs[1]->setPauseRequested(true);
for (const auto& i : rgs) {
rgman_->addReservedGroup(i);
}
rgman_->fillRequestGroupFromReserver(e_.get());
CPPUNIT_ASSERT_EQUAL((size_t)2, rgman_->getReservedGroups().size());
}
void RequestGroupManTest::testFillRequestGroupFromReserver_uriParser()
{
std::shared_ptr<RequestGroup> rgs[] = {
createRequestGroup(0, 0, "mem1", "http://mem1", util::copy(option_)),
createRequestGroup(0, 0, "mem2", "http://mem2", util::copy(option_)),
};
rgs[0]->setPauseRequested(true);
for (const auto& i : rgs) {
rgman_->addReservedGroup(i);
}
std::shared_ptr<UriListParser> flp(
new UriListParser(A2_TEST_DIR "/filelist2.txt"));
rgman_->setUriListParser(flp);
rgman_->fillRequestGroupFromReserver(e_.get());
RequestGroupList::const_iterator itr;
CPPUNIT_ASSERT_EQUAL((size_t)1, rgman_->getReservedGroups().size());
itr = rgman_->getReservedGroups().begin();
CPPUNIT_ASSERT_EQUAL(rgs[0]->getGID(), (*itr)->getGID());
CPPUNIT_ASSERT_EQUAL((size_t)3, rgman_->getRequestGroups().size());
}
void RequestGroupManTest::testInsertReservedGroup()
{
std::vector<std::shared_ptr<RequestGroup>> rgs1{
std::shared_ptr<RequestGroup>(
new RequestGroup(GroupId::create(), util::copy(option_))),
std::shared_ptr<RequestGroup>(
new RequestGroup(GroupId::create(), util::copy(option_)))};
std::vector<std::shared_ptr<RequestGroup>> rgs2{
std::shared_ptr<RequestGroup>(
new RequestGroup(GroupId::create(), util::copy(option_))),
std::shared_ptr<RequestGroup>(
new RequestGroup(GroupId::create(), util::copy(option_)))};
rgman_->insertReservedGroup(0, rgs1);
CPPUNIT_ASSERT_EQUAL((size_t)2, rgman_->getReservedGroups().size());
RequestGroupList::const_iterator itr;
itr = rgman_->getReservedGroups().begin();
CPPUNIT_ASSERT_EQUAL(rgs1[0]->getGID(), (*itr++)->getGID());
CPPUNIT_ASSERT_EQUAL(rgs1[1]->getGID(), (*itr++)->getGID());
rgman_->insertReservedGroup(1, rgs2);
CPPUNIT_ASSERT_EQUAL((size_t)4, rgman_->getReservedGroups().size());
itr = rgman_->getReservedGroups().begin();
++itr;
CPPUNIT_ASSERT_EQUAL(rgs2[0]->getGID(), (*itr++)->getGID());
CPPUNIT_ASSERT_EQUAL(rgs2[1]->getGID(), (*itr++)->getGID());
}
void RequestGroupManTest::testAddDownloadResult()
{
std::string uri = "http://example.org";
rgman_->setMaxDownloadResult(3);
rgman_->addDownloadResult(createDownloadResult(error_code::TIME_OUT, uri));
rgman_->addDownloadResult(createDownloadResult(error_code::FINISHED, uri));
rgman_->addDownloadResult(createDownloadResult(error_code::FINISHED, uri));
rgman_->addDownloadResult(createDownloadResult(error_code::FINISHED, uri));
rgman_->addDownloadResult(createDownloadResult(error_code::FINISHED, uri));
CPPUNIT_ASSERT_EQUAL(error_code::TIME_OUT,
rgman_->getDownloadStat().getLastErrorResult());
}
} // namespace aria2
Reference in New Issue View Git Blame Copy Permalink