DefaultPeerStorageTest.cc 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. #include "DefaultPeerStorage.h"
  2. #include "DefaultBtContext.h"
  3. #include "Util.h"
  4. #include "Exception.h"
  5. #include "Peer.h"
  6. #include "Option.h"
  7. #include "BtRuntime.h"
  8. #include <algorithm>
  9. #include <cppunit/extensions/HelperMacros.h>
  10. namespace aria2 {
  11. class DefaultPeerStorageTest:public CppUnit::TestFixture {
  12. CPPUNIT_TEST_SUITE(DefaultPeerStorageTest);
  13. CPPUNIT_TEST(testCountPeer);
  14. CPPUNIT_TEST(testDeleteUnusedPeer);
  15. CPPUNIT_TEST(testAddPeer);
  16. CPPUNIT_TEST(testGetUnusedPeer);
  17. CPPUNIT_TEST(testIsPeerAvailable);
  18. CPPUNIT_TEST(testActivatePeer);
  19. CPPUNIT_TEST(testCalculateStat);
  20. CPPUNIT_TEST(testReturnPeer);
  21. CPPUNIT_TEST(testOnErasingPeer);
  22. CPPUNIT_TEST_SUITE_END();
  23. private:
  24. SharedHandle<BtContext> btContext;
  25. SharedHandle<BtRuntime> btRuntime;
  26. Option* option;
  27. public:
  28. void setUp() {
  29. btContext.reset(new DefaultBtContext());
  30. btContext->load("test.torrent");
  31. option = new Option();
  32. btRuntime.reset(new BtRuntime());
  33. }
  34. void tearDown() {
  35. delete option;
  36. }
  37. void testCountPeer();
  38. void testDeleteUnusedPeer();
  39. void testAddPeer();
  40. void testGetUnusedPeer();
  41. void testIsPeerAvailable();
  42. void testActivatePeer();
  43. void testCalculateStat();
  44. void testReturnPeer();
  45. void testOnErasingPeer();
  46. };
  47. CPPUNIT_TEST_SUITE_REGISTRATION(DefaultPeerStorageTest);
  48. void DefaultPeerStorageTest::testCountPeer() {
  49. DefaultPeerStorage ps(btContext, option);
  50. CPPUNIT_ASSERT_EQUAL((size_t)0, ps.countPeer());
  51. SharedHandle<Peer> peer(new Peer("192.168.0.1", 6889));
  52. ps.addPeer(peer);
  53. CPPUNIT_ASSERT_EQUAL((size_t)1, ps.countPeer());
  54. }
  55. void DefaultPeerStorageTest::testDeleteUnusedPeer() {
  56. DefaultPeerStorage ps(btContext, option);
  57. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 6889));
  58. SharedHandle<Peer> peer2(new Peer("192.168.0.2", 6889));
  59. SharedHandle<Peer> peer3(new Peer("192.168.0.3", 6889));
  60. ps.addPeer(peer1);
  61. ps.addPeer(peer2);
  62. ps.addPeer(peer3);
  63. ps.deleteUnusedPeer(2);
  64. CPPUNIT_ASSERT_EQUAL((size_t)1, ps.countPeer());
  65. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.3"),
  66. ps.getPeer("192.168.0.3", 6889)->ipaddr);
  67. ps.addPeer(peer1);
  68. ps.addPeer(peer2);
  69. peer2->usedBy(1);
  70. ps.deleteUnusedPeer(3);
  71. // peer2 has been in use, so it did't deleted.
  72. CPPUNIT_ASSERT_EQUAL((size_t)1, ps.countPeer());
  73. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.2"),
  74. ps.getPeer("192.168.0.2", 6889)->ipaddr);
  75. }
  76. void DefaultPeerStorageTest::testAddPeer() {
  77. DefaultPeerStorage ps(btContext, option);
  78. SharedHandle<BtRuntime> btRuntime(new BtRuntime());
  79. btRuntime->setMaxPeers(3);
  80. ps.setBtRuntime(btRuntime);
  81. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 6889));
  82. SharedHandle<Peer> peer2(new Peer("192.168.0.2", 6889));
  83. SharedHandle<Peer> peer3(new Peer("192.168.0.3", 6889));
  84. ps.addPeer(peer1);
  85. ps.addPeer(peer2);
  86. ps.addPeer(peer3);
  87. CPPUNIT_ASSERT_EQUAL((size_t)3, ps.countPeer());
  88. // this returns false, because peer1 is already in the container
  89. CPPUNIT_ASSERT_EQUAL(false, ps.addPeer(peer1));
  90. // the number of peers doesn't change.
  91. CPPUNIT_ASSERT_EQUAL((size_t)3, ps.countPeer());
  92. SharedHandle<Peer> peer4(new Peer("192.168.0.4", 6889));
  93. peer1->usedBy(1);
  94. CPPUNIT_ASSERT(ps.addPeer(peer4));
  95. // peer2 was deleted. While peer1 is oldest, its cuid is not 0.
  96. CPPUNIT_ASSERT_EQUAL((size_t)3, ps.countPeer());
  97. CPPUNIT_ASSERT(std::find(ps.getPeers().begin(), ps.getPeers().end(), peer2) == ps.getPeers().end());
  98. SharedHandle<Peer> peer5(new Peer("192.168.0.4", 0));
  99. peer5->port = 6889;
  100. // this returns false because the peer which has same ip and port has already added
  101. CPPUNIT_ASSERT_EQUAL(false, ps.addPeer(peer5));
  102. }
  103. void DefaultPeerStorageTest::testGetUnusedPeer() {
  104. DefaultPeerStorage ps(btContext, option);
  105. ps.setBtRuntime(btRuntime);
  106. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 6889));
  107. ps.addPeer(peer1);
  108. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"),
  109. ps.getUnusedPeer()->ipaddr);
  110. peer1->usedBy(1);
  111. CPPUNIT_ASSERT(ps.getUnusedPeer().isNull());
  112. peer1->resetStatus();
  113. peer1->startBadCondition();
  114. CPPUNIT_ASSERT(ps.getUnusedPeer().isNull());
  115. }
  116. void DefaultPeerStorageTest::testIsPeerAvailable() {
  117. DefaultPeerStorage ps(btContext, option);
  118. ps.setBtRuntime(btRuntime);
  119. CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());
  120. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 6889));
  121. ps.addPeer(peer1);
  122. CPPUNIT_ASSERT_EQUAL(true, ps.isPeerAvailable());
  123. peer1->usedBy(1);
  124. CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());
  125. peer1->resetStatus();
  126. peer1->startBadCondition();
  127. CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());
  128. }
  129. void DefaultPeerStorageTest::testActivatePeer() {
  130. DefaultPeerStorage ps(btContext, option);
  131. {
  132. std::deque<SharedHandle<Peer> > peers;
  133. ps.getActivePeers(peers);
  134. CPPUNIT_ASSERT_EQUAL((size_t)0, peers.size());
  135. }
  136. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 6889));
  137. ps.addPeer(peer1);
  138. {
  139. std::deque<SharedHandle<Peer> > activePeers;
  140. ps.getActivePeers(activePeers);
  141. CPPUNIT_ASSERT_EQUAL((size_t)0, activePeers.size());
  142. }
  143. {
  144. peer1->allocateSessionResource(btContext->getPieceLength(),
  145. btContext->getTotalLength());
  146. std::deque<SharedHandle<Peer> > activePeers;
  147. ps.getActivePeers(activePeers);
  148. CPPUNIT_ASSERT_EQUAL((size_t)1, activePeers.size());
  149. }
  150. }
  151. void DefaultPeerStorageTest::testCalculateStat() {
  152. }
  153. void DefaultPeerStorageTest::testReturnPeer()
  154. {
  155. DefaultPeerStorage ps(btContext, option);
  156. SharedHandle<Peer> peer1(new Peer("192.168.0.1", 0));
  157. peer1->allocateSessionResource(btContext->getPieceLength(),
  158. btContext->getTotalLength());
  159. SharedHandle<Peer> peer2(new Peer("192.168.0.2", 6889));
  160. peer2->allocateSessionResource(btContext->getPieceLength(),
  161. btContext->getTotalLength());
  162. SharedHandle<Peer> peer3(new Peer("192.168.0.1", 6889));
  163. ps.addPeer(peer1);
  164. ps.addPeer(peer2);
  165. ps.addPeer(peer3);
  166. ps.returnPeer(peer2); // peer2 removed from the container
  167. CPPUNIT_ASSERT_EQUAL((size_t)2, ps.getPeers().size());
  168. CPPUNIT_ASSERT(std::find(ps.getPeers().begin(), ps.getPeers().end(), peer2)
  169. == ps.getPeers().end());
  170. ps.returnPeer(peer1); // peer1 is removed from the container
  171. CPPUNIT_ASSERT_EQUAL((size_t)1, ps.getPeers().size());
  172. CPPUNIT_ASSERT(std::find(ps.getPeers().begin(), ps.getPeers().end(), peer1) == ps.getPeers().end());
  173. }
  174. void DefaultPeerStorageTest::testOnErasingPeer()
  175. {
  176. // test this
  177. }
  178. } // namespace aria2