mirror of
https://github.com/aria2/aria2.git
synced 2025-01-04 09:03:46 +00:00
114 lines
3.4 KiB
C++
114 lines
3.4 KiB
C++
#include "MetalinkPostDownloadHandler.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "RequestGroup.h"
|
|
#include "Option.h"
|
|
#include "DownloadContext.h"
|
|
#include "FileEntry.h"
|
|
#include "PieceStorage.h"
|
|
#include "DiskAdaptor.h"
|
|
#include "RequestGroupCriteria.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MetalinkPostDownloadHandlerTest : public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(MetalinkPostDownloadHandlerTest);
|
|
CPPUNIT_TEST(testCanHandle_extension);
|
|
CPPUNIT_TEST(testCanHandle_contentType);
|
|
CPPUNIT_TEST(testGetNextRequestGroups);
|
|
CPPUNIT_TEST(testGetNextRequestGroups_withBaseUri);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
private:
|
|
std::shared_ptr<Option> option_;
|
|
|
|
public:
|
|
void setUp() { option_.reset(new Option()); }
|
|
|
|
void testCanHandle_extension();
|
|
void testCanHandle_contentType();
|
|
void testGetNextRequestGroups();
|
|
void testGetNextRequestGroups_withBaseUri();
|
|
};
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(MetalinkPostDownloadHandlerTest);
|
|
|
|
void MetalinkPostDownloadHandlerTest::testCanHandle_extension()
|
|
{
|
|
std::shared_ptr<DownloadContext> dctx(
|
|
new DownloadContext(0, 0, "test.metalink"));
|
|
RequestGroup rg(GroupId::create(), option_);
|
|
rg.setDownloadContext(dctx);
|
|
|
|
MetalinkPostDownloadHandler handler;
|
|
|
|
CPPUNIT_ASSERT(handler.canHandle(&rg));
|
|
|
|
dctx->getFirstFileEntry()->setPath("test.metalink2");
|
|
CPPUNIT_ASSERT(!handler.canHandle(&rg));
|
|
}
|
|
|
|
void MetalinkPostDownloadHandlerTest::testCanHandle_contentType()
|
|
{
|
|
std::shared_ptr<DownloadContext> dctx(new DownloadContext(0, 0, "test"));
|
|
dctx->getFirstFileEntry()->setContentType("application/metalink+xml");
|
|
RequestGroup rg(GroupId::create(), option_);
|
|
rg.setDownloadContext(dctx);
|
|
|
|
MetalinkPostDownloadHandler handler;
|
|
|
|
CPPUNIT_ASSERT(handler.canHandle(&rg));
|
|
|
|
dctx->getFirstFileEntry()->setContentType("application/octet-stream");
|
|
CPPUNIT_ASSERT(!handler.canHandle(&rg));
|
|
}
|
|
|
|
void MetalinkPostDownloadHandlerTest::testGetNextRequestGroups()
|
|
{
|
|
std::shared_ptr<DownloadContext> dctx(
|
|
new DownloadContext(1_k, 0, A2_TEST_DIR "/test.xml"));
|
|
RequestGroup rg(GroupId::create(), option_);
|
|
rg.setDownloadContext(dctx);
|
|
rg.initPieceStorage();
|
|
rg.getPieceStorage()->getDiskAdaptor()->enableReadOnly();
|
|
|
|
MetalinkPostDownloadHandler handler;
|
|
std::vector<std::shared_ptr<RequestGroup>> groups;
|
|
handler.getNextRequestGroups(groups, &rg);
|
|
#ifdef ENABLE_BITTORRENT
|
|
CPPUNIT_ASSERT_EQUAL((size_t)6 /* 5 + 1 torrent file download */,
|
|
groups.size());
|
|
#else
|
|
CPPUNIT_ASSERT_EQUAL((size_t)5, groups.size());
|
|
#endif // ENABLE_BITTORRENT
|
|
|
|
for (auto& nrg : groups) {
|
|
CPPUNIT_ASSERT_EQUAL(rg.getGID(), nrg->following());
|
|
}
|
|
}
|
|
|
|
void MetalinkPostDownloadHandlerTest::testGetNextRequestGroups_withBaseUri()
|
|
{
|
|
std::shared_ptr<DownloadContext> dctx(
|
|
new DownloadContext(1_k, 0, A2_TEST_DIR "/base_uri.xml"));
|
|
dctx->getFirstFileEntry()->addUri("http://base/dir/base_uri.xml");
|
|
RequestGroup rg(GroupId::create(), option_);
|
|
rg.setDownloadContext(dctx);
|
|
rg.initPieceStorage();
|
|
rg.getPieceStorage()->getDiskAdaptor()->enableReadOnly();
|
|
|
|
MetalinkPostDownloadHandler handler;
|
|
std::vector<std::shared_ptr<RequestGroup>> groups;
|
|
handler.getNextRequestGroups(groups, &rg);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, groups.size());
|
|
CPPUNIT_ASSERT_EQUAL(std::string("http://base/dir/example.ext"),
|
|
groups[0]
|
|
->getDownloadContext()
|
|
->getFirstFileEntry()
|
|
->getRemainingUris()[0]);
|
|
}
|
|
|
|
} // namespace aria2
|