BtNotInterestedMessageTest.cc 2.5 KB

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