DHTFindNodeReplyMessageTest.cc 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. #include "DHTFindNodeReplyMessage.h"
  2. #include <cppunit/extensions/HelperMacros.h>
  3. #include "DHTNode.h"
  4. #include "Exception.h"
  5. #include "util.h"
  6. #include "DHTBucket.h"
  7. #include "bittorrent_helper.h"
  8. #include "bencode.h"
  9. namespace aria2 {
  10. class DHTFindNodeReplyMessageTest:public CppUnit::TestFixture {
  11. CPPUNIT_TEST_SUITE(DHTFindNodeReplyMessageTest);
  12. CPPUNIT_TEST(testGetBencodedMessage);
  13. CPPUNIT_TEST_SUITE_END();
  14. public:
  15. void setUp() {}
  16. void tearDown() {}
  17. void testGetBencodedMessage();
  18. };
  19. CPPUNIT_TEST_SUITE_REGISTRATION(DHTFindNodeReplyMessageTest);
  20. void DHTFindNodeReplyMessageTest::testGetBencodedMessage()
  21. {
  22. SharedHandle<DHTNode> localNode(new DHTNode());
  23. SharedHandle<DHTNode> remoteNode(new DHTNode());
  24. unsigned char tid[DHT_TRANSACTION_ID_LENGTH];
  25. util::generateRandomData(tid, DHT_TRANSACTION_ID_LENGTH);
  26. std::string transactionID(&tid[0], &tid[DHT_TRANSACTION_ID_LENGTH]);
  27. DHTFindNodeReplyMessage msg(localNode, remoteNode, transactionID);
  28. msg.setVersion("A200");
  29. std::string compactNodeInfo;
  30. SharedHandle<DHTNode> nodes[8];
  31. for(size_t i = 0; i < DHTBucket::K; ++i) {
  32. nodes[i].reset(new DHTNode());
  33. nodes[i]->setIPAddress("192.168.0."+util::uitos(i+1));
  34. nodes[i]->setPort(6881+i);
  35. unsigned char buf[6];
  36. CPPUNIT_ASSERT(bittorrent::createcompact
  37. (buf, nodes[i]->getIPAddress(), nodes[i]->getPort()));
  38. compactNodeInfo +=
  39. std::string(&nodes[i]->getID()[0], &nodes[i]->getID()[DHT_ID_LENGTH])+
  40. std::string(&buf[0], &buf[sizeof(buf)]);
  41. }
  42. msg.setClosestKNodes
  43. (std::vector<SharedHandle<DHTNode> >(&nodes[0], &nodes[DHTBucket::K]));
  44. std::string msgbody = msg.getBencodedMessage();
  45. BDE dict = BDE::dict();
  46. dict["t"] = transactionID;
  47. dict["v"] = BDE("A200");
  48. dict["y"] = BDE("r");
  49. BDE rDict = BDE::dict();
  50. rDict["id"] = BDE(localNode->getID(), DHT_ID_LENGTH);
  51. rDict["nodes"] = compactNodeInfo;
  52. dict["r"] = rDict;
  53. CPPUNIT_ASSERT_EQUAL(bencode::encode(dict), msgbody);
  54. }
  55. } // namespace aria2