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