mirror of https://github.com/aria2/aria2
137 lines
4.2 KiB
C++
137 lines
4.2 KiB
C++
#include "BtDependency.h"
|
|
|
|
#include <iostream>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "SingleFileDownloadContext.h"
|
|
#include "DefaultPieceStorage.h"
|
|
#include "BtContext.h"
|
|
#include "RequestGroup.h"
|
|
#include "Option.h"
|
|
#include "Exception.h"
|
|
#include "SegmentMan.h"
|
|
#include "FileEntry.h"
|
|
#include "PieceSelector.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class BtDependencyTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtDependencyTest);
|
|
CPPUNIT_TEST(testResolve);
|
|
CPPUNIT_TEST(testResolve_loadError);
|
|
CPPUNIT_TEST(testResolve_dependeeFailure);
|
|
CPPUNIT_TEST(testResolve_dependeeInProgress);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
SharedHandle<RequestGroup> createDependant(const Option* option)
|
|
{
|
|
SharedHandle<RequestGroup> dependant(new RequestGroup(option, std::deque<std::string>()));
|
|
SharedHandle<SingleFileDownloadContext> dctx
|
|
(new SingleFileDownloadContext(0, 0, "/tmp/outfile.path"));
|
|
dctx->setDir("/tmp");
|
|
dependant->setDownloadContext(dctx);
|
|
return dependant;
|
|
}
|
|
|
|
SharedHandle<RequestGroup>
|
|
createDependee(const Option* option, const std::string& torrentFile, int64_t length)
|
|
{
|
|
SharedHandle<RequestGroup> dependee
|
|
(new RequestGroup(option, std::deque<std::string>()));
|
|
SharedHandle<SingleFileDownloadContext> dctx
|
|
(new SingleFileDownloadContext(1024*1024, length, torrentFile));
|
|
dctx->setDir(".");
|
|
dependee->setDownloadContext(dctx);
|
|
DefaultPieceStorageHandle ps(new DefaultPieceStorage(dctx, option));
|
|
dependee->setPieceStorage(ps);
|
|
ps->initStorage();
|
|
dependee->initSegmentMan();
|
|
return dependee;
|
|
}
|
|
|
|
public:
|
|
void setUp() {}
|
|
|
|
void testResolve();
|
|
void testResolve_loadError();
|
|
void testResolve_dependeeFailure();
|
|
void testResolve_dependeeInProgress();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( BtDependencyTest );
|
|
|
|
void BtDependencyTest::testResolve()
|
|
{
|
|
std::string filename = "test.torrent";
|
|
Option option;
|
|
SharedHandle<RequestGroup> dependant = createDependant(&option);
|
|
SharedHandle<RequestGroup> dependee = createDependee(&option, filename, File(filename).size());
|
|
dependee->getPieceStorage()->markAllPiecesDone();
|
|
|
|
BtDependency dep(dependant, dependee, &option);
|
|
CPPUNIT_ASSERT(dep.resolve());
|
|
|
|
SharedHandle<BtContext> btContext
|
|
(dynamic_pointer_cast<BtContext>(dependant->getDownloadContext()));
|
|
CPPUNIT_ASSERT(!btContext.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp/outfile.path"),
|
|
btContext->getActualBasePath());
|
|
}
|
|
|
|
void BtDependencyTest::testResolve_loadError()
|
|
{
|
|
try {
|
|
Option option;
|
|
SharedHandle<RequestGroup> dependant = createDependant(&option);
|
|
SharedHandle<RequestGroup> dependee = createDependee(&option, "notExist", 40);
|
|
dependee->getPieceStorage()->markAllPiecesDone();
|
|
|
|
BtDependency dep(dependant, dependee, &option);
|
|
CPPUNIT_ASSERT(dep.resolve());
|
|
|
|
SharedHandle<SingleFileDownloadContext> dctx
|
|
(dynamic_pointer_cast<SingleFileDownloadContext>
|
|
(dependant->getDownloadContext()));
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp/outfile.path"),
|
|
dctx->getActualBasePath());
|
|
} catch(Exception& e) {
|
|
std::cerr << e.stackTrace() << std::endl;
|
|
CPPUNIT_FAIL("an exception was thrown.");
|
|
}
|
|
}
|
|
|
|
void BtDependencyTest::testResolve_dependeeFailure()
|
|
{
|
|
Option option;
|
|
SharedHandle<RequestGroup> dependant = createDependant(&option);
|
|
SharedHandle<RequestGroup> dependee = createDependee(&option, "notExist", 40);
|
|
|
|
BtDependency dep(dependant, dependee, &option);
|
|
CPPUNIT_ASSERT(dep.resolve());
|
|
|
|
SharedHandle<SingleFileDownloadContext> dctx
|
|
(dynamic_pointer_cast<SingleFileDownloadContext>
|
|
(dependant->getDownloadContext()));
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp/outfile.path"),
|
|
dctx->getActualBasePath());
|
|
}
|
|
|
|
void BtDependencyTest::testResolve_dependeeInProgress()
|
|
{
|
|
std::string filename = "test.torrent";
|
|
Option option;
|
|
SharedHandle<RequestGroup> dependant = createDependant(&option);
|
|
SharedHandle<RequestGroup> dependee = createDependee(&option, filename, File(filename).size());
|
|
dependee->increaseNumCommand();
|
|
|
|
BtDependency dep(dependant, dependee, &option);
|
|
CPPUNIT_ASSERT(!dep.resolve());
|
|
}
|
|
|
|
} // namespace aria2
|