123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- #ifndef D_MOCK_BT_MESSAGE_FACTORY_H
- #define D_MOCK_BT_MESSAGE_FACTORY_H
- #include "BtMessageFactory.h"
- #include "BtHandshakeMessage.h"
- #include "BtRequestMessage.h"
- #include "BtCancelMessage.h"
- #include "BtPieceMessage.h"
- #include "BtHaveMessage.h"
- #include "BtChokeMessage.h"
- #include "BtUnchokeMessage.h"
- #include "BtInterestedMessage.h"
- #include "BtNotInterestedMessage.h"
- #include "BtBitfieldMessage.h"
- #include "BtKeepAliveMessage.h"
- #include "BtHaveAllMessage.h"
- #include "BtHaveNoneMessage.h"
- #include "BtRejectMessage.h"
- #include "BtAllowedFastMessage.h"
- #include "BtPortMessage.h"
- #include "BtExtendedMessage.h"
- namespace aria2 {
- class ExtensionMessage;
- class MockBtMessageFactory : public BtMessageFactory {
- public:
- MockBtMessageFactory() {}
- virtual ~MockBtMessageFactory() {}
- virtual std::unique_ptr<BtMessage>
- createBtMessage(const unsigned char* msg, size_t msgLength) {
- return std::unique_ptr<BtMessage>{};
- };
- virtual std::unique_ptr<BtHandshakeMessage>
- createHandshakeMessage(const unsigned char* msg, size_t msgLength) {
- return std::unique_ptr<BtHandshakeMessage>{};
- }
- virtual std::unique_ptr<BtHandshakeMessage>
- createHandshakeMessage(const unsigned char* infoHash,
- const unsigned char* peerId) {
- return std::unique_ptr<BtHandshakeMessage>{};
- }
- virtual std::unique_ptr<BtRequestMessage>
- createRequestMessage(const std::shared_ptr<Piece>& piece, size_t blockIndex) {
- return std::unique_ptr<BtRequestMessage>{};
- }
- virtual std::unique_ptr<BtCancelMessage>
- createCancelMessage(size_t index, int32_t begin, int32_t length) {
- return std::unique_ptr<BtCancelMessage>{};
- }
- virtual std::unique_ptr<BtPieceMessage>
- createPieceMessage(size_t index, int32_t begin, int32_t length) {
- return std::unique_ptr<BtPieceMessage>{};
- }
- virtual std::unique_ptr<BtHaveMessage> createHaveMessage(size_t index) {
- return std::unique_ptr<BtHaveMessage>{};
- }
- virtual std::unique_ptr<BtChokeMessage> createChokeMessage() {
- return std::unique_ptr<BtChokeMessage>{};
- }
- virtual std::unique_ptr<BtUnchokeMessage> createUnchokeMessage() {
- return std::unique_ptr<BtUnchokeMessage>{};
- }
- virtual std::unique_ptr<BtInterestedMessage> createInterestedMessage() {
- return std::unique_ptr<BtInterestedMessage>{};
- }
- virtual std::unique_ptr<BtNotInterestedMessage> createNotInterestedMessage() {
- return std::unique_ptr<BtNotInterestedMessage>{};
- }
- virtual std::unique_ptr<BtBitfieldMessage> createBitfieldMessage() {
- return std::unique_ptr<BtBitfieldMessage>{};
- }
- virtual std::unique_ptr<BtKeepAliveMessage> createKeepAliveMessage() {
- return std::unique_ptr<BtKeepAliveMessage>{};
- }
- virtual std::unique_ptr<BtHaveAllMessage> createHaveAllMessage() {
- return std::unique_ptr<BtHaveAllMessage>{};
- }
- virtual std::unique_ptr<BtHaveNoneMessage> createHaveNoneMessage() {
- return std::unique_ptr<BtHaveNoneMessage>{};
- }
- virtual std::unique_ptr<BtRejectMessage>
- createRejectMessage(size_t index, int32_t begin, int32_t length) {
- return std::unique_ptr<BtRejectMessage>{};
- }
- virtual std::unique_ptr<BtAllowedFastMessage> createAllowedFastMessage
- (size_t index) {
- return std::unique_ptr<BtAllowedFastMessage>{};
- }
- virtual std::unique_ptr<BtPortMessage> createPortMessage(uint16_t port)
- {
- return std::unique_ptr<BtPortMessage>{};
- }
- virtual std::unique_ptr<BtExtendedMessage>
- createBtExtendedMessage(const std::shared_ptr<ExtensionMessage>& extmsg)
- {
- return std::unique_ptr<BtExtendedMessage>{};
- }
- };
- } // namespace aria2
- #endif // D_MOCK_BT_MESSAGE_FACTORY_H
|