BtHaveAllMessageTest.cc 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. #include "BtHaveAllMessage.h"
  2. #include "PeerMessageUtil.h"
  3. #include "Peer.h"
  4. #include "MockPieceStorage.h"
  5. #include <cstring>
  6. #include <cppunit/extensions/HelperMacros.h>
  7. namespace aria2 {
  8. class BtHaveAllMessageTest:public CppUnit::TestFixture {
  9. CPPUNIT_TEST_SUITE(BtHaveAllMessageTest);
  10. CPPUNIT_TEST(testCreate);
  11. CPPUNIT_TEST(testGetMessage);
  12. CPPUNIT_TEST(testDoReceivedAction);
  13. CPPUNIT_TEST_SUITE_END();
  14. private:
  15. public:
  16. void setUp() {
  17. }
  18. void testCreate();
  19. void testGetMessage();
  20. void testDoReceivedAction();
  21. };
  22. CPPUNIT_TEST_SUITE_REGISTRATION(BtHaveAllMessageTest);
  23. void BtHaveAllMessageTest::testCreate() {
  24. unsigned char msg[5];
  25. PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 1, 14);
  26. SharedHandle<BtHaveAllMessage> pm = BtHaveAllMessage::create(&msg[4], 1);
  27. CPPUNIT_ASSERT_EQUAL((uint8_t)14, pm->getId());
  28. // case: payload size is wrong
  29. try {
  30. unsigned char msg[6];
  31. PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 2, 14);
  32. BtHaveAllMessage::create(&msg[4], 2);
  33. CPPUNIT_FAIL("exception must be thrown.");
  34. } catch(...) {
  35. }
  36. // case: id is wrong
  37. try {
  38. unsigned char msg[5];
  39. PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 1, 15);
  40. BtHaveAllMessage::create(&msg[4], 1);
  41. CPPUNIT_FAIL("exception must be thrown.");
  42. } catch(...) {
  43. }
  44. }
  45. void BtHaveAllMessageTest::testGetMessage() {
  46. BtHaveAllMessage msg;
  47. unsigned char data[5];
  48. PeerMessageUtil::createPeerMessageString(data, sizeof(data), 1, 14);
  49. CPPUNIT_ASSERT(memcmp(msg.getMessage(), data, 5) == 0);
  50. }
  51. void BtHaveAllMessageTest::testDoReceivedAction() {
  52. BtHaveAllMessage msg;
  53. SharedHandle<Peer> peer(new Peer("host", 6969));
  54. peer->allocateSessionResource(16*1024, 256*1024);
  55. peer->setFastExtensionEnabled(true);
  56. msg.setPeer(peer);
  57. SharedHandle<MockPieceStorage> pieceStorage(new MockPieceStorage());
  58. msg.setPieceStorage(pieceStorage);
  59. msg.doReceivedAction();
  60. CPPUNIT_ASSERT(peer->isSeeder());
  61. peer->setFastExtensionEnabled(false);
  62. try {
  63. msg.doReceivedAction();
  64. CPPUNIT_FAIL("exception must be thrown.");
  65. } catch(...) {}
  66. }
  67. } // namespace aria2