DHTAnnouncePeerReplyMessageTest.cc 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #include "DHTAnnouncePeerReplyMessage.h"
  2. #include <cppunit/extensions/HelperMacros.h>
  3. #include "DHTNode.h"
  4. #include "DHTUtil.h"
  5. #include "Exception.h"
  6. #include "Util.h"
  7. #include "bencode.h"
  8. namespace aria2 {
  9. class DHTAnnouncePeerReplyMessageTest:public CppUnit::TestFixture {
  10. CPPUNIT_TEST_SUITE(DHTAnnouncePeerReplyMessageTest);
  11. CPPUNIT_TEST(testGetBencodedMessage);
  12. CPPUNIT_TEST_SUITE_END();
  13. public:
  14. void setUp() {}
  15. void tearDown() {}
  16. void testGetBencodedMessage();
  17. };
  18. CPPUNIT_TEST_SUITE_REGISTRATION(DHTAnnouncePeerReplyMessageTest);
  19. void DHTAnnouncePeerReplyMessageTest::testGetBencodedMessage()
  20. {
  21. SharedHandle<DHTNode> localNode(new DHTNode());
  22. SharedHandle<DHTNode> remoteNode(new DHTNode());
  23. unsigned char tid[DHT_TRANSACTION_ID_LENGTH];
  24. DHTUtil::generateRandomData(tid, DHT_TRANSACTION_ID_LENGTH);
  25. std::string transactionID(&tid[0], &tid[DHT_TRANSACTION_ID_LENGTH]);
  26. DHTAnnouncePeerReplyMessage msg(localNode, remoteNode, transactionID);
  27. std::string msgbody = msg.getBencodedMessage();
  28. BDE dict = BDE::dict();
  29. dict["t"] = transactionID;
  30. dict["y"] = BDE("r");
  31. BDE rDict = BDE::dict();
  32. rDict["id"] = BDE(localNode->getID(), DHT_ID_LENGTH);
  33. dict["r"] = rDict;
  34. CPPUNIT_ASSERT_EQUAL(bencode::encode(dict), msgbody);
  35. }
  36. } // namespace aria2