summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/DefaultBtMessageDispatcher.cc12
-rw-r--r--src/DefaultBtMessageDispatcher.h8
-rw-r--r--src/PeerInteractionCommand.cc2
-rw-r--r--test/DefaultBtMessageDispatcherTest.cc36
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));