summaryrefslogtreecommitdiff
path: root/test/HandshakeExtensionMessageTest.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2013-06-23 00:59:55 +0900
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2013-06-23 00:59:55 +0900
commitbef6236da80cb18a3212bde7c33c556883cbd536 (patch)
tree8faeefde90feeaeb0d7c52d75dcc5189f8c25ea0 /test/HandshakeExtensionMessageTest.cc
parent1a299c4d7cf5ab8d130e678d152c4c47ae3430ea (diff)
DownloadContext::attrs_ now holds std::unique_ptr
DownloadContext::getAttribute() returns a raw pointer.
Diffstat (limited to 'test/HandshakeExtensionMessageTest.cc')
-rw-r--r--test/HandshakeExtensionMessageTest.cc4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/HandshakeExtensionMessageTest.cc b/test/HandshakeExtensionMessageTest.cc
index 218b3e27..bdd16cdf 100644
--- a/test/HandshakeExtensionMessageTest.cc
+++ b/test/HandshakeExtensionMessageTest.cc
@@ -97,8 +97,7 @@ void HandshakeExtensionMessageTest::testDoReceivedAction()
RequestGroup rg(GroupId::create(), op);
rg.setDownloadContext(dctx);
- std::shared_ptr<TorrentAttribute> attrs(new TorrentAttribute());
- dctx->setAttribute(CTX_ATTR_BT, attrs);
+ dctx->setAttribute(CTX_ATTR_BT, make_unique<TorrentAttribute>());
dctx->markTotalLengthIsUnknown();
std::shared_ptr<Peer> peer(new Peer("192.168.0.1", 0));
@@ -122,6 +121,7 @@ void HandshakeExtensionMessageTest::testDoReceivedAction()
peer->getExtensionMessageID
(ExtensionMessageRegistry::UT_METADATA));
CPPUNIT_ASSERT(peer->isSeeder());
+ auto attrs = bittorrent::getTorrentAttrs(dctx);
CPPUNIT_ASSERT_EQUAL((size_t)1024, attrs->metadataSize);
CPPUNIT_ASSERT_EQUAL((int64_t)1024, dctx->getTotalLength());
CPPUNIT_ASSERT(dctx->knowsTotalLength());