XmlRpcMethodTest.cc 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629
  1. #include "XmlRpcMethod.h"
  2. #include <cppunit/extensions/HelperMacros.h>
  3. #include "DownloadEngine.h"
  4. #include "SelectEventPoll.h"
  5. #include "Option.h"
  6. #include "RequestGroupMan.h"
  7. #include "ServerStatMan.h"
  8. #include "RequestGroup.h"
  9. #include "XmlRpcMethodImpl.h"
  10. #include "BDE.h"
  11. #include "OptionParser.h"
  12. #include "OptionHandler.h"
  13. #include "XmlRpcRequest.h"
  14. #include "XmlRpcResponse.h"
  15. #include "prefs.h"
  16. #include "TestUtil.h"
  17. #include "DownloadContext.h"
  18. #include "FeatureConfig.h"
  19. #ifdef ENABLE_BITTORRENT
  20. # include "BtRegistry.h"
  21. # include "BtRuntime.h"
  22. # include "PieceStorage.h"
  23. # include "PeerStorage.h"
  24. # include "BtProgressInfoFile.h"
  25. # include "BtAnnounce.h"
  26. #endif // ENABLE_BITTORRENT
  27. namespace aria2 {
  28. namespace xmlrpc {
  29. class XmlRpcMethodTest:public CppUnit::TestFixture {
  30. CPPUNIT_TEST_SUITE(XmlRpcMethodTest);
  31. CPPUNIT_TEST(testAddUri);
  32. CPPUNIT_TEST(testAddUri_withoutUri);
  33. CPPUNIT_TEST(testAddUri_notUri);
  34. CPPUNIT_TEST(testAddUri_withBadOption);
  35. CPPUNIT_TEST(testAddUri_withPosition);
  36. CPPUNIT_TEST(testAddUri_withBadPosition);
  37. #ifdef ENABLE_BITTORRENT
  38. CPPUNIT_TEST(testAddTorrent);
  39. CPPUNIT_TEST(testAddTorrent_withoutTorrent);
  40. CPPUNIT_TEST(testAddTorrent_notBase64Torrent);
  41. CPPUNIT_TEST(testAddTorrent_withPosition);
  42. #endif // ENABLE_BITTORRENT
  43. #ifdef ENABLE_METALINK
  44. CPPUNIT_TEST(testAddMetalink);
  45. CPPUNIT_TEST(testAddMetalink_withoutMetalink);
  46. CPPUNIT_TEST(testAddMetalink_notBase64Metalink);
  47. CPPUNIT_TEST(testAddMetalink_withPosition);
  48. #endif // ENABLE_METALINK
  49. CPPUNIT_TEST(testChangeOption);
  50. CPPUNIT_TEST(testChangeOption_withBadOption);
  51. CPPUNIT_TEST(testChangeOption_withNotAllowedOption);
  52. CPPUNIT_TEST(testChangeOption_withoutGid);
  53. CPPUNIT_TEST(testChangeGlobalOption);
  54. CPPUNIT_TEST(testChangeGlobalOption_withBadOption);
  55. CPPUNIT_TEST(testChangeGlobalOption_withNotAllowedOption);
  56. CPPUNIT_TEST(testTellStatus_withoutGid);
  57. CPPUNIT_TEST(testTellWaiting);
  58. CPPUNIT_TEST(testTellWaiting_fail);
  59. CPPUNIT_TEST(testGetVersion);
  60. CPPUNIT_TEST(testNoSuchMethod);
  61. CPPUNIT_TEST_SUITE_END();
  62. private:
  63. SharedHandle<DownloadEngine> _e;
  64. SharedHandle<Option> _option;
  65. public:
  66. void setUp()
  67. {
  68. RequestGroup::resetGIDCounter();
  69. _option.reset(new Option());
  70. _option->put(PREF_DIR, "/tmp");
  71. _option->put(PREF_SEGMENT_SIZE, "1048576");
  72. _e.reset(new DownloadEngine(SharedHandle<EventPoll>(new SelectEventPoll())));
  73. _e->option = _option.get();
  74. _e->_requestGroupMan.reset
  75. (new RequestGroupMan(std::deque<SharedHandle<RequestGroup> >(),
  76. 1, _option.get()));
  77. }
  78. void tearDown() {}
  79. void testAddUri();
  80. void testAddUri_withoutUri();
  81. void testAddUri_notUri();
  82. void testAddUri_withBadOption();
  83. void testAddUri_withPosition();
  84. void testAddUri_withBadPosition();
  85. #ifdef ENABLE_BITTORRENT
  86. void testAddTorrent();
  87. void testAddTorrent_withoutTorrent();
  88. void testAddTorrent_notBase64Torrent();
  89. void testAddTorrent_withPosition();
  90. #endif // ENABLE_BITTORRENT
  91. #ifdef ENABLE_METALINK
  92. void testAddMetalink();
  93. void testAddMetalink_withoutMetalink();
  94. void testAddMetalink_notBase64Metalink();
  95. void testAddMetalink_withPosition();
  96. #endif // ENABLE_METALINK
  97. void testChangeOption();
  98. void testChangeOption_withBadOption();
  99. void testChangeOption_withNotAllowedOption();
  100. void testChangeOption_withoutGid();
  101. void testChangeGlobalOption();
  102. void testChangeGlobalOption_withBadOption();
  103. void testChangeGlobalOption_withNotAllowedOption();
  104. void testTellStatus_withoutGid();
  105. void testTellWaiting();
  106. void testTellWaiting_fail();
  107. void testGetVersion();
  108. void testNoSuchMethod();
  109. };
  110. CPPUNIT_TEST_SUITE_REGISTRATION(XmlRpcMethodTest);
  111. void XmlRpcMethodTest::testAddUri()
  112. {
  113. AddUriXmlRpcMethod m;
  114. XmlRpcRequest req("aria2.addUri", BDE::list());
  115. req._params << BDE::list();
  116. req._params[0] << BDE("http://localhost/");
  117. {
  118. XmlRpcResponse res = m.execute(req, _e.get());
  119. CPPUNIT_ASSERT_EQUAL(0, res._code);
  120. const std::deque<SharedHandle<RequestGroup> > rgs =
  121. _e->_requestGroupMan->getReservedGroups();
  122. CPPUNIT_ASSERT_EQUAL((size_t)1, rgs.size());
  123. CPPUNIT_ASSERT_EQUAL(std::string("http://localhost/"),
  124. rgs.front()->getDownloadContext()->getFirstFileEntry()->getRemainingUris().front());
  125. }
  126. // with options
  127. BDE opt = BDE::dict();
  128. opt[PREF_DIR] = BDE("/sink");
  129. req._params << opt;
  130. {
  131. XmlRpcResponse res = m.execute(req, _e.get());
  132. CPPUNIT_ASSERT_EQUAL(0, res._code);
  133. CPPUNIT_ASSERT_EQUAL(std::string("/sink"),
  134. _e->_requestGroupMan->findReservedGroup(2)->
  135. getDownloadContext()->getDir());
  136. }
  137. }
  138. void XmlRpcMethodTest::testAddUri_withoutUri()
  139. {
  140. AddUriXmlRpcMethod m;
  141. XmlRpcRequest req("aria2.addUri", BDE::list());
  142. XmlRpcResponse res = m.execute(req, _e.get());
  143. CPPUNIT_ASSERT_EQUAL(1, res._code);
  144. }
  145. void XmlRpcMethodTest::testAddUri_notUri()
  146. {
  147. AddUriXmlRpcMethod m;
  148. XmlRpcRequest req("aria2.addUri", BDE::list());
  149. req._params << BDE::list();
  150. req._params[0] << BDE("not uri");
  151. XmlRpcResponse res = m.execute(req, _e.get());
  152. CPPUNIT_ASSERT_EQUAL(1, res._code);
  153. }
  154. void XmlRpcMethodTest::testAddUri_withBadOption()
  155. {
  156. AddUriXmlRpcMethod m;
  157. XmlRpcRequest req("aria2.addUri", BDE::list());
  158. req._params << BDE::list();
  159. req._params[0] << BDE("http://localhost");
  160. BDE opt = BDE::dict();
  161. opt[PREF_FILE_ALLOCATION] = BDE("badvalue");
  162. req._params << opt;
  163. XmlRpcResponse res = m.execute(req, _e.get());
  164. CPPUNIT_ASSERT_EQUAL(1, res._code);
  165. }
  166. void XmlRpcMethodTest::testAddUri_withPosition()
  167. {
  168. AddUriXmlRpcMethod m;
  169. XmlRpcRequest req1("aria2.addUri", BDE::list());
  170. req1._params << BDE::list();
  171. req1._params[0] << BDE("http://uri1");
  172. XmlRpcResponse res1 = m.execute(req1, _e.get());
  173. CPPUNIT_ASSERT_EQUAL(0, res1._code);
  174. XmlRpcRequest req2("aria2.addUri", BDE::list());
  175. req2._params << BDE::list();
  176. req2._params[0] << BDE("http://uri2");
  177. req2._params << BDE::dict();
  178. req2._params << BDE((int64_t)0);
  179. m.execute(req2, _e.get());
  180. std::string uri =
  181. _e->_requestGroupMan->getReservedGroups()[0]->getDownloadContext()->getFirstFileEntry()->getRemainingUris()[0];
  182. CPPUNIT_ASSERT_EQUAL(std::string("http://uri2"), uri);
  183. }
  184. void XmlRpcMethodTest::testAddUri_withBadPosition()
  185. {
  186. AddUriXmlRpcMethod m;
  187. XmlRpcRequest req("aria2.addUri", BDE::list());
  188. req._params << BDE::list();
  189. req._params[0] << BDE("http://localhost/");
  190. req._params << BDE::dict();
  191. req._params << BDE((int64_t)-1);
  192. XmlRpcResponse res = m.execute(req, _e.get());
  193. CPPUNIT_ASSERT_EQUAL(1, res._code);
  194. }
  195. #ifdef ENABLE_BITTORRENT
  196. void XmlRpcMethodTest::testAddTorrent()
  197. {
  198. AddTorrentXmlRpcMethod m;
  199. XmlRpcRequest req("aria2.addTorrent", BDE::list());
  200. req._params << BDE(readFile("single.torrent"));
  201. BDE uris = BDE::list();
  202. uris << BDE("http://localhost/aria2-0.8.2.tar.bz2");
  203. req._params << uris;
  204. {
  205. XmlRpcResponse res = m.execute(req, _e.get());
  206. CPPUNIT_ASSERT_EQUAL(0, res._code);
  207. CPPUNIT_ASSERT_EQUAL(std::string("1"), res._param.s());
  208. SharedHandle<RequestGroup> group = _e->_requestGroupMan->findReservedGroup(1);
  209. CPPUNIT_ASSERT(!group.isNull());
  210. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/aria2-0.8.2.tar.bz2"),
  211. group->getFirstFilePath());
  212. CPPUNIT_ASSERT_EQUAL((size_t)1, group->getDownloadContext()->getFirstFileEntry()->getRemainingUris().size());
  213. CPPUNIT_ASSERT_EQUAL(std::string("http://localhost/aria2-0.8.2.tar.bz2"),
  214. group->getDownloadContext()->getFirstFileEntry()->getRemainingUris()[0]);
  215. }
  216. // with options
  217. BDE opt = BDE::dict();
  218. opt[PREF_DIR] = BDE("/sink");
  219. req._params << opt;
  220. {
  221. XmlRpcResponse res = m.execute(req, _e.get());
  222. CPPUNIT_ASSERT_EQUAL(0, res._code);
  223. CPPUNIT_ASSERT_EQUAL(std::string("/sink/aria2-0.8.2.tar.bz2"),
  224. _e->_requestGroupMan->findReservedGroup(2)->getFirstFilePath());
  225. }
  226. }
  227. void XmlRpcMethodTest::testAddTorrent_withoutTorrent()
  228. {
  229. AddTorrentXmlRpcMethod m;
  230. XmlRpcRequest req("aria2.addTorrent", BDE::list());
  231. XmlRpcResponse res = m.execute(req, _e.get());
  232. CPPUNIT_ASSERT_EQUAL(1, res._code);
  233. }
  234. void XmlRpcMethodTest::testAddTorrent_notBase64Torrent()
  235. {
  236. AddTorrentXmlRpcMethod m;
  237. XmlRpcRequest req("aria2.addTorrent", BDE::list());
  238. req._params << BDE("not torrent");
  239. XmlRpcResponse res = m.execute(req, _e.get());
  240. CPPUNIT_ASSERT_EQUAL(1, res._code);
  241. }
  242. void XmlRpcMethodTest::testAddTorrent_withPosition()
  243. {
  244. AddTorrentXmlRpcMethod m;
  245. XmlRpcRequest req1("aria2.addTorrent", BDE::list());
  246. req1._params << BDE(readFile("test.torrent"));
  247. req1._params << BDE::list();
  248. req1._params << BDE::dict();
  249. XmlRpcResponse res1 = m.execute(req1, _e.get());
  250. CPPUNIT_ASSERT_EQUAL(0, res1._code);
  251. XmlRpcRequest req2("aria2.addTorrent", BDE::list());
  252. req2._params << BDE(readFile("single.torrent"));
  253. req2._params << BDE::list();
  254. req2._params << BDE::dict();
  255. req2._params << BDE((int64_t)0);
  256. m.execute(req2, _e.get());
  257. CPPUNIT_ASSERT_EQUAL((size_t)1,
  258. _e->_requestGroupMan->getReservedGroups()[0]->
  259. getDownloadContext()->getFileEntries().size());
  260. }
  261. #endif // ENABLE_BITTORRENT
  262. #ifdef ENABLE_METALINK
  263. void XmlRpcMethodTest::testAddMetalink()
  264. {
  265. AddMetalinkXmlRpcMethod m;
  266. XmlRpcRequest req("aria2.addMetalink", BDE::list());
  267. req._params << BDE(readFile("2files.metalink"));
  268. {
  269. XmlRpcResponse res = m.execute(req, _e.get());
  270. CPPUNIT_ASSERT_EQUAL(0, res._code);
  271. CPPUNIT_ASSERT_EQUAL((size_t)2, res._param.size());
  272. CPPUNIT_ASSERT_EQUAL(std::string("1"), res._param[0].s());
  273. CPPUNIT_ASSERT_EQUAL(std::string("2"), res._param[1].s());
  274. SharedHandle<RequestGroup> tar = _e->_requestGroupMan->findReservedGroup(1);
  275. CPPUNIT_ASSERT(!tar.isNull());
  276. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/aria2-5.0.0.tar.bz2"),
  277. tar->getFirstFilePath());
  278. SharedHandle<RequestGroup> deb = _e->_requestGroupMan->findReservedGroup(2);
  279. CPPUNIT_ASSERT(!deb.isNull());
  280. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/aria2-5.0.0.deb"),
  281. deb->getFirstFilePath());
  282. }
  283. // with options
  284. BDE opt = BDE::dict();
  285. opt[PREF_DIR] = BDE("/sink");
  286. req._params << opt;
  287. {
  288. XmlRpcResponse res = m.execute(req, _e.get());
  289. CPPUNIT_ASSERT_EQUAL(0, res._code);
  290. CPPUNIT_ASSERT_EQUAL(std::string("/sink/aria2-5.0.0.tar.bz2"),
  291. _e->_requestGroupMan->findReservedGroup(3)->getFirstFilePath());
  292. }
  293. }
  294. void XmlRpcMethodTest::testAddMetalink_withoutMetalink()
  295. {
  296. AddMetalinkXmlRpcMethod m;
  297. XmlRpcRequest req("aria2.addMetalink", BDE::list());
  298. XmlRpcResponse res = m.execute(req, _e.get());
  299. CPPUNIT_ASSERT_EQUAL(1, res._code);
  300. }
  301. void XmlRpcMethodTest::testAddMetalink_notBase64Metalink()
  302. {
  303. AddMetalinkXmlRpcMethod m;
  304. XmlRpcRequest req("aria2.addMetalink", BDE::list());
  305. req._params << BDE("not metalink");
  306. XmlRpcResponse res = m.execute(req, _e.get());
  307. CPPUNIT_ASSERT_EQUAL(1, res._code);
  308. }
  309. void XmlRpcMethodTest::testAddMetalink_withPosition()
  310. {
  311. AddUriXmlRpcMethod m1;
  312. XmlRpcRequest req1("aria2.addUri", BDE::list());
  313. req1._params << BDE::list();
  314. req1._params[0] << BDE("http://uri");
  315. XmlRpcResponse res1 = m1.execute(req1, _e.get());
  316. CPPUNIT_ASSERT_EQUAL(0, res1._code);
  317. AddMetalinkXmlRpcMethod m2;
  318. XmlRpcRequest req2("ari2.addMetalink", BDE::list());
  319. req2._params << BDE(readFile("2files.metalink"));
  320. req2._params << BDE::dict();
  321. req2._params << BDE((int64_t)0);
  322. XmlRpcResponse res2 = m2.execute(req2, _e.get());
  323. CPPUNIT_ASSERT_EQUAL(0, res2._code);
  324. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/aria2-5.0.0.tar.bz2"),
  325. _e->_requestGroupMan->getReservedGroups()[0]->
  326. getFirstFilePath());
  327. }
  328. #endif // ENABLE_METALINK
  329. void XmlRpcMethodTest::testChangeOption()
  330. {
  331. SharedHandle<RequestGroup> group(new RequestGroup(_option));
  332. _e->_requestGroupMan->addReservedGroup(group);
  333. ChangeOptionXmlRpcMethod m;
  334. XmlRpcRequest req("aria2.changeOption", BDE::list());
  335. req._params << BDE("1");
  336. BDE opt = BDE::dict();
  337. opt[PREF_MAX_DOWNLOAD_LIMIT] = BDE("100K");
  338. #ifdef ENABLE_BITTORRENT
  339. opt[PREF_BT_MAX_PEERS] = BDE("100");
  340. opt[PREF_BT_REQUEST_PEER_SPEED_LIMIT] = BDE("300K");
  341. opt[PREF_MAX_UPLOAD_LIMIT] = BDE("50K");
  342. BtObject btObject;
  343. btObject._btRuntime = SharedHandle<BtRuntime>(new BtRuntime());
  344. _e->getBtRegistry()->put(group->getGID(), btObject);
  345. #endif // ENABLE_BITTORRENT
  346. req._params << opt;
  347. XmlRpcResponse res = m.execute(req, _e.get());
  348. SharedHandle<Option> option = group->getOption();
  349. CPPUNIT_ASSERT_EQUAL(0, res._code);
  350. CPPUNIT_ASSERT_EQUAL((unsigned int)100*1024,
  351. group->getMaxDownloadSpeedLimit());
  352. CPPUNIT_ASSERT_EQUAL(std::string("102400"),
  353. option->get(PREF_MAX_DOWNLOAD_LIMIT));
  354. #ifdef ENABLE_BITTORRENT
  355. CPPUNIT_ASSERT_EQUAL(std::string("307200"),
  356. option->get(PREF_BT_REQUEST_PEER_SPEED_LIMIT));
  357. CPPUNIT_ASSERT_EQUAL(std::string("100"), option->get(PREF_BT_MAX_PEERS));
  358. CPPUNIT_ASSERT_EQUAL((unsigned int)100, btObject._btRuntime->getMaxPeers());
  359. CPPUNIT_ASSERT_EQUAL((unsigned int)50*1024,
  360. group->getMaxUploadSpeedLimit());
  361. CPPUNIT_ASSERT_EQUAL(std::string("51200"),
  362. option->get(PREF_MAX_UPLOAD_LIMIT));
  363. #endif // ENABLE_BITTORRENT
  364. }
  365. void XmlRpcMethodTest::testChangeOption_withBadOption()
  366. {
  367. SharedHandle<RequestGroup> group(new RequestGroup(_option));
  368. _e->_requestGroupMan->addReservedGroup(group);
  369. ChangeOptionXmlRpcMethod m;
  370. XmlRpcRequest req("aria2.changeOption", BDE::list());
  371. req._params << BDE("1");
  372. BDE opt = BDE::dict();
  373. opt[PREF_MAX_DOWNLOAD_LIMIT] = BDE("badvalue");
  374. req._params << opt;
  375. XmlRpcResponse res = m.execute(req, _e.get());
  376. CPPUNIT_ASSERT_EQUAL(1, res._code);
  377. }
  378. void XmlRpcMethodTest::testChangeOption_withNotAllowedOption()
  379. {
  380. SharedHandle<RequestGroup> group(new RequestGroup(_option));
  381. _e->_requestGroupMan->addReservedGroup(group);
  382. ChangeOptionXmlRpcMethod m;
  383. XmlRpcRequest req("aria2.changeOption", BDE::list());
  384. req._params << BDE("1");
  385. BDE opt = BDE::dict();
  386. opt[PREF_MAX_OVERALL_DOWNLOAD_LIMIT] = BDE("100K");
  387. req._params << opt;
  388. XmlRpcResponse res = m.execute(req, _e.get());
  389. CPPUNIT_ASSERT_EQUAL(1, res._code);
  390. }
  391. void XmlRpcMethodTest::testChangeOption_withoutGid()
  392. {
  393. ChangeOptionXmlRpcMethod m;
  394. XmlRpcRequest req("aria2.changeOption", BDE::list());
  395. XmlRpcResponse res = m.execute(req, _e.get());
  396. CPPUNIT_ASSERT_EQUAL(1, res._code);
  397. }
  398. void XmlRpcMethodTest::testChangeGlobalOption()
  399. {
  400. ChangeGlobalOptionXmlRpcMethod m;
  401. XmlRpcRequest req("aria2.changeGlobalOption", BDE::list());
  402. BDE opt = BDE::dict();
  403. opt[PREF_MAX_OVERALL_DOWNLOAD_LIMIT] = BDE("100K");
  404. #ifdef ENABLE_BITTORRENT
  405. opt[PREF_MAX_OVERALL_UPLOAD_LIMIT] = BDE("50K");
  406. #endif // ENABLE_BITTORRENT
  407. req._params << opt;
  408. XmlRpcResponse res = m.execute(req, _e.get());
  409. CPPUNIT_ASSERT_EQUAL(0, res._code);
  410. CPPUNIT_ASSERT_EQUAL((unsigned int)100*1024,
  411. _e->_requestGroupMan->getMaxOverallDownloadSpeedLimit());
  412. CPPUNIT_ASSERT_EQUAL(std::string("102400"),
  413. _e->option->get(PREF_MAX_OVERALL_DOWNLOAD_LIMIT));
  414. #ifdef ENABLE_BITTORRENT
  415. CPPUNIT_ASSERT_EQUAL((unsigned int)50*1024,
  416. _e->_requestGroupMan->getMaxOverallUploadSpeedLimit());
  417. CPPUNIT_ASSERT_EQUAL(std::string("51200"),
  418. _e->option->get(PREF_MAX_OVERALL_UPLOAD_LIMIT));
  419. #endif // ENABLE_BITTORRENT
  420. }
  421. void XmlRpcMethodTest::testChangeGlobalOption_withBadOption()
  422. {
  423. ChangeGlobalOptionXmlRpcMethod m;
  424. XmlRpcRequest req("aria2.changeGlobalOption", BDE::list());
  425. BDE opt = BDE::dict();
  426. opt[PREF_MAX_OVERALL_DOWNLOAD_LIMIT] = BDE("badvalue");
  427. req._params << opt;
  428. XmlRpcResponse res = m.execute(req, _e.get());
  429. CPPUNIT_ASSERT_EQUAL(1, res._code);
  430. }
  431. void XmlRpcMethodTest::testChangeGlobalOption_withNotAllowedOption()
  432. {
  433. ChangeGlobalOptionXmlRpcMethod m;
  434. XmlRpcRequest req("aria2.changeGlobalOption", BDE::list());
  435. BDE opt = BDE::dict();
  436. opt[PREF_MAX_DOWNLOAD_LIMIT] = BDE("100K");
  437. req._params << opt;
  438. XmlRpcResponse res = m.execute(req, _e.get());
  439. CPPUNIT_ASSERT_EQUAL(1, res._code);
  440. }
  441. void XmlRpcMethodTest::testNoSuchMethod()
  442. {
  443. NoSuchMethodXmlRpcMethod m;
  444. XmlRpcRequest req("make.hamburger", BDE::none);
  445. XmlRpcResponse res = m.execute(req, 0);
  446. CPPUNIT_ASSERT_EQUAL(1, res._code);
  447. CPPUNIT_ASSERT_EQUAL(std::string("No such method: make.hamburger"),
  448. res._param["faultString"].s());
  449. CPPUNIT_ASSERT_EQUAL
  450. (std::string("<?xml version=\"1.0\"?>"
  451. "<methodResponse>"
  452. "<fault>"
  453. "<value>"
  454. "<struct>"
  455. "<member>"
  456. "<name>faultCode</name><value><int>1</int></value>"
  457. "</member>"
  458. "<member>"
  459. "<name>faultString</name>"
  460. "<value>"
  461. "<string>No such method: make.hamburger</string>"
  462. "</value>"
  463. "</member>"
  464. "</struct>"
  465. "</value>"
  466. "</fault>"
  467. "</methodResponse>"),
  468. res.toXml());
  469. }
  470. void XmlRpcMethodTest::testTellStatus_withoutGid()
  471. {
  472. TellStatusXmlRpcMethod m;
  473. XmlRpcRequest req("aria2.tellStatus", BDE::list());
  474. XmlRpcResponse res = m.execute(req, _e.get());
  475. CPPUNIT_ASSERT_EQUAL(1, res._code);
  476. }
  477. static void addUri(const std::string& uri,
  478. const SharedHandle<DownloadEngine>& e)
  479. {
  480. AddUriXmlRpcMethod m;
  481. XmlRpcRequest req("aria2.addUri", BDE::list());
  482. req._params << BDE::list();
  483. req._params[0] << BDE(uri);
  484. CPPUNIT_ASSERT_EQUAL(0, m.execute(req, e.get())._code);
  485. }
  486. #ifdef ENABLE_BITTORRENT
  487. static void addTorrent
  488. (const std::string& torrentFile, const SharedHandle<DownloadEngine>& e)
  489. {
  490. AddTorrentXmlRpcMethod m;
  491. XmlRpcRequest req("aria2.addTorrent", BDE::list());
  492. req._params << BDE(readFile(torrentFile));
  493. XmlRpcResponse res = m.execute(req, e.get());
  494. }
  495. #endif // ENABLE_BITTORRENT
  496. void XmlRpcMethodTest::testTellWaiting()
  497. {
  498. addUri("http://1/", _e);
  499. addUri("http://2/", _e);
  500. addUri("http://3/", _e);
  501. #ifdef ENABLE_BITTORRENT
  502. addTorrent("single.torrent", _e);
  503. #endif // ENABLE_BITTORRENT
  504. TellWaitingXmlRpcMethod m;
  505. XmlRpcRequest req("aria2.tellWaiting", BDE::list());
  506. req._params << BDE((int64_t)1);
  507. req._params << BDE((int64_t)2);
  508. XmlRpcResponse res = m.execute(req, _e.get());
  509. CPPUNIT_ASSERT_EQUAL(0, res._code);
  510. CPPUNIT_ASSERT_EQUAL((size_t)2, res._param.size());
  511. CPPUNIT_ASSERT_EQUAL(std::string("2"), res._param[0]["gid"].s());
  512. CPPUNIT_ASSERT_EQUAL(std::string("3"), res._param[1]["gid"].s());
  513. // waiting.size() == offset+num
  514. req = XmlRpcRequest("aria2.tellWaiting", BDE::list());
  515. req._params << BDE((int64_t)1);
  516. #ifdef ENABLE_BITTORRENT
  517. req._params << BDE((int64_t)3);
  518. #else // !ENABLE_BITTORRENT
  519. req._params << BDE((int64_t)2);
  520. #endif // !ENABLE_BITTORRENT
  521. res = m.execute(req, _e.get());
  522. CPPUNIT_ASSERT_EQUAL(0, res._code);
  523. #ifdef ENABLE_BITTORRENT
  524. CPPUNIT_ASSERT_EQUAL((size_t)3, res._param.size());
  525. #else // !ENABLE_BITTORRENT
  526. CPPUNIT_ASSERT_EQUAL((size_t)2, res._param.size());
  527. #endif // !ENABLE_BITTORRENT
  528. // waiting.size() < offset+num
  529. req = XmlRpcRequest("aria2.tellWaiting", BDE::list());
  530. req._params << BDE((int64_t)1);
  531. req._params << BDE((int64_t)4);
  532. res = m.execute(req, _e.get());
  533. CPPUNIT_ASSERT_EQUAL(0, res._code);
  534. #ifdef ENABLE_BITTORRENT
  535. CPPUNIT_ASSERT_EQUAL((size_t)3, res._param.size());
  536. #else //!ENABLE_BITTORRENT
  537. CPPUNIT_ASSERT_EQUAL((size_t)2, res._param.size());
  538. #endif // !ENABLE_BITTORRENT
  539. }
  540. void XmlRpcMethodTest::testTellWaiting_fail()
  541. {
  542. TellWaitingXmlRpcMethod m;
  543. XmlRpcRequest req("aria2.tellWaiting", BDE::list());
  544. XmlRpcResponse res = m.execute(req, _e.get());
  545. CPPUNIT_ASSERT_EQUAL(1, res._code);
  546. }
  547. void XmlRpcMethodTest::testGetVersion()
  548. {
  549. GetVersionXmlRpcMethod m;
  550. XmlRpcRequest req("aria2.getVersion", BDE::none);
  551. XmlRpcResponse res = m.execute(req, _e.get());
  552. CPPUNIT_ASSERT_EQUAL(0, res._code);
  553. CPPUNIT_ASSERT_EQUAL(std::string(PACKAGE_VERSION), res._param["version"].s());
  554. const BDE& featureList = res._param["enabledFeatures"];
  555. std::string features;
  556. for(BDE::List::const_iterator i = featureList.listBegin();
  557. i != featureList.listEnd(); ++i) {
  558. features += (*i).s();
  559. features += ", ";
  560. }
  561. CPPUNIT_ASSERT_EQUAL(FeatureConfig::getInstance()->featureSummary()+", ",
  562. features);
  563. }
  564. } // namespace xmlrpc
  565. } // namespace aria2