aria2/test/MockPieceStorage.h
2015-12-27 18:40:08 +09:00

302 lines
8.0 KiB
C++

#ifndef D_MOCK_PIECE_STORAGE_H
#define D_MOCK_PIECE_STORAGE_H
#include "PieceStorage.h"
#include <algorithm>
#include "BitfieldMan.h"
#include "Piece.h"
#include "DiskAdaptor.h"
namespace aria2 {
class MockPieceStorage : public PieceStorage {
private:
int64_t totalLength;
int64_t filteredTotalLength;
int64_t completedLength;
int64_t filteredCompletedLength;
BitfieldMan* bitfieldMan;
bool selectiveDownloadingMode;
bool endGame;
std::shared_ptr<DiskAdaptor> diskAdaptor;
std::deque<int32_t> pieceLengthList;
std::deque<std::shared_ptr<Piece>> inFlightPieces;
bool downloadFinished_;
bool allDownloadFinished_;
public:
MockPieceStorage()
: totalLength(0),
filteredTotalLength(0),
completedLength(0),
filteredCompletedLength(0),
bitfieldMan(0),
selectiveDownloadingMode(false),
endGame(false),
downloadFinished_(false),
allDownloadFinished_(false)
{
}
virtual ~MockPieceStorage() {}
#ifdef ENABLE_BITTORRENT
virtual bool hasMissingPiece(const std::shared_ptr<Peer>& peer) CXX11_OVERRIDE
{
return false;
}
virtual void getMissingPiece(std::vector<std::shared_ptr<Piece>>& pieces,
size_t minMissingBlocks,
const std::shared_ptr<Peer>& peer,
cuid_t cuid) CXX11_OVERRIDE
{
}
virtual void getMissingPiece(std::vector<std::shared_ptr<Piece>>& pieces,
size_t minMissingBlocks,
const std::shared_ptr<Peer>& peer,
const std::vector<size_t>& excludedIndexes,
cuid_t cuid) CXX11_OVERRIDE
{
}
virtual void getMissingFastPiece(std::vector<std::shared_ptr<Piece>>& pieces,
size_t minMissingBlocks,
const std::shared_ptr<Peer>& peer,
cuid_t cuid) CXX11_OVERRIDE
{
}
virtual void getMissingFastPiece(std::vector<std::shared_ptr<Piece>>& pieces,
size_t minMissingBlocks,
const std::shared_ptr<Peer>& peer,
const std::vector<size_t>& excludedIndexes,
cuid_t cuid) CXX11_OVERRIDE
{
}
virtual std::shared_ptr<Piece>
getMissingPiece(const std::shared_ptr<Peer>& peer, cuid_t cuid) CXX11_OVERRIDE
{
return std::shared_ptr<Piece>(new Piece());
}
virtual std::shared_ptr<Piece>
getMissingPiece(const std::shared_ptr<Peer>& peer,
const std::vector<size_t>& excludedIndexes,
cuid_t cuid) CXX11_OVERRIDE
{
return std::shared_ptr<Piece>(new Piece());
}
#endif // ENABLE_BITTORRENT
virtual bool hasMissingUnusedPiece() CXX11_OVERRIDE { return false; }
virtual std::shared_ptr<Piece>
getMissingPiece(size_t minSplitSize, const unsigned char* ignoreBitfield,
size_t length, cuid_t cuid) CXX11_OVERRIDE
{
return std::shared_ptr<Piece>(new Piece());
}
virtual std::shared_ptr<Piece> getMissingPiece(size_t index,
cuid_t cuid) CXX11_OVERRIDE
{
return std::shared_ptr<Piece>(new Piece());
}
virtual bool isPieceUsed(size_t index) CXX11_OVERRIDE { return false; }
virtual void markPieceMissing(size_t index) CXX11_OVERRIDE {}
virtual void markPiecesDone(int64_t) CXX11_OVERRIDE {}
virtual std::shared_ptr<Piece> getPiece(size_t index) CXX11_OVERRIDE
{
return std::shared_ptr<Piece>(new Piece());
}
virtual void completePiece(const std::shared_ptr<Piece>& piece) CXX11_OVERRIDE
{
}
virtual void cancelPiece(const std::shared_ptr<Piece>& piece,
cuid_t cuid) CXX11_OVERRIDE
{
}
virtual bool hasPiece(size_t index) CXX11_OVERRIDE { return false; }
virtual int64_t getTotalLength() CXX11_OVERRIDE { return totalLength; }
void setTotalLength(int64_t totalLength) { this->totalLength = totalLength; }
virtual int64_t getFilteredTotalLength() CXX11_OVERRIDE
{
return filteredTotalLength;
}
void setFilteredTotalLength(int64_t totalLength)
{
this->filteredTotalLength = totalLength;
}
virtual int64_t getCompletedLength() CXX11_OVERRIDE
{
return completedLength;
}
void setCompletedLength(int64_t completedLength)
{
this->completedLength = completedLength;
}
virtual int64_t getFilteredCompletedLength() CXX11_OVERRIDE
{
return filteredCompletedLength;
}
void setFilteredCompletedLength(int64_t completedLength)
{
this->filteredCompletedLength = completedLength;
}
virtual void setupFileFilter() CXX11_OVERRIDE {}
virtual void clearFileFilter() CXX11_OVERRIDE {}
virtual bool downloadFinished() CXX11_OVERRIDE { return downloadFinished_; }
void setDownloadFinished(bool f) { downloadFinished_ = f; }
virtual bool allDownloadFinished() CXX11_OVERRIDE
{
return allDownloadFinished_;
}
void setAllDownloadFinished(bool f) { allDownloadFinished_ = f; }
virtual void initStorage() CXX11_OVERRIDE {}
virtual const unsigned char* getBitfield() CXX11_OVERRIDE
{
return bitfieldMan->getBitfield();
}
virtual void setBitfield(const unsigned char* bitfield,
size_t bitfieldLength) CXX11_OVERRIDE
{
bitfieldMan->setBitfield(bitfield, bitfieldLength);
}
virtual size_t getBitfieldLength() CXX11_OVERRIDE
{
return bitfieldMan->getBitfieldLength();
}
void setBitfield(BitfieldMan* bitfieldMan)
{
this->bitfieldMan = bitfieldMan;
}
virtual bool isSelectiveDownloadingMode() CXX11_OVERRIDE
{
return selectiveDownloadingMode;
}
void setSelectiveDownloadingMode(bool flag)
{
this->selectiveDownloadingMode = flag;
}
virtual bool isEndGame() CXX11_OVERRIDE { return endGame; }
virtual void setEndGamePieceNum(size_t num) CXX11_OVERRIDE {}
virtual void enterEndGame() CXX11_OVERRIDE { this->endGame = true; }
virtual std::shared_ptr<DiskAdaptor> getDiskAdaptor() CXX11_OVERRIDE
{
return diskAdaptor;
}
virtual WrDiskCache* getWrDiskCache() CXX11_OVERRIDE { return 0; }
virtual void flushWrDiskCacheEntry() CXX11_OVERRIDE {}
void setDiskAdaptor(const std::shared_ptr<DiskAdaptor>& adaptor)
{
this->diskAdaptor = adaptor;
}
virtual int32_t getPieceLength(size_t index) CXX11_OVERRIDE
{
return pieceLengthList.at(index);
}
void addPieceLengthList(int32_t length) { pieceLengthList.push_back(length); }
virtual void advertisePiece(cuid_t cuid, size_t index) CXX11_OVERRIDE {}
virtual void
getAdvertisedPieceIndexes(std::vector<size_t>& indexes, cuid_t myCuid,
const Timer& lastCheckTime) CXX11_OVERRIDE
{
}
virtual void
removeAdvertisedPiece(const std::chrono::seconds& elapsed) CXX11_OVERRIDE
{
}
virtual void markAllPiecesDone() CXX11_OVERRIDE {}
virtual void addInFlightPiece(
const std::vector<std::shared_ptr<Piece>>& pieces) CXX11_OVERRIDE
{
std::copy(pieces.begin(), pieces.end(), back_inserter(inFlightPieces));
}
virtual size_t countInFlightPiece() CXX11_OVERRIDE
{
return inFlightPieces.size();
}
virtual void
getInFlightPieces(std::vector<std::shared_ptr<Piece>>& pieces) CXX11_OVERRIDE
{
pieces.insert(pieces.end(), inFlightPieces.begin(), inFlightPieces.end());
}
virtual void addPieceStats(size_t index) CXX11_OVERRIDE {}
virtual void addPieceStats(const unsigned char* bitfield,
size_t bitfieldLength) CXX11_OVERRIDE
{
}
virtual void subtractPieceStats(const unsigned char* bitfield,
size_t bitfieldLength) CXX11_OVERRIDE
{
}
virtual void updatePieceStats(const unsigned char* newBitfield,
size_t newBitfieldLength,
const unsigned char* oldBitfield) CXX11_OVERRIDE
{
}
virtual size_t getNextUsedIndex(size_t index) CXX11_OVERRIDE { return 0; }
virtual void onDownloadIncomplete() CXX11_OVERRIDE {}
};
} // namespace aria2
#endif // D_MOCK_PIECE_STORAGE_H