1 #ifndef D_MOCK_BT_MESSAGE_FACTORY_H 2 #define D_MOCK_BT_MESSAGE_FACTORY_H 3 4 #include "BtMessageFactory.h" 5 6 #include "BtHandshakeMessage.h" 7 #include "BtRequestMessage.h" 8 #include "BtCancelMessage.h" 9 #include "BtPieceMessage.h" 10 #include "BtHaveMessage.h" 11 #include "BtChokeMessage.h" 12 #include "BtUnchokeMessage.h" 13 #include "BtInterestedMessage.h" 14 #include "BtNotInterestedMessage.h" 15 #include "BtBitfieldMessage.h" 16 #include "BtKeepAliveMessage.h" 17 #include "BtHaveAllMessage.h" 18 #include "BtHaveNoneMessage.h" 19 #include "BtRejectMessage.h" 20 #include "BtAllowedFastMessage.h" 21 #include "BtPortMessage.h" 22 #include "BtExtendedMessage.h" 23 #include "ExtensionMessage.h" 24 25 namespace aria2 { 26 27 class ExtensionMessage; 28 29 class MockBtMessageFactory : public BtMessageFactory { 30 public: MockBtMessageFactory()31 MockBtMessageFactory() {} 32 ~MockBtMessageFactory()33 virtual ~MockBtMessageFactory() {} 34 35 virtual std::unique_ptr<BtMessage> createBtMessage(const unsigned char * msg,size_t msgLength)36 createBtMessage(const unsigned char* msg, size_t msgLength) CXX11_OVERRIDE 37 { 38 return nullptr; 39 }; 40 41 virtual std::unique_ptr<BtHandshakeMessage> createHandshakeMessage(const unsigned char * msg,size_t msgLength)42 createHandshakeMessage(const unsigned char* msg, 43 size_t msgLength) CXX11_OVERRIDE 44 { 45 return nullptr; 46 } 47 48 virtual std::unique_ptr<BtHandshakeMessage> createHandshakeMessage(const unsigned char * infoHash,const unsigned char * peerId)49 createHandshakeMessage(const unsigned char* infoHash, 50 const unsigned char* peerId) CXX11_OVERRIDE 51 { 52 return nullptr; 53 } 54 55 virtual std::unique_ptr<BtRequestMessage> createRequestMessage(const std::shared_ptr<Piece> & piece,size_t blockIndex)56 createRequestMessage(const std::shared_ptr<Piece>& piece, 57 size_t blockIndex) CXX11_OVERRIDE 58 { 59 return nullptr; 60 } 61 62 virtual std::unique_ptr<BtCancelMessage> createCancelMessage(size_t index,int32_t begin,int32_t length)63 createCancelMessage(size_t index, int32_t begin, 64 int32_t length) CXX11_OVERRIDE 65 { 66 return nullptr; 67 } 68 69 virtual std::unique_ptr<BtPieceMessage> createPieceMessage(size_t index,int32_t begin,int32_t length)70 createPieceMessage(size_t index, int32_t begin, int32_t length) CXX11_OVERRIDE 71 { 72 return nullptr; 73 } 74 75 virtual std::unique_ptr<BtHaveMessage> createHaveMessage(size_t index)76 createHaveMessage(size_t index) CXX11_OVERRIDE 77 { 78 return nullptr; 79 } 80 createChokeMessage()81 virtual std::unique_ptr<BtChokeMessage> createChokeMessage() CXX11_OVERRIDE 82 { 83 return nullptr; 84 } 85 86 virtual std::unique_ptr<BtUnchokeMessage> createUnchokeMessage()87 createUnchokeMessage() CXX11_OVERRIDE 88 { 89 return nullptr; 90 } 91 92 virtual std::unique_ptr<BtInterestedMessage> createInterestedMessage()93 createInterestedMessage() CXX11_OVERRIDE 94 { 95 return nullptr; 96 } 97 98 virtual std::unique_ptr<BtNotInterestedMessage> createNotInterestedMessage()99 createNotInterestedMessage() CXX11_OVERRIDE 100 { 101 return nullptr; 102 } 103 104 virtual std::unique_ptr<BtBitfieldMessage> createBitfieldMessage()105 createBitfieldMessage() CXX11_OVERRIDE 106 { 107 return nullptr; 108 } 109 110 virtual std::unique_ptr<BtKeepAliveMessage> createKeepAliveMessage()111 createKeepAliveMessage() CXX11_OVERRIDE 112 { 113 return nullptr; 114 } 115 116 virtual std::unique_ptr<BtHaveAllMessage> createHaveAllMessage()117 createHaveAllMessage() CXX11_OVERRIDE 118 { 119 return nullptr; 120 } 121 122 virtual std::unique_ptr<BtHaveNoneMessage> createHaveNoneMessage()123 createHaveNoneMessage() CXX11_OVERRIDE 124 { 125 return nullptr; 126 } 127 128 virtual std::unique_ptr<BtRejectMessage> createRejectMessage(size_t index,int32_t begin,int32_t length)129 createRejectMessage(size_t index, int32_t begin, 130 int32_t length) CXX11_OVERRIDE 131 { 132 return nullptr; 133 } 134 135 virtual std::unique_ptr<BtAllowedFastMessage> createAllowedFastMessage(size_t index)136 createAllowedFastMessage(size_t index) CXX11_OVERRIDE 137 { 138 return nullptr; 139 } 140 141 virtual std::unique_ptr<BtPortMessage> createPortMessage(uint16_t port)142 createPortMessage(uint16_t port) CXX11_OVERRIDE 143 { 144 return nullptr; 145 } 146 createBtExtendedMessage(std::unique_ptr<ExtensionMessage> extmsg)147 virtual std::unique_ptr<BtExtendedMessage> createBtExtendedMessage( 148 std::unique_ptr<ExtensionMessage> extmsg) CXX11_OVERRIDE 149 { 150 return nullptr; 151 } 152 }; 153 154 } // namespace aria2 155 156 #endif // D_MOCK_BT_MESSAGE_FACTORY_H 157