mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 21:47:37 +03:00
2952abf064
Added following 2 keys, followedBy and belongsTo, to the response of tellStatus. followedBy: List of GIDs which are generated by the consequence of this download. For example, when aria2 downloaded Metalink file, it generates downloads described in it(see *--follow-metalink* option). This value is useful to track these auto generated downloads. If there is no such downloads, this key will not be included in the response. belongsTo: GID of a parent download. Some downloads are a part of another download. For example, if a file in Metalink has BitTorrent resource, the download of .torrent is a part of that file. If this download has no parent, this key will not be included in the response. * src/BtPostDownloadHandler.cc * src/DownloadResult.h * src/Metalink2RequestGroup.cc * src/MetalinkPostDownloadHandler.cc * src/RequestGroup.cc * src/RequestGroup.h * src/UTMetadataPostDownloadHandler.cc * src/XmlRpcMethodImpl.cc * src/XmlRpcMethodImpl.h
124 lines
3.9 KiB
C++
124 lines
3.9 KiB
C++
#include "Metalink2RequestGroup.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "DownloadContext.h"
|
|
#include "prefs.h"
|
|
#include "Option.h"
|
|
#include "RequestGroup.h"
|
|
#include "FileEntry.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class Metalink2RequestGroupTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(Metalink2RequestGroupTest);
|
|
CPPUNIT_TEST(testGenerate);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<Option> _option;
|
|
|
|
public:
|
|
void setUp()
|
|
{
|
|
_option.reset(new Option());
|
|
_option->put(PREF_SPLIT, "1");
|
|
}
|
|
|
|
void testGenerate();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( Metalink2RequestGroupTest );
|
|
|
|
void Metalink2RequestGroupTest::testGenerate()
|
|
{
|
|
std::deque<SharedHandle<RequestGroup> > groups;
|
|
_option->put(PREF_DIR, "/tmp");
|
|
Metalink2RequestGroup().generate(groups, "test.xml", _option);
|
|
// first file
|
|
{
|
|
SharedHandle<RequestGroup> rg = groups[0];
|
|
std::deque<std::string> uris;
|
|
rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
|
|
std::sort(uris.begin(), uris.end());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("ftp://ftphost/aria2-0.5.2.tar.bz2"), uris[0]);
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("http://httphost/aria2-0.5.2.tar.bz2"), uris[1]);
|
|
|
|
const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
|
|
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL((uint64_t)0ULL, dctx->getTotalLength());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
CPPUNIT_ASSERT_EQUAL(std::string("sha1"), dctx->getChecksumHashAlgo());
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("a96cf3f0266b91d87d5124cf94326422800b627d"),
|
|
dctx->getChecksum());
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
CPPUNIT_ASSERT(!dctx->getSignature().isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("pgp"), dctx->getSignature()->getType());
|
|
}
|
|
// second file
|
|
{
|
|
SharedHandle<RequestGroup> rg = groups[1];
|
|
std::deque<std::string> uris;
|
|
rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
|
|
|
|
const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
|
|
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
CPPUNIT_ASSERT_EQUAL(std::string("sha1"), dctx->getPieceHashAlgo());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, dctx->getPieceHashes().size());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)262144, dctx->getPieceLength());
|
|
CPPUNIT_ASSERT_EQUAL(std::string(""), dctx->getChecksumHashAlgo());
|
|
CPPUNIT_ASSERT_EQUAL(std::string(""), dctx->getChecksum());
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
CPPUNIT_ASSERT(dctx->getSignature().isNull());
|
|
}
|
|
|
|
#ifdef ENABLE_BITTORRENT
|
|
// fifth file <- downloading .torrent file
|
|
{
|
|
SharedHandle<RequestGroup> rg = groups[4];
|
|
std::deque<std::string> uris;
|
|
rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("http://host/torrent-http.integrated.torrent"), uris[0]);
|
|
const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
|
|
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(groups[5]->getGID(), rg->belongsTo());
|
|
}
|
|
#endif // ENABLE_BITTORRENT
|
|
|
|
// sixth file <- depends on fifth file to download .torrent file.
|
|
{
|
|
#ifdef ENABLE_BITTORRENT
|
|
SharedHandle<RequestGroup> rg = groups[5];
|
|
#else
|
|
SharedHandle<RequestGroup> rg = groups[4];
|
|
#endif // ENABLE_BITTORRENT
|
|
std::deque<std::string> uris;
|
|
rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("http://host/torrent-http.integrated"), uris[0]);
|
|
|
|
const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
|
|
|
|
CPPUNIT_ASSERT(!dctx.isNull());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
|
|
}
|
|
}
|
|
|
|
} // namespace aria2
|