diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/BtAllowedFastMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtBitfieldMessageTest.cc | 7 | ||||
| -rw-r--r-- | test/BtCancelMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtChokeMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtExtendedMessageTest.cc | 7 | ||||
| -rw-r--r-- | test/BtHandshakeMessageTest.cc | 22 | ||||
| -rw-r--r-- | test/BtHaveAllMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtHaveMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtHaveNoneMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtInterestedMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtKeepAliveMessageTest.cc | 7 | ||||
| -rw-r--r-- | test/BtNotInterestedMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtPortMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtRejectMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtRequestMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtSuggestPieceMessageTest.cc | 6 | ||||
| -rw-r--r-- | test/BtUnchokeMessageTest.cc | 6 |
17 files changed, 59 insertions, 62 deletions
diff --git a/test/BtAllowedFastMessageTest.cc b/test/BtAllowedFastMessageTest.cc index ccc2cad8..ee2509ae 100644 --- a/test/BtAllowedFastMessageTest.cc +++ b/test/BtAllowedFastMessageTest.cc @@ -70,9 +70,9 @@ void BtAllowedFastMessageTest::testCreateMessage() unsigned char data[9]; bittorrent::createPeerMessageString(data, sizeof(data), 5, 17); bittorrent::setIntParam(&data[5], 12345); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 9) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)9, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtAllowedFastMessageTest::testDoReceivedAction() diff --git a/test/BtBitfieldMessageTest.cc b/test/BtBitfieldMessageTest.cc index a3059381..90b07810 100644 --- a/test/BtBitfieldMessageTest.cc +++ b/test/BtBitfieldMessageTest.cc @@ -76,10 +76,9 @@ void BtBitfieldMessageTest::testCreateMessage() unsigned char data[5 + 2]; bittorrent::createPeerMessageString(data, sizeof(data), 3, 5); memcpy(&data[5], bitfield, sizeof(bitfield)); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 7) == 0); - delete[] rawmsg; - CPPUNIT_ASSERT_EQUAL((size_t)7, msg.getMessageLength()); + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)7, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtBitfieldMessageTest::testDoReceivedAction() diff --git a/test/BtCancelMessageTest.cc b/test/BtCancelMessageTest.cc index f1a7e0a8..37ee4987 100644 --- a/test/BtCancelMessageTest.cc +++ b/test/BtCancelMessageTest.cc @@ -95,9 +95,9 @@ void BtCancelMessageTest::testCreateMessage() bittorrent::setIntParam(&data[5], 12345); bittorrent::setIntParam(&data[9], 256); bittorrent::setIntParam(&data[13], 1_k); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 17) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtCancelMessageTest::testDoReceivedAction() diff --git a/test/BtChokeMessageTest.cc b/test/BtChokeMessageTest.cc index fa73742b..debd2530 100644 --- a/test/BtChokeMessageTest.cc +++ b/test/BtChokeMessageTest.cc @@ -109,9 +109,9 @@ void BtChokeMessageTest::testCreateMessage() BtChokeMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 0); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtChokeMessageTest::testDoReceivedAction() diff --git a/test/BtExtendedMessageTest.cc b/test/BtExtendedMessageTest.cc index 65d4fdbc..f32f3426 100644 --- a/test/BtExtendedMessageTest.cc +++ b/test/BtExtendedMessageTest.cc @@ -78,10 +78,9 @@ void BtExtendedMessageTest::testCreateMessage() bittorrent::createPeerMessageString(data, sizeof(data), 13, 20); *(data + 5) = extendedMessageID; memcpy(data + 6, payload.c_str(), payload.size()); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 17) == 0); - delete[] rawmsg; - CPPUNIT_ASSERT_EQUAL((size_t)17, msg.getMessageLength()); + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtExtendedMessageTest::testDoReceivedAction() diff --git a/test/BtHandshakeMessageTest.cc b/test/BtHandshakeMessageTest.cc index a86ae54a..b8569d73 100644 --- a/test/BtHandshakeMessageTest.cc +++ b/test/BtHandshakeMessageTest.cc @@ -73,23 +73,23 @@ void BtHandshakeMessageTest::testCreate() void BtHandshakeMessageTest::testCreateMessage() { - unsigned char infoHash[] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, - 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, - 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}; - unsigned char peerId[] = {0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, - 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, - 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0}; - - std::shared_ptr<BtHandshakeMessage> msg(new BtHandshakeMessage()); + constexpr unsigned char infoHash[] = { + 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}; + constexpr unsigned char peerId[] = {0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, + 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, + 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0}; + + auto msg = std::make_shared<BtHandshakeMessage>(); msg->setInfoHash(infoHash); msg->setPeerId(peerId); unsigned char data[68]; createHandshakeMessageData(data); - unsigned char* rawmsg = msg->createMessage(); + auto rawmsg = msg->createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)68, rawmsg.size()); CPPUNIT_ASSERT_EQUAL(util::toHex((const unsigned char*)data, 68), - util::toHex(rawmsg, 68)); - delete[] rawmsg; + util::toHex(rawmsg.data(), 68)); } void BtHandshakeMessageTest::testToString() diff --git a/test/BtHaveAllMessageTest.cc b/test/BtHaveAllMessageTest.cc index 56e9504f..9bff0632 100644 --- a/test/BtHaveAllMessageTest.cc +++ b/test/BtHaveAllMessageTest.cc @@ -65,9 +65,9 @@ void BtHaveAllMessageTest::testCreateMessage() BtHaveAllMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 14); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtHaveAllMessageTest::testDoReceivedAction() diff --git a/test/BtHaveMessageTest.cc b/test/BtHaveMessageTest.cc index 562a5033..677ff2b2 100644 --- a/test/BtHaveMessageTest.cc +++ b/test/BtHaveMessageTest.cc @@ -71,9 +71,9 @@ void BtHaveMessageTest::testCreateMessage() unsigned char data[9]; bittorrent::createPeerMessageString(data, sizeof(data), 5, 4); bittorrent::setIntParam(&data[5], 12345); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 9) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)9, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtHaveMessageTest::testDoReceivedAction() diff --git a/test/BtHaveNoneMessageTest.cc b/test/BtHaveNoneMessageTest.cc index c5ee6d0d..71ab3f3e 100644 --- a/test/BtHaveNoneMessageTest.cc +++ b/test/BtHaveNoneMessageTest.cc @@ -62,9 +62,9 @@ void BtHaveNoneMessageTest::testCreateMessage() BtHaveNoneMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 15); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtHaveNoneMessageTest::testDoReceivedAction() diff --git a/test/BtInterestedMessageTest.cc b/test/BtInterestedMessageTest.cc index c3096d3e..79e73216 100644 --- a/test/BtInterestedMessageTest.cc +++ b/test/BtInterestedMessageTest.cc @@ -64,9 +64,9 @@ void BtInterestedMessageTest::testCreateMessage() BtInterestedMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 2); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtInterestedMessageTest::testDoReceivedAction() diff --git a/test/BtKeepAliveMessageTest.cc b/test/BtKeepAliveMessageTest.cc index 3bd27543..2d9d4872 100644 --- a/test/BtKeepAliveMessageTest.cc +++ b/test/BtKeepAliveMessageTest.cc @@ -27,10 +27,9 @@ void BtKeepAliveMessageTest::testCreateMessage() memset(data, 0, sizeof(data)); BtKeepAliveMessage message; CPPUNIT_ASSERT_EQUAL((uint8_t)99, message.getId()); - CPPUNIT_ASSERT_EQUAL((size_t)4, message.getMessageLength()); - unsigned char* rawmsg = message.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 4) == 0); - delete[] rawmsg; + auto rawmsg = message.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)4, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtKeepAliveMessageTest::testToString() diff --git a/test/BtNotInterestedMessageTest.cc b/test/BtNotInterestedMessageTest.cc index 1f3bbac5..88d233c2 100644 --- a/test/BtNotInterestedMessageTest.cc +++ b/test/BtNotInterestedMessageTest.cc @@ -64,9 +64,9 @@ void BtNotInterestedMessageTest::testCreateMessage() BtNotInterestedMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 3); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtNotInterestedMessageTest::testDoReceivedAction() diff --git a/test/BtPortMessageTest.cc b/test/BtPortMessageTest.cc index cceb893c..c0370efc 100644 --- a/test/BtPortMessageTest.cc +++ b/test/BtPortMessageTest.cc @@ -99,9 +99,9 @@ void BtPortMessageTest::testCreateMessage() unsigned char data[7]; bittorrent::createPeerMessageString(data, sizeof(data), 3, 9); bittorrent::setShortIntParam(&data[5], 6881); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 7) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)7, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtPortMessageTest::testDoReceivedAction() diff --git a/test/BtRejectMessageTest.cc b/test/BtRejectMessageTest.cc index 3e3b7f8e..3c85ff92 100644 --- a/test/BtRejectMessageTest.cc +++ b/test/BtRejectMessageTest.cc @@ -129,9 +129,9 @@ void BtRejectMessageTest::testCreateMessage() bittorrent::setIntParam(&data[5], 12345); bittorrent::setIntParam(&data[9], 256); bittorrent::setIntParam(&data[13], 1_k); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 17) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtRejectMessageTest::testDoReceivedAction() diff --git a/test/BtRequestMessageTest.cc b/test/BtRequestMessageTest.cc index 3ff3d7a9..a68fb4d4 100644 --- a/test/BtRequestMessageTest.cc +++ b/test/BtRequestMessageTest.cc @@ -151,9 +151,9 @@ void BtRequestMessageTest::testCreateMessage() bittorrent::setIntParam(&data[5], 12345); bittorrent::setIntParam(&data[9], 256); bittorrent::setIntParam(&data[13], 1_k); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 17) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtRequestMessageTest::testDoReceivedAction_hasPieceAndAmNotChoking() diff --git a/test/BtSuggestPieceMessageTest.cc b/test/BtSuggestPieceMessageTest.cc index 7e48628e..9be0b8b1 100644 --- a/test/BtSuggestPieceMessageTest.cc +++ b/test/BtSuggestPieceMessageTest.cc @@ -62,9 +62,9 @@ void BtSuggestPieceMessageTest::testCreateMessage() unsigned char data[9]; bittorrent::createPeerMessageString(data, sizeof(data), 5, 13); bittorrent::setIntParam(&data[5], 12345); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 9) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)9, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtSuggestPieceMessageTest::testToString() diff --git a/test/BtUnchokeMessageTest.cc b/test/BtUnchokeMessageTest.cc index dea19a9d..1fe0bbe9 100644 --- a/test/BtUnchokeMessageTest.cc +++ b/test/BtUnchokeMessageTest.cc @@ -64,9 +64,9 @@ void BtUnchokeMessageTest::testCreateMessage() BtUnchokeMessage msg; unsigned char data[5]; bittorrent::createPeerMessageString(data, sizeof(data), 1, 1); - unsigned char* rawmsg = msg.createMessage(); - CPPUNIT_ASSERT(memcmp(rawmsg, data, 5) == 0); - delete[] rawmsg; + auto rawmsg = msg.createMessage(); + CPPUNIT_ASSERT_EQUAL((size_t)5, rawmsg.size()); + CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data)); } void BtUnchokeMessageTest::testDoReceivedAction() |
