diff options
| -rw-r--r-- | src/DefaultBtMessageDispatcher.cc | 12 | ||||
| -rw-r--r-- | src/DefaultBtMessageDispatcher.h | 8 | ||||
| -rw-r--r-- | src/PeerInteractionCommand.cc | 2 | ||||
| -rw-r--r-- | test/DefaultBtMessageDispatcherTest.cc | 36 |
4 files changed, 0 insertions, 58 deletions
diff --git a/src/DefaultBtMessageDispatcher.cc b/src/DefaultBtMessageDispatcher.cc index a0749470..e94427a4 100644 --- a/src/DefaultBtMessageDispatcher.cc +++ b/src/DefaultBtMessageDispatcher.cc @@ -64,8 +64,6 @@ namespace aria2 { DefaultBtMessageDispatcher::DefaultBtMessageDispatcher() : cuid_{0}, downloadContext_{nullptr}, - peerStorage_{nullptr}, - pieceStorage_{nullptr}, peerConnection_{nullptr}, messageFactory_{nullptr}, requestGroupMan_{nullptr}, @@ -315,16 +313,6 @@ void DefaultBtMessageDispatcher::setDownloadContext downloadContext_ = downloadContext; } -void DefaultBtMessageDispatcher::setPieceStorage(PieceStorage* pieceStorage) -{ - pieceStorage_ = pieceStorage; -} - -void DefaultBtMessageDispatcher::setPeerStorage(PeerStorage* peerStorage) -{ - peerStorage_ = peerStorage; -} - void DefaultBtMessageDispatcher::setBtMessageFactory(BtMessageFactory* factory) { messageFactory_ = factory; diff --git a/src/DefaultBtMessageDispatcher.h b/src/DefaultBtMessageDispatcher.h index 35afbd5b..a67c28ab 100644 --- a/src/DefaultBtMessageDispatcher.h +++ b/src/DefaultBtMessageDispatcher.h @@ -45,8 +45,6 @@ namespace aria2 { class DownloadContext; -class PeerStorage; -class PieceStorage; class BtMessage; class BtMessageFactory; class Peer; @@ -60,8 +58,6 @@ private: std::deque<std::unique_ptr<BtMessage> > messageQueue_; std::deque<std::unique_ptr<RequestSlot>> requestSlots_; DownloadContext* downloadContext_; - PeerStorage* peerStorage_; - PieceStorage* pieceStorage_; PeerConnection* peerConnection_; BtMessageFactory* messageFactory_; std::shared_ptr<Peer> peer_; @@ -135,10 +131,6 @@ public: void setDownloadContext(DownloadContext* downloadContext); - void setPieceStorage(PieceStorage* pieceStorage); - - void setPeerStorage(PeerStorage* peerStorage); - void setBtMessageFactory(BtMessageFactory* factory); void setRequestGroupMan(RequestGroupMan* rgman); diff --git a/src/PeerInteractionCommand.cc b/src/PeerInteractionCommand.cc index 8cc9635e..022cee14 100644 --- a/src/PeerInteractionCommand.cc +++ b/src/PeerInteractionCommand.cc @@ -180,8 +180,6 @@ PeerInteractionCommand::PeerInteractionCommand dispatcher->setCuid(cuid); dispatcher->setPeer(getPeer()); dispatcher->setDownloadContext(requestGroup_->getDownloadContext().get()); - dispatcher->setPieceStorage(pieceStorage.get()); - dispatcher->setPeerStorage(peerStorage.get()); dispatcher->setRequestTimeout(getOption()-> getAsInt(PREF_BT_REQUEST_TIMEOUT)); dispatcher->setBtMessageFactory(factory.get()); diff --git a/test/DefaultBtMessageDispatcherTest.cc b/test/DefaultBtMessageDispatcherTest.cc index 5ce5adef..7139db4a 100644 --- a/test/DefaultBtMessageDispatcherTest.cc +++ b/test/DefaultBtMessageDispatcherTest.cc @@ -6,8 +6,6 @@ #include "util.h" #include "Exception.h" -#include "MockPieceStorage.h" -#include "MockPeerStorage.h" #include "MockBtMessage.h" #include "MockBtMessageFactory.h" #include "prefs.h" @@ -44,8 +42,6 @@ private: std::shared_ptr<DownloadContext> dctx_; std::shared_ptr<Peer> peer; std::unique_ptr<DefaultBtMessageDispatcher> btMessageDispatcher; - std::unique_ptr<MockPeerStorage> peerStorage; - std::unique_ptr<MockPieceStorage> pieceStorage; std::unique_ptr<MockBtMessageFactory> messageFactory_; std::unique_ptr<RequestGroupMan> rgman_; std::shared_ptr<Option> option_; @@ -106,21 +102,6 @@ public: } }; - class MockPieceStorage2 : public MockPieceStorage { - private: - std::shared_ptr<Piece> piece; - public: - virtual std::shared_ptr<Piece> getPiece(size_t index) CXX11_OVERRIDE - { - return piece; - } - - void setPiece(const std::shared_ptr<Piece>& piece) - { - this->piece = piece; - } - }; - class MockBtMessageFactory2 : public MockBtMessageFactory { public: virtual std::unique_ptr<BtCancelMessage> @@ -146,9 +127,6 @@ public: peer = std::make_shared<Peer>("192.168.0.1", 6969); peer->allocateSessionResource (dctx_->getPieceLength(), dctx_->getTotalLength()); - peerStorage = make_unique<MockPeerStorage>(); - pieceStorage = make_unique<MockPieceStorage>(); - messageFactory_ = make_unique<MockBtMessageFactory2>(); rgman_ = make_unique<RequestGroupMan> @@ -157,8 +135,6 @@ public: btMessageDispatcher = make_unique<DefaultBtMessageDispatcher>(); btMessageDispatcher->setPeer(peer); btMessageDispatcher->setDownloadContext(dctx_.get()); - btMessageDispatcher->setPieceStorage(pieceStorage.get()); - btMessageDispatcher->setPeerStorage(peerStorage.get()); btMessageDispatcher->setBtMessageFactory(messageFactory_.get()); btMessageDispatcher->setCuid(1); btMessageDispatcher->setRequestGroupMan(rgman_.get()); @@ -233,11 +209,7 @@ void DefaultBtMessageDispatcherTest::testCheckRequestSlotAndDoNecessaryThing() CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index)); CPPUNIT_ASSERT_EQUAL((size_t)0, index); - auto pieceStorage = make_unique<MockPieceStorage2>(); - pieceStorage->setPiece(piece); - btMessageDispatcher->setRequestTimeout(60); - btMessageDispatcher->setPieceStorage(pieceStorage.get()); btMessageDispatcher->addOutstandingRequest (make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece)); @@ -256,11 +228,7 @@ testCheckRequestSlotAndDoNecessaryThing_timeout() { CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index)); CPPUNIT_ASSERT_EQUAL((size_t)0, index); - auto pieceStorage = make_unique<MockPieceStorage2>(); - pieceStorage->setPiece(piece); - btMessageDispatcher->setRequestTimeout(60); - btMessageDispatcher->setPieceStorage(pieceStorage.get()); auto slot = make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece); // make this slot timeout slot->setDispatchedTime(0); @@ -279,11 +247,7 @@ void DefaultBtMessageDispatcherTest:: testCheckRequestSlotAndDoNecessaryThing_completeBlock() { auto piece = std::make_shared<Piece>(0, MY_PIECE_LENGTH); piece->completeBlock(0); - auto pieceStorage = make_unique<MockPieceStorage2>(); - pieceStorage->setPiece(piece); - btMessageDispatcher->setRequestTimeout(60); - btMessageDispatcher->setPieceStorage(pieceStorage.get()); btMessageDispatcher->addOutstandingRequest (make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece)); |
