12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184 |
- #include "RpcMethod.h"
- #include <cppunit/extensions/HelperMacros.h>
- #include "DownloadEngine.h"
- #include "SelectEventPoll.h"
- #include "Option.h"
- #include "RequestGroupMan.h"
- #include "RequestGroup.h"
- #include "RpcMethodImpl.h"
- #include "OptionParser.h"
- #include "OptionHandler.h"
- #include "RpcRequest.h"
- #include "RpcResponse.h"
- #include "prefs.h"
- #include "TestUtil.h"
- #include "DownloadContext.h"
- #include "FeatureConfig.h"
- #include "util.h"
- #include "array_fun.h"
- #include "download_helper.h"
- #include "FileEntry.h"
- #ifdef ENABLE_BITTORRENT
- # include "BtRegistry.h"
- # include "BtRuntime.h"
- # include "bittorrent_helper.h"
- #endif // ENABLE_BITTORRENT
- namespace aria2 {
- namespace rpc {
- class RpcMethodTest:public CppUnit::TestFixture {
- CPPUNIT_TEST_SUITE(RpcMethodTest);
- CPPUNIT_TEST(testAddUri);
- CPPUNIT_TEST(testAddUri_withoutUri);
- CPPUNIT_TEST(testAddUri_notUri);
- CPPUNIT_TEST(testAddUri_withBadOption);
- CPPUNIT_TEST(testAddUri_withPosition);
- CPPUNIT_TEST(testAddUri_withBadPosition);
- #ifdef ENABLE_BITTORRENT
- CPPUNIT_TEST(testAddTorrent);
- CPPUNIT_TEST(testAddTorrent_withoutTorrent);
- CPPUNIT_TEST(testAddTorrent_notBase64Torrent);
- CPPUNIT_TEST(testAddTorrent_withPosition);
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- CPPUNIT_TEST(testAddMetalink);
- CPPUNIT_TEST(testAddMetalink_withoutMetalink);
- CPPUNIT_TEST(testAddMetalink_notBase64Metalink);
- CPPUNIT_TEST(testAddMetalink_withPosition);
- #endif // ENABLE_METALINK
- CPPUNIT_TEST(testChangeOption);
- CPPUNIT_TEST(testChangeOption_withBadOption);
- CPPUNIT_TEST(testChangeOption_withNotAllowedOption);
- CPPUNIT_TEST(testChangeOption_withoutGid);
- CPPUNIT_TEST(testChangeGlobalOption);
- CPPUNIT_TEST(testChangeGlobalOption_withBadOption);
- CPPUNIT_TEST(testChangeGlobalOption_withNotAllowedOption);
- CPPUNIT_TEST(testTellStatus_withoutGid);
- CPPUNIT_TEST(testTellWaiting);
- CPPUNIT_TEST(testTellWaiting_fail);
- CPPUNIT_TEST(testGetVersion);
- CPPUNIT_TEST(testNoSuchMethod);
- CPPUNIT_TEST(testGatherStoppedDownload);
- CPPUNIT_TEST(testGatherProgressCommon);
- #ifdef ENABLE_BITTORRENT
- CPPUNIT_TEST(testGatherBitTorrentMetadata);
- #endif // ENABLE_BITTORRENT
- CPPUNIT_TEST(testChangePosition);
- CPPUNIT_TEST(testChangePosition_fail);
- CPPUNIT_TEST(testGetSessionInfo);
- CPPUNIT_TEST(testChangeUri);
- CPPUNIT_TEST(testChangeUri_fail);
- CPPUNIT_TEST(testPause);
- CPPUNIT_TEST(testSystemMulticall);
- CPPUNIT_TEST(testSystemMulticall_fail);
- CPPUNIT_TEST_SUITE_END();
- private:
- SharedHandle<DownloadEngine> e_;
- SharedHandle<Option> option_;
- public:
- void setUp()
- {
- RequestGroup::resetGIDCounter();
- option_.reset(new Option());
- option_->put(PREF_DIR, A2_TEST_OUT_DIR"/aria2_RpcMethodTest");
- option_->put(PREF_PIECE_LENGTH, "1048576");
- File(option_->get(PREF_DIR)).mkdirs();
- e_.reset
- (new DownloadEngine(SharedHandle<EventPoll>(new SelectEventPoll())));
- e_->setOption(option_.get());
- e_->setRequestGroupMan
- (SharedHandle<RequestGroupMan>
- (new RequestGroupMan(std::vector<SharedHandle<RequestGroup> >(),
- 1, option_.get())));
- }
- void testAddUri();
- void testAddUri_withoutUri();
- void testAddUri_notUri();
- void testAddUri_withBadOption();
- void testAddUri_withPosition();
- void testAddUri_withBadPosition();
- #ifdef ENABLE_BITTORRENT
- void testAddTorrent();
- void testAddTorrent_withoutTorrent();
- void testAddTorrent_notBase64Torrent();
- void testAddTorrent_withPosition();
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- void testAddMetalink();
- void testAddMetalink_withoutMetalink();
- void testAddMetalink_notBase64Metalink();
- void testAddMetalink_withPosition();
- #endif // ENABLE_METALINK
- void testChangeOption();
- void testChangeOption_withBadOption();
- void testChangeOption_withNotAllowedOption();
- void testChangeOption_withoutGid();
- void testChangeGlobalOption();
- void testChangeGlobalOption_withBadOption();
- void testChangeGlobalOption_withNotAllowedOption();
- void testTellStatus_withoutGid();
- void testTellWaiting();
- void testTellWaiting_fail();
- void testGetVersion();
- void testNoSuchMethod();
- void testGatherStoppedDownload();
- void testGatherProgressCommon();
- #ifdef ENABLE_BITTORRENT
- void testGatherBitTorrentMetadata();
- #endif // ENABLE_BITTORRENT
- void testChangePosition();
- void testChangePosition_fail();
- void testGetSessionInfo();
- void testChangeUri();
- void testChangeUri_fail();
- void testPause();
- void testSystemMulticall();
- void testSystemMulticall_fail();
- };
- CPPUNIT_TEST_SUITE_REGISTRATION(RpcMethodTest);
- namespace {
- std::string getString(const Dict* dict, const std::string& key)
- {
- return downcast<String>(dict->get(key))->s();
- }
- } // namespace
- void RpcMethodTest::testAddUri()
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam = List::g();
- urisParam->append("http://localhost/");
- req.params->append(urisParam);
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- const std::deque<SharedHandle<RequestGroup> > rgs =
- e_->getRequestGroupMan()->getReservedGroups();
- CPPUNIT_ASSERT_EQUAL((size_t)1, rgs.size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://localhost/"),
- rgs.front()->getDownloadContext()->
- getFirstFileEntry()->getRemainingUris().front());
- }
- // with options
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_DIR->k, "/sink");
- req.params->append(opt);
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL(std::string("/sink"),
- e_->getRequestGroupMan()->findReservedGroup(2)->
- getOption()->get(PREF_DIR));
- }
- }
- void RpcMethodTest::testAddUri_withoutUri()
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddUri_notUri()
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam = List::g();
- urisParam->append("not uri");
- req.params->append(urisParam);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddUri_withBadOption()
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam = List::g();
- urisParam->append("http://localhost");
- req.params->append(urisParam);
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_FILE_ALLOCATION->k, "badvalue");
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddUri_withPosition()
- {
- AddUriRpcMethod m;
- RpcRequest req1(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam1 = List::g();
- urisParam1->append("http://uri1");
- req1.params->append(urisParam1);
- RpcResponse res1 = m.execute(req1, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res1.code);
-
- RpcRequest req2(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam2 = List::g();
- urisParam2->append("http://uri2");
- req2.params->append(urisParam2);
- req2.params->append(Dict::g());
- req2.params->append(Integer::g(0));
- m.execute(req2, e_.get());
- std::string uri =
- e_->getRequestGroupMan()->getReservedGroups()[0]->
- getDownloadContext()->getFirstFileEntry()->getRemainingUris()[0];
- CPPUNIT_ASSERT_EQUAL(std::string("http://uri2"), uri);
- }
- void RpcMethodTest::testAddUri_withBadPosition()
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam = List::g();
- urisParam->append("http://localhost/");
- req.params->append(urisParam);
- req.params->append(Dict::g());
- req.params->append(Integer::g(-1));
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- #ifdef ENABLE_BITTORRENT
- void RpcMethodTest::testAddTorrent()
- {
- File(e_->getOption()->get(PREF_DIR)+
- "/0a3893293e27ac0490424c06de4d09242215f0a6.torrent").remove();
- AddTorrentRpcMethod m;
- RpcRequest req(AddTorrentRpcMethod::getMethodName(), List::g());
- req.params->append(readFile(A2_TEST_DIR"/single.torrent"));
- SharedHandle<List> uris = List::g();
- uris->append("http://localhost/aria2-0.8.2.tar.bz2");
- req.params->append(uris);
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT
- (File(e_->getOption()->get(PREF_DIR)+
- "/0a3893293e27ac0490424c06de4d09242215f0a6.torrent").exists());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL(std::string("1"), downcast<String>(res.param)->s());
- SharedHandle<RequestGroup> group =
- e_->getRequestGroupMan()->findReservedGroup(1);
- CPPUNIT_ASSERT(group);
- CPPUNIT_ASSERT_EQUAL(e_->getOption()->get(PREF_DIR)+"/aria2-0.8.2.tar.bz2",
- group->getFirstFilePath());
- CPPUNIT_ASSERT_EQUAL((size_t)1,
- group->getDownloadContext()->getFirstFileEntry()->
- getRemainingUris().size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://localhost/aria2-0.8.2.tar.bz2"),
- group->getDownloadContext()->getFirstFileEntry()->
- getRemainingUris()[0]);
- }
- // with options
- std::string dir = A2_TEST_OUT_DIR"/aria2_RpcMethodTest_testAddTorrent";
- File(dir).mkdirs();
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_DIR->k, dir);
- File(dir+"/0a3893293e27ac0490424c06de4d09242215f0a6.torrent").remove();
- req.params->append(opt);
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL
- (dir+"/aria2-0.8.2.tar.bz2",
- e_->getRequestGroupMan()->findReservedGroup(2)->getFirstFilePath());
- CPPUNIT_ASSERT
- (File(dir+"/0a3893293e27ac0490424c06de4d09242215f0a6.torrent").exists());
- }
- }
- void RpcMethodTest::testAddTorrent_withoutTorrent()
- {
- AddTorrentRpcMethod m;
- RpcRequest req(AddTorrentRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddTorrent_notBase64Torrent()
- {
- AddTorrentRpcMethod m;
- RpcRequest req(AddTorrentRpcMethod::getMethodName(), List::g());
- req.params->append("not torrent");
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddTorrent_withPosition()
- {
- AddTorrentRpcMethod m;
- RpcRequest req1(AddTorrentRpcMethod::getMethodName(), List::g());
- req1.params->append(readFile(A2_TEST_DIR"/test.torrent"));
- req1.params->append(List::g());
- req1.params->append(Dict::g());
- RpcResponse res1 = m.execute(req1, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res1.code);
- RpcRequest req2(AddTorrentRpcMethod::getMethodName(), List::g());
- req2.params->append(readFile(A2_TEST_DIR"/single.torrent"));
- req2.params->append(List::g());
- req2.params->append(Dict::g());
- req2.params->append(Integer::g(0));
- m.execute(req2, e_.get());
- CPPUNIT_ASSERT_EQUAL((size_t)1,
- e_->getRequestGroupMan()->getReservedGroups()[0]->
- getDownloadContext()->getFileEntries().size());
- }
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- void RpcMethodTest::testAddMetalink()
- {
- File(e_->getOption()->get(PREF_DIR)+
- "/c908634fbc257fd56f0114912c2772aeeb4064f4.meta4").remove();
- AddMetalinkRpcMethod m;
- RpcRequest req(AddMetalinkRpcMethod::getMethodName(), List::g());
- req.params->append(readFile(A2_TEST_DIR"/2files.metalink"));
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- const List* resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)2, resParams->size());
- CPPUNIT_ASSERT_EQUAL(std::string("1"), downcast<String>(resParams->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("2"), downcast<String>(resParams->get(1))->s());
- #ifdef ENABLE_MESSAGE_DIGEST
- CPPUNIT_ASSERT
- (File(e_->getOption()->get(PREF_DIR)+
- "/c908634fbc257fd56f0114912c2772aeeb4064f4.meta4").exists());
- #endif // ENABLE_MESSAGE_DIGEST
- SharedHandle<RequestGroup> tar =
- e_->getRequestGroupMan()->findReservedGroup(1);
- CPPUNIT_ASSERT(tar);
- CPPUNIT_ASSERT_EQUAL(e_->getOption()->get(PREF_DIR)+"/aria2-5.0.0.tar.bz2",
- tar->getFirstFilePath());
- SharedHandle<RequestGroup> deb =
- e_->getRequestGroupMan()->findReservedGroup(2);
- CPPUNIT_ASSERT(deb);
- CPPUNIT_ASSERT_EQUAL(e_->getOption()->get(PREF_DIR)+"/aria2-5.0.0.deb",
- deb->getFirstFilePath());
- }
- // with options
- std::string dir = A2_TEST_OUT_DIR"/aria2_RpcMethodTest_testAddMetalink";
- File(dir).mkdirs();
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_DIR->k, dir);
- File(dir+"/c908634fbc257fd56f0114912c2772aeeb4064f4.meta4").remove();
- req.params->append(opt);
- {
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL(dir+"/aria2-5.0.0.tar.bz2",
- e_->getRequestGroupMan()->findReservedGroup(3)->
- getFirstFilePath());
- #ifdef ENABLE_MESSAGE_DIGEST
- CPPUNIT_ASSERT
- (File(dir+"/c908634fbc257fd56f0114912c2772aeeb4064f4.meta4").exists());
- #endif // ENABLE_MESSAGE_DIGEST
- }
- }
- void RpcMethodTest::testAddMetalink_withoutMetalink()
- {
- AddMetalinkRpcMethod m;
- RpcRequest req(AddMetalinkRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddMetalink_notBase64Metalink()
- {
- AddMetalinkRpcMethod m;
- RpcRequest req(AddMetalinkRpcMethod::getMethodName(), List::g());
- req.params->append("not metalink");
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testAddMetalink_withPosition()
- {
- AddUriRpcMethod m1;
- RpcRequest req1(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam1 = List::g();
- urisParam1->append("http://uri");
- req1.params->append(urisParam1);
- RpcResponse res1 = m1.execute(req1, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res1.code);
- AddMetalinkRpcMethod m2;
- RpcRequest req2("ari2.addMetalink", List::g());
- req2.params->append(readFile(A2_TEST_DIR"/2files.metalink"));
- req2.params->append(Dict::g());
- req2.params->append(Integer::g(0));
- RpcResponse res2 = m2.execute(req2, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res2.code);
- CPPUNIT_ASSERT_EQUAL(e_->getOption()->get(PREF_DIR)+"/aria2-5.0.0.tar.bz2",
- e_->getRequestGroupMan()->getReservedGroups()[0]->
- getFirstFilePath());
- }
- #endif // ENABLE_METALINK
- void RpcMethodTest::testChangeOption()
- {
- SharedHandle<RequestGroup> group(new RequestGroup(option_));
- e_->getRequestGroupMan()->addReservedGroup(group);
- ChangeOptionRpcMethod m;
- RpcRequest req(ChangeOptionRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_MAX_DOWNLOAD_LIMIT->k, "100K");
- #ifdef ENABLE_BITTORRENT
- opt->put(PREF_BT_MAX_PEERS->k, "100");
- opt->put(PREF_BT_REQUEST_PEER_SPEED_LIMIT->k, "300K");
- opt->put(PREF_MAX_UPLOAD_LIMIT->k, "50K");
- SharedHandle<BtObject> btObject(new BtObject());
- btObject->btRuntime = SharedHandle<BtRuntime>(new BtRuntime());
- e_->getBtRegistry()->put(group->getGID(), btObject);
- #endif // ENABLE_BITTORRENT
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- SharedHandle<Option> option = group->getOption();
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL(100*1024,
- group->getMaxDownloadSpeedLimit());
- CPPUNIT_ASSERT_EQUAL(std::string("102400"),
- option->get(PREF_MAX_DOWNLOAD_LIMIT));
- #ifdef ENABLE_BITTORRENT
- CPPUNIT_ASSERT_EQUAL(std::string("307200"),
- option->get(PREF_BT_REQUEST_PEER_SPEED_LIMIT));
- CPPUNIT_ASSERT_EQUAL(std::string("100"), option->get(PREF_BT_MAX_PEERS));
- CPPUNIT_ASSERT_EQUAL(100, btObject->btRuntime->getMaxPeers());
- CPPUNIT_ASSERT_EQUAL(50*1024,
- group->getMaxUploadSpeedLimit());
- CPPUNIT_ASSERT_EQUAL(std::string("51200"),
- option->get(PREF_MAX_UPLOAD_LIMIT));
- #endif // ENABLE_BITTORRENT
- }
- void RpcMethodTest::testChangeOption_withBadOption()
- {
- SharedHandle<RequestGroup> group(new RequestGroup(option_));
- e_->getRequestGroupMan()->addReservedGroup(group);
- ChangeOptionRpcMethod m;
- RpcRequest req(ChangeOptionRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_MAX_DOWNLOAD_LIMIT->k, "badvalue");
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testChangeOption_withNotAllowedOption()
- {
- SharedHandle<RequestGroup> group(new RequestGroup(option_));
- e_->getRequestGroupMan()->addReservedGroup(group);
- ChangeOptionRpcMethod m;
- RpcRequest req(ChangeOptionRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_MAX_OVERALL_DOWNLOAD_LIMIT->k, "100K");
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- // The unacceptable options are just ignored.
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- void RpcMethodTest::testChangeOption_withoutGid()
- {
- ChangeOptionRpcMethod m;
- RpcRequest req(ChangeOptionRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testChangeGlobalOption()
- {
- ChangeGlobalOptionRpcMethod m;
- RpcRequest req
- (ChangeGlobalOptionRpcMethod::getMethodName(), List::g());
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_MAX_OVERALL_DOWNLOAD_LIMIT->k, "100K");
- #ifdef ENABLE_BITTORRENT
- opt->put(PREF_MAX_OVERALL_UPLOAD_LIMIT->k, "50K");
- #endif // ENABLE_BITTORRENT
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL
- (100*1024,
- e_->getRequestGroupMan()->getMaxOverallDownloadSpeedLimit());
- CPPUNIT_ASSERT_EQUAL(std::string("102400"),
- e_->getOption()->get(PREF_MAX_OVERALL_DOWNLOAD_LIMIT));
- #ifdef ENABLE_BITTORRENT
- CPPUNIT_ASSERT_EQUAL
- (50*1024,
- e_->getRequestGroupMan()->getMaxOverallUploadSpeedLimit());
- CPPUNIT_ASSERT_EQUAL(std::string("51200"),
- e_->getOption()->get(PREF_MAX_OVERALL_UPLOAD_LIMIT));
- #endif // ENABLE_BITTORRENT
- }
- void RpcMethodTest::testChangeGlobalOption_withBadOption()
- {
- ChangeGlobalOptionRpcMethod m;
- RpcRequest req
- (ChangeGlobalOptionRpcMethod::getMethodName(), List::g());
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_MAX_OVERALL_DOWNLOAD_LIMIT->k, "badvalue");
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testChangeGlobalOption_withNotAllowedOption()
- {
- ChangeGlobalOptionRpcMethod m;
- RpcRequest req
- (ChangeGlobalOptionRpcMethod::getMethodName(), List::g());
- SharedHandle<Dict> opt = Dict::g();
- opt->put(PREF_ENABLE_RPC->k, "100K");
- req.params->append(opt);
- RpcResponse res = m.execute(req, e_.get());
- // The unacceptable options are just ignored.
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- void RpcMethodTest::testNoSuchMethod()
- {
- NoSuchMethodRpcMethod m;
- RpcRequest req("make.hamburger", List::g());
- RpcResponse res = m.execute(req, 0);
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- CPPUNIT_ASSERT_EQUAL(std::string("No such method: make.hamburger"),
- getString(downcast<Dict>(res.param), "faultString"));
- }
- void RpcMethodTest::testTellStatus_withoutGid()
- {
- TellStatusRpcMethod m;
- RpcRequest req(TellStatusRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- namespace {
- void addUri(const std::string& uri, const SharedHandle<DownloadEngine>& e)
- {
- AddUriRpcMethod m;
- RpcRequest req(AddUriRpcMethod::getMethodName(), List::g());
- SharedHandle<List> urisParam = List::g();
- urisParam->append(uri);
- req.params->append(urisParam);
- CPPUNIT_ASSERT_EQUAL(0, m.execute(req, e.get()).code);
- }
- } // namespace
- #ifdef ENABLE_BITTORRENT
- namespace {
- void addTorrent
- (const std::string& torrentFile, const SharedHandle<DownloadEngine>& e)
- {
- AddTorrentRpcMethod m;
- RpcRequest req(AddTorrentRpcMethod::getMethodName(), List::g());
- req.params->append(readFile(torrentFile));
- RpcResponse res = m.execute(req, e.get());
- }
- } // namespace
- #endif // ENABLE_BITTORRENT
- void RpcMethodTest::testTellWaiting()
- {
- addUri("http://1/", e_);
- addUri("http://2/", e_);
- addUri("http://3/", e_);
- #ifdef ENABLE_BITTORRENT
- addTorrent(A2_TEST_DIR"/single.torrent", e_);
- #else // !ENABLE_BITTORRENT
- addUri("http://4/", e_);
- #endif // !ENABLE_BITTORRENT
- TellWaitingRpcMethod m;
- RpcRequest req(TellWaitingRpcMethod::getMethodName(), List::g());
- req.params->append(Integer::g(1));
- req.params->append(Integer::g(2));
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- const List* resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)2, resParams->size());
- CPPUNIT_ASSERT_EQUAL(std::string("2"),
- getString(downcast<Dict>(resParams->get(0)), "gid"));
- CPPUNIT_ASSERT_EQUAL(std::string("3"),
- getString(downcast<Dict>(resParams->get(1)), "gid"));
- // waiting.size() == offset+num
- req = RpcRequest(TellWaitingRpcMethod::getMethodName(), List::g());
- req.params->append(Integer::g(1));
- req.params->append(Integer::g(3));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)3, resParams->size());
- // waiting.size() < offset+num
- req = RpcRequest(TellWaitingRpcMethod::getMethodName(), List::g());
- req.params->append(Integer::g(1));
- req.params->append(Integer::g(4));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)3, resParams->size());
- // offset = INT32_MAX
- req.params->set(0, Integer::g(INT32_MAX));
- req.params->set(1, Integer::g(1));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)0, resParams->size());
- // num = INT32_MAX
- req.params->set(0, Integer::g(1));
- req.params->set(1, Integer::g(INT32_MAX));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)3, resParams->size());
- // offset=INT32_MAX and num = INT32_MAX
- req.params->set(0, Integer::g(INT32_MAX));
- req.params->set(1, Integer::g(INT32_MAX));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)0, resParams->size());
- // offset=INT32_MIN and num = INT32_MAX
- req.params->set(0, Integer::g(INT32_MIN));
- req.params->set(1, Integer::g(INT32_MAX));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)0, resParams->size());
- // negative offset
- req = RpcRequest(TellWaitingRpcMethod::getMethodName(), List::g());
- req.params->append(Integer::g(-1));
- req.params->append(Integer::g(2));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)2, resParams->size());
- CPPUNIT_ASSERT_EQUAL(std::string("4"),
- getString(downcast<Dict>(resParams->get(0)), "gid"));
- CPPUNIT_ASSERT_EQUAL(std::string("3"),
- getString(downcast<Dict>(resParams->get(1)), "gid"));
- // negative offset and size < num
- req.params->set(1, Integer::g(100));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)4, resParams->size());
- // nagative offset and normalized offset < 0
- req.params->set(0, Integer::g(-5));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)0, resParams->size());
- // nagative offset and normalized offset == 0
- req.params->set(0, Integer::g(-4));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)1, resParams->size());
- }
- void RpcMethodTest::testTellWaiting_fail()
- {
- TellWaitingRpcMethod m;
- RpcRequest req(TellWaitingRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testGetVersion()
- {
- GetVersionRpcMethod m;
- RpcRequest req(GetVersionRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- const Dict* resParams = downcast<Dict>(res.param);
- CPPUNIT_ASSERT_EQUAL(std::string(PACKAGE_VERSION),
- getString(resParams, "version"));
- const List* featureList = downcast<List>(resParams->get("enabledFeatures"));
- std::string features;
- for(List::ValueType::const_iterator i = featureList->begin();
- i != featureList->end(); ++i) {
- const String* s = downcast<String>(*i);
- features += s->s();
- features += ", ";
- }
-
- CPPUNIT_ASSERT_EQUAL(FeatureConfig::getInstance()->featureSummary()+", ",
- features);
- }
- void RpcMethodTest::testGatherStoppedDownload()
- {
- std::vector<SharedHandle<FileEntry> > fileEntries;
- std::vector<a2_gid_t> followedBy;
- followedBy.push_back(3);
- followedBy.push_back(4);
- SharedHandle<DownloadResult> d(new DownloadResult());
- d->gid = 1;
- d->fileEntries = fileEntries;
- d->inMemoryDownload = false;
- d->sessionDownloadLength = UINT64_MAX;
- d->sessionTime = 1000;
- d->result = error_code::FINISHED;
- d->followedBy = followedBy;
- d->belongsTo = 2;
- SharedHandle<Dict> entry = Dict::g();
- std::vector<std::string> keys;
- gatherStoppedDownload(entry, d, keys);
- const List* followedByRes = downcast<List>(entry->get("followedBy"));
- CPPUNIT_ASSERT_EQUAL(std::string("3"), downcast<String>(followedByRes->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("4"), downcast<String>(followedByRes->get(1))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("2"),
- downcast<String>(entry->get("belongsTo"))->s());
- keys.push_back("gid");
- entry = Dict::g();
- gatherStoppedDownload(entry, d, keys);
- CPPUNIT_ASSERT_EQUAL((size_t)1, entry->size());
- CPPUNIT_ASSERT(entry->containsKey("gid"));
- }
- void RpcMethodTest::testGatherProgressCommon()
- {
- SharedHandle<DownloadContext> dctx(new DownloadContext(0, 0,"aria2.tar.bz2"));
- std::string uris[] = { "http://localhost/aria2.tar.bz2" };
- dctx->getFirstFileEntry()->addUris(vbegin(uris), vend(uris));
- SharedHandle<RequestGroup> group(new RequestGroup(util::copy(option_)));
- group->setDownloadContext(dctx);
- std::vector<SharedHandle<RequestGroup> > followedBy;
- for(int i = 0; i < 2; ++i) {
- followedBy.push_back(SharedHandle<RequestGroup>(new RequestGroup(util::copy(option_))));
- }
- group->followedBy(followedBy.begin(), followedBy.end());
- group->belongsTo(2);
- SharedHandle<Dict> entry = Dict::g();
- std::vector<std::string> keys;
- gatherProgressCommon(entry, group, keys);
-
- const List* followedByRes = downcast<List>(entry->get("followedBy"));
- CPPUNIT_ASSERT_EQUAL(util::itos(followedBy[0]->getGID()),
- downcast<String>(followedByRes->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(util::itos(followedBy[1]->getGID()),
- downcast<String>(followedByRes->get(1))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("2"),
- downcast<String>(entry->get("belongsTo"))->s());
- const List* files = downcast<List>(entry->get("files"));
- CPPUNIT_ASSERT_EQUAL((size_t)1, files->size());
- const Dict* file = downcast<Dict>(files->get(0));
- CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"),
- downcast<String>(file->get("path"))->s());
- CPPUNIT_ASSERT_EQUAL(uris[0],
- downcast<String>
- (downcast<Dict>
- (downcast<List>(file->get("uris"))->get(0))
- ->get("uri"))
- ->s());
- CPPUNIT_ASSERT_EQUAL(e_->getOption()->get(PREF_DIR),
- downcast<String>(entry->get("dir"))->s());
- keys.push_back("gid");
- entry = Dict::g();
- gatherProgressCommon(entry, group, keys);
- CPPUNIT_ASSERT_EQUAL((size_t)1, entry->size());
- CPPUNIT_ASSERT(entry->containsKey("gid"));
-
- }
- #ifdef ENABLE_BITTORRENT
- void RpcMethodTest::testGatherBitTorrentMetadata()
- {
- SharedHandle<Option> option(new Option());
- option->put(PREF_DIR, ".");
- SharedHandle<DownloadContext> dctx(new DownloadContext());
- bittorrent::load(A2_TEST_DIR"/test.torrent", dctx, option);
- SharedHandle<Dict> btDict = Dict::g();
- gatherBitTorrentMetadata(btDict, bittorrent::getTorrentAttrs(dctx));
- CPPUNIT_ASSERT_EQUAL(std::string("REDNOAH.COM RULES"),
- downcast<String>(btDict->get("comment"))->s());
- CPPUNIT_ASSERT_EQUAL((int64_t)1123456789,
- downcast<Integer>(btDict->get("creationDate"))->i());
- CPPUNIT_ASSERT_EQUAL(std::string("multi"),
- downcast<String>(btDict->get("mode"))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("aria2-test"),
- downcast<String>
- (downcast<Dict>
- (btDict->get("info"))
- ->get("name"))
- ->s());
- const List* announceList = downcast<List>(btDict->get("announceList"));
- CPPUNIT_ASSERT_EQUAL((size_t)3, announceList->size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://tracker1"),
- downcast<String>(downcast<List>(announceList->get(0))->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("http://tracker2"),
- downcast<String>(downcast<List>(announceList->get(1))->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("http://tracker3"),
- downcast<String>(downcast<List>(announceList->get(2))->get(0))->s());
- // Remove some keys
- SharedHandle<TorrentAttribute> modBtAttrs = bittorrent::getTorrentAttrs(dctx);
- modBtAttrs->comment.clear();
- modBtAttrs->creationDate = 0;
- modBtAttrs->mode.clear();
- modBtAttrs->metadata.clear();
- btDict = Dict::g();
- gatherBitTorrentMetadata(btDict, modBtAttrs);
- CPPUNIT_ASSERT(!btDict->containsKey("comment"));
- CPPUNIT_ASSERT(!btDict->containsKey("creationDate"));
- CPPUNIT_ASSERT(!btDict->containsKey("mode"));
- CPPUNIT_ASSERT(!btDict->containsKey("info"));
- CPPUNIT_ASSERT(btDict->containsKey("announceList"));
- }
- #endif // ENABLE_BITTORRENT
- void RpcMethodTest::testChangePosition()
- {
- e_->getRequestGroupMan()->addReservedGroup
- (SharedHandle<RequestGroup>(new RequestGroup(util::copy(option_))));
- e_->getRequestGroupMan()->addReservedGroup
- (SharedHandle<RequestGroup>(new RequestGroup(util::copy(option_))));
- ChangePositionRpcMethod m;
- RpcRequest req(ChangePositionRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- req.params->append(Integer::g(1));
- req.params->append("POS_SET");
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL((int64_t)1, downcast<Integer>(res.param)->i());
- CPPUNIT_ASSERT_EQUAL
- ((a2_gid_t)1, e_->getRequestGroupMan()->getReservedGroups()[1]->getGID());
- }
- void RpcMethodTest::testChangePosition_fail()
- {
- ChangePositionRpcMethod m;
- RpcRequest req(ChangePositionRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->append("1");
- req.params->append(Integer::g(2));
- req.params->append("bad keyword");
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testChangeUri()
- {
- SharedHandle<FileEntry> files[3];
- for(int i = 0; i < 3; ++i) {
- files[i].reset(new FileEntry());
- }
- files[1]->addUri("http://example.org/aria2.tar.bz2");
- files[1]->addUri("http://example.org/mustremove1");
- files[1]->addUri("http://example.org/mustremove2");
- SharedHandle<DownloadContext> dctx(new DownloadContext());
- dctx->setFileEntries(&files[0], &files[3]);
- SharedHandle<RequestGroup> group(new RequestGroup(option_));
- group->setDownloadContext(dctx);
- e_->getRequestGroupMan()->addReservedGroup(group);
- ChangeUriRpcMethod m;
- RpcRequest req(ChangeUriRpcMethod::getMethodName(), List::g());
- req.params->append("1"); // GID
- req.params->append(Integer::g(2)); // index of FileEntry
- SharedHandle<List> removeuris = List::g();
- removeuris->append("http://example.org/mustremove1");
- removeuris->append("http://example.org/mustremove2");
- removeuris->append("http://example.org/notexist");
- req.params->append(removeuris);
- SharedHandle<List> adduris = List::g();
- adduris->append("http://example.org/added1");
- adduris->append("http://example.org/added2");
- adduris->append("baduri");
- adduris->append("http://example.org/added3");
- req.params->append(adduris);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL((int64_t)2, downcast<Integer>(downcast<List>(res.param)->get(0))->i());
- CPPUNIT_ASSERT_EQUAL((int64_t)3, downcast<Integer>(downcast<List>(res.param)->get(1))->i());
- CPPUNIT_ASSERT_EQUAL((size_t)0, files[0]->getRemainingUris().size());
- CPPUNIT_ASSERT_EQUAL((size_t)0, files[2]->getRemainingUris().size());
- std::deque<std::string> uris = files[1]->getRemainingUris();
- CPPUNIT_ASSERT_EQUAL((size_t)4, uris.size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/aria2.tar.bz2"),uris[0]);
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added1"), uris[1]);
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added2"), uris[2]);
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added3"), uris[3]);
- // Change adduris
- adduris = List::g();
- adduris->append("http://example.org/added1-1");
- adduris->append("http://example.org/added1-2");
- req.params->set(3, adduris);
- // Set position parameter
- req.params->append(Integer::g(2));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL((int64_t)0, downcast<Integer>(downcast<List>(res.param)->get(0))->i());
- CPPUNIT_ASSERT_EQUAL((int64_t)2, downcast<Integer>(downcast<List>(res.param)->get(1))->i());
- uris = files[1]->getRemainingUris();
- CPPUNIT_ASSERT_EQUAL((size_t)6, uris.size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added1-1"), uris[2]);
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added1-2"), uris[3]);
- // Change index of FileEntry
- req.params->set(1, Integer::g(1));
- // Set position far beyond the size of uris in FileEntry.
- req.params->set(4, Integer::g(1000));
- res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL((int64_t)0, downcast<Integer>(downcast<List>(res.param)->get(0))->i());
- CPPUNIT_ASSERT_EQUAL((int64_t)2, downcast<Integer>(downcast<List>(res.param)->get(1))->i());
- uris = files[0]->getRemainingUris();
- CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added1-1"), uris[0]);
- CPPUNIT_ASSERT_EQUAL(std::string("http://example.org/added1-2"), uris[1]);
- }
- void RpcMethodTest::testChangeUri_fail()
- {
- SharedHandle<FileEntry> files[3];
- for(int i = 0; i < 3; ++i) {
- files[i].reset(new FileEntry());
- }
- SharedHandle<DownloadContext> dctx(new DownloadContext());
- dctx->setFileEntries(&files[0], &files[3]);
- SharedHandle<RequestGroup> group(new RequestGroup(option_));
- group->setDownloadContext(dctx);
- e_->getRequestGroupMan()->addReservedGroup(group);
- ChangeUriRpcMethod m;
- RpcRequest req(ChangeUriRpcMethod::getMethodName(), List::g());
- req.params->append("1"); // GID
- req.params->append(Integer::g(1)); // index of FileEntry
- SharedHandle<List> removeuris = List::g();
- req.params->append(removeuris);
- SharedHandle<List> adduris = List::g();
- req.params->append(adduris);
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- req.params->set(1, Integer::g(0));
- res = m.execute(req, e_.get());
- // RPC request fails because 2nd argument is less than 1.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->set(1, Integer::g(1));
- req.params->set(0, String::g("2"));
- res = m.execute(req, e_.get());
- // RPC request fails because GID#2 does not exist.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->set(0, String::g("1"));
- req.params->set(1, Integer::g(4));
- res = m.execute(req, e_.get());
- // RPC request fails because FileEntry#3 does not exist.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->set(1, String::g("0"));
- res = m.execute(req, e_.get());
- // RPC request fails because index of FileEntry is string.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->set(1, Integer::g(1));
- req.params->set(2, String::g("http://url"));
- res = m.execute(req, e_.get());
- // RPC request fails because 3rd param is not list.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- req.params->set(2, List::g());
- req.params->set(3, String::g("http://url"));
- res = m.execute(req, e_.get());
- // RPC request fails because 4th param is not list.
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- void RpcMethodTest::testGetSessionInfo()
- {
- GetSessionInfoRpcMethod m;
- RpcRequest req(GetSessionInfoRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- CPPUNIT_ASSERT_EQUAL(util::toHex(e_->getSessionId()),
- getString(downcast<Dict>(res.param), "sessionId"));
- }
- void RpcMethodTest::testPause()
- {
- const std::string URIS[] = {
- "http://url1",
- "http://url2",
- "http://url3",
- };
- std::vector<std::string> uris(vbegin(URIS), vend(URIS));
- option_->put(PREF_FORCE_SEQUENTIAL, A2_V_TRUE);
- std::vector<SharedHandle<RequestGroup> > groups;
- createRequestGroupForUri(groups, option_, uris);
- CPPUNIT_ASSERT_EQUAL((size_t)3, groups.size());
- e_->getRequestGroupMan()->addReservedGroup(groups);
- {
- PauseRpcMethod m;
- RpcRequest req(PauseRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- CPPUNIT_ASSERT(groups[0]->isPauseRequested());
- {
- UnpauseRpcMethod m;
- RpcRequest req(UnpauseRpcMethod::getMethodName(), List::g());
- req.params->append("1");
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- CPPUNIT_ASSERT(!groups[0]->isPauseRequested());
- {
- PauseAllRpcMethod m;
- RpcRequest req(PauseAllRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- for(size_t i = 0; i < groups.size(); ++i) {
- CPPUNIT_ASSERT(groups[i]->isPauseRequested());
- }
- {
- UnpauseAllRpcMethod m;
- RpcRequest req(UnpauseAllRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- for(size_t i = 0; i < groups.size(); ++i) {
- CPPUNIT_ASSERT(!groups[i]->isPauseRequested());
- }
- {
- ForcePauseAllRpcMethod m;
- RpcRequest req(ForcePauseAllRpcMethod::getMethodName(), List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- }
- for(size_t i = 0; i < groups.size(); ++i) {
- CPPUNIT_ASSERT(groups[i]->isPauseRequested());
- }
- }
- void RpcMethodTest::testSystemMulticall()
- {
- SystemMulticallRpcMethod m;
- RpcRequest req("system.multicall", List::g());
- SharedHandle<List> reqparams = List::g();
- req.params->append(reqparams);
- for(int i = 0; i < 2; ++i) {
- SharedHandle<Dict> dict = Dict::g();
- dict->put("methodName", AddUriRpcMethod::getMethodName());
- SharedHandle<List> params = List::g();
- SharedHandle<List> urisParam = List::g();
- urisParam->append("http://localhost/"+util::itos(i));
- params->append(urisParam);
- dict->put("params", params);
- reqparams->append(dict);
- }
- {
- SharedHandle<Dict> dict = Dict::g();
- dict->put("methodName", "not exists");
- dict->put("params", List::g());
- reqparams->append(dict);
- }
- {
- reqparams->append("not struct");
- }
- {
- SharedHandle<Dict> dict = Dict::g();
- dict->put("methodName", "system.multicall");
- dict->put("params", List::g());
- reqparams->append(dict);
- }
- {
- // missing params
- SharedHandle<Dict> dict = Dict::g();
- dict->put("methodName", GetVersionRpcMethod::getMethodName());
- reqparams->append(dict);
- }
- {
- SharedHandle<Dict> dict = Dict::g();
- dict->put("methodName", GetVersionRpcMethod::getMethodName());
- dict->put("params", List::g());
- reqparams->append(dict);
- }
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(0, res.code);
- const List* resParams = downcast<List>(res.param);
- CPPUNIT_ASSERT_EQUAL((size_t)7, resParams->size());
- CPPUNIT_ASSERT_EQUAL(std::string("1"),
- downcast<String>(downcast<List>(resParams->get(0))->get(0))->s());
- CPPUNIT_ASSERT_EQUAL(std::string("2"),
- downcast<String>(downcast<List>(resParams->get(1))->get(0))->s());
- CPPUNIT_ASSERT_EQUAL((int64_t)1,
- downcast<Integer>
- (downcast<Dict>(resParams->get(2))->get("faultCode"))
- ->i());
- CPPUNIT_ASSERT_EQUAL((int64_t)1,
- downcast<Integer>
- (downcast<Dict>(resParams->get(3))->get("faultCode"))
- ->i());
- CPPUNIT_ASSERT_EQUAL((int64_t)1,
- downcast<Integer>
- (downcast<Dict>(resParams->get(4))->get("faultCode"))
- ->i());
- CPPUNIT_ASSERT(downcast<List>(resParams->get(5)));
- CPPUNIT_ASSERT(downcast<List>(resParams->get(6)));
- }
- void RpcMethodTest::testSystemMulticall_fail()
- {
- SystemMulticallRpcMethod m;
- RpcRequest req("system.multicall", List::g());
- RpcResponse res = m.execute(req, e_.get());
- CPPUNIT_ASSERT_EQUAL(1, res.code);
- }
- } // namespace rpc
- } // namespace aria2
|