summaryrefslogtreecommitdiff
path: root/test/BtExtendedMessageTest.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2016-05-15 23:40:09 +0900
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2016-05-15 23:40:52 +0900
commita7237c69f7befce7af174087bf770acd0187fb9b (patch)
treec91ede2c506785956c55bfad0dce82a0c73ceea2 /test/BtExtendedMessageTest.cc
parentcc8375f0b04918ace028bf58e09f9e25a0440c23 (diff)
Refactor BitTorrent message buffer usage
Diffstat (limited to 'test/BtExtendedMessageTest.cc')
-rw-r--r--test/BtExtendedMessageTest.cc6
1 files changed, 3 insertions, 3 deletions
diff --git a/test/BtExtendedMessageTest.cc b/test/BtExtendedMessageTest.cc
index 65d4fdbc..5e7d4ca6 100644
--- a/test/BtExtendedMessageTest.cc
+++ b/test/BtExtendedMessageTest.cc
@@ -78,9 +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;
+ auto rawmsg = msg.createMessage();
+ CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size());
+ CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data));
CPPUNIT_ASSERT_EQUAL((size_t)17, msg.getMessageLength());
}