diff options
Diffstat (limited to 'test/BtExtendedMessageTest.cc')
| -rw-r--r-- | test/BtExtendedMessageTest.cc | 7 |
1 files changed, 3 insertions, 4 deletions
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() |