MockBtMessageFactory.h 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. #ifndef _D_MOCK_BT_MESSAGE_FACTORY_H_
  2. #define _D_MOCK_BT_MESSAGE_FACTORY_H_
  3. #include "BtMessageFactory.h"
  4. class ExtensionMessage;
  5. typedef SharedHandle<ExtensionMessage> ExensionMessageHandle;
  6. class MockBtMessageFactory : public BtMessageFactory {
  7. public:
  8. MockBtMessageFactory() {}
  9. virtual ~MockBtMessageFactory() {}
  10. virtual BtMessageHandle
  11. createBtMessage(const unsigned char* msg, int32_t msgLength) {
  12. return BtMessageHandle(0);
  13. };
  14. virtual BtMessageHandle
  15. createHandshakeMessage(const unsigned char* msg, int32_t msgLength) {
  16. return BtMessageHandle(0);
  17. }
  18. virtual BtMessageHandle
  19. createHandshakeMessage(const unsigned char* infoHash,
  20. const unsigned char* peerId) {
  21. return BtMessageHandle(0);
  22. }
  23. virtual BtMessageHandle
  24. createRequestMessage(const PieceHandle& piece, int32_t blockIndex) {
  25. return BtMessageHandle(0);
  26. }
  27. virtual BtMessageHandle
  28. createCancelMessage(int32_t index, int32_t begin, int32_t length) {
  29. return BtMessageHandle(0);
  30. }
  31. virtual BtMessageHandle
  32. createPieceMessage(int32_t index, int32_t begin, int32_t length) {
  33. return BtMessageHandle(0);
  34. }
  35. virtual BtMessageHandle createHaveMessage(int32_t index) {
  36. return BtMessageHandle(0);
  37. }
  38. virtual BtMessageHandle createChokeMessage() {
  39. return BtMessageHandle(0);
  40. }
  41. virtual BtMessageHandle createUnchokeMessage() {
  42. return BtMessageHandle(0);
  43. }
  44. virtual BtMessageHandle createInterestedMessage() {
  45. return BtMessageHandle(0);
  46. }
  47. virtual BtMessageHandle createNotInterestedMessage() {
  48. return BtMessageHandle(0);
  49. }
  50. virtual BtMessageHandle createBitfieldMessage() {
  51. return BtMessageHandle(0);
  52. }
  53. virtual BtMessageHandle createKeepAliveMessage() {
  54. return BtMessageHandle(0);
  55. }
  56. virtual BtMessageHandle createHaveAllMessage() {
  57. return BtMessageHandle(0);
  58. }
  59. virtual BtMessageHandle createHaveNoneMessage() {
  60. return BtMessageHandle(0);
  61. }
  62. virtual BtMessageHandle
  63. createRejectMessage(int32_t index, int32_t begin, int32_t length) {
  64. return BtMessageHandle(0);
  65. }
  66. virtual BtMessageHandle createAllowedFastMessage(int32_t index) {
  67. return BtMessageHandle(0);
  68. }
  69. virtual BtMessageHandle createBtExtendedMessage(const ExensionMessageHandle&)
  70. {
  71. return BtMessageHandle(0);
  72. }
  73. };
  74. typedef SharedHandle<MockBtMessageFactory> MockBtMessageFactoryHandle;
  75. #endif // _D_MOCK_BT_MESSAGE_FACTORY_H_