test/MockBtMessageDispatcher.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

#ifndef _D_MOCK_BT_MESSAGE_DISPATCHER_H_
#define _D_MOCK_BT_MESSAGE_DISPATCHER_H_

#include "BtMessageDispatcher.h"

class MockBtMessageDispatcher : public BtMessageDispatcher {
public:
  BtMessages messageQueue;

  virtual ~MockBtMessageDispatcher() {}

  virtual void addMessageToQueue(const BtMessageHandle& btMessage) {
    messageQueue.push_back(btMessage);
  }

  virtual void addMessageToQueue(const BtMessages& btMessages) {
    copy(btMessages.begin(), btMessages.end(), back_inserter(messageQueue));
  }

  virtual void sendMessages() {}

  virtual void doCancelSendingPieceAction(int32_t index, int32_t begin, int32_t length) {}

  virtual void doCancelSendingPieceAction(const PieceHandle& piece) {}

  virtual void doAbortOutstandingRequestAction(const PieceHandle& piece) {}

  virtual void doChokedAction() {}

  virtual void doChokingAction() {}

  virtual void checkRequestSlotAndDoNecessaryThing() {}

  virtual bool isSendingInProgress() {
    return false;
  }

  virtual int32_t countMessageInQueue() {
    return messageQueue.size();
  }

  virtual int32_t countOutstandingRequest() {
    return 0;
  }

  virtual bool isOutstandingRequest(int32_t index, int32_t blockIndex) {
    return false;
  }

  virtual RequestSlot getOutstandingRequest(int32_t index, int32_t begin, int32_t length) {
    return RequestSlot::nullSlot;
  }

  virtual void removeOutstandingRequest(const RequestSlot& slot) {}
  
  virtual void addOutstandingRequest(const RequestSlot& slot) {}
};

typedef SharedHandle<MockBtMessageDispatcher> MockBtMessageDispatcherHandle;

#endif // _D_MOCK_BT_MESSAGE_DISPATCHER_H_