diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2009-11-22 15:31:47 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2009-11-22 15:31:47 +0000 |
| commit | c1730aeea945f80b5bf152dc6e7b7d9a42a40b5c (patch) | |
| tree | 801f8ea1893fdf0e6d3b22ba29cef3d0a2262ad9 /test/HandshakeExtensionMessageTest.cc | |
| parent | b563621dd13a5598cfa805290a95caec51bb7e8b (diff) | |
2009-11-23 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Fixed ut_metadata data handling. Implemented
UTMetadataDataExtensionMessage::doReceivedAction(). Initialize
PeerStorage in HandshakeExtensionMessage::doReceivedAction() when
metadata_size is received.
* src/DefaultExtensionMessageFactory.cc
* src/DefaultExtensionMessageFactory.h
* src/HandshakeExtensionMessage.cc
* src/HandshakeExtensionMessage.h
* src/UTMetadataDataExtensionMessage.cc
* src/UTMetadataDataExtensionMessage.h
* src/UTMetadataRequestExtensionMessage.cc
* src/UTMetadataRequestFactory.cc
* src/UTMetadataRequestFactory.h
* src/UTMetadataRequestTracker.cc
* src/UTMetadataRequestTracker.h
* test/DefaultExtensionMessageFactoryTest.cc
* test/HandshakeExtensionMessageTest.cc
* test/MockBtMessage.h
* test/UTMetadataDataExtensionMessageTest.cc
* test/UTMetadataRequestExtensionMessageTest.cc
* test/UTMetadataRequestFactoryTest.cc
* test/UTMetadataRequestTrackerTest.cc
* test/extension_message_test_helper.h
Diffstat (limited to 'test/HandshakeExtensionMessageTest.cc')
| -rw-r--r-- | test/HandshakeExtensionMessageTest.cc | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/test/HandshakeExtensionMessageTest.cc b/test/HandshakeExtensionMessageTest.cc index b8ceb4dc..03818f6e 100644 --- a/test/HandshakeExtensionMessageTest.cc +++ b/test/HandshakeExtensionMessageTest.cc @@ -9,6 +9,8 @@ #include "FileEntry.h" #include "DownloadContext.h" #include "bittorrent_helper.h" +#include "Option.h" +#include "RequestGroup.h" namespace aria2 { @@ -88,9 +90,16 @@ void HandshakeExtensionMessageTest::testToString() void HandshakeExtensionMessageTest::testDoReceivedAction() { - SharedHandle<DownloadContext> ctx(new DownloadContext()); + SharedHandle<DownloadContext> dctx + (new DownloadContext(METADATA_PIECE_SIZE, 0)); + SharedHandle<Option> op(new Option()); + RequestGroup rg(op); + rg.setDownloadContext(dctx); + dctx->setOwnerRequestGroup(&rg); + BDE attrs = BDE::dict(); - ctx->setAttribute(bittorrent::BITTORRENT, attrs); + dctx->setAttribute(bittorrent::BITTORRENT, attrs); + dctx->markTotalLengthIsUnknown(); SharedHandle<Peer> peer(new Peer("192.168.0.1", 0)); peer->allocateSessionResource(1024, 1024*1024); @@ -101,7 +110,7 @@ void HandshakeExtensionMessageTest::testDoReceivedAction() msg.setExtension("a2_dht", 2); msg.setMetadataSize(1024); msg.setPeer(peer); - msg.setDownloadContext(ctx); + msg.setDownloadContext(dctx); msg.doReceivedAction(); @@ -109,6 +118,8 @@ void HandshakeExtensionMessageTest::testDoReceivedAction() CPPUNIT_ASSERT_EQUAL((uint8_t)1, peer->getExtensionMessageID("ut_pex")); CPPUNIT_ASSERT_EQUAL((uint8_t)2, peer->getExtensionMessageID("a2_dht")); CPPUNIT_ASSERT_EQUAL((int64_t)1024, attrs[bittorrent::METADATA_SIZE].i()); + CPPUNIT_ASSERT_EQUAL((uint64_t)1024, dctx->getTotalLength()); + CPPUNIT_ASSERT(dctx->knowsTotalLength()); } void HandshakeExtensionMessageTest::testCreate() |