summaryrefslogtreecommitdiff
path: root/test/MockDHTTask.h
blob: cba8047bda2d6612c524fdce8b320c072ee599ec (plain)
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
#ifndef _D_MOCK_DHT_TASK_H_
#define _D_MOCK_DHT_TASK_H_

#include "DHTTask.h"
#include "DHTNode.h"
#include "DHTConstants.h"
#include <cstring>

namespace aria2 {

class MockDHTTask:public DHTTask {
public:
  SharedHandle<DHTNode> remoteNode_;

  unsigned char targetID_[DHT_ID_LENGTH];

  MockDHTTask(const SharedHandle<DHTNode>& remoteNode):remoteNode_(remoteNode) {}

  virtual ~MockDHTTask() {}

  virtual void startup() {}

  virtual bool finished()
  {
    return false;
  }

  void setTargetID(const unsigned char* targetID)
  {
    memcpy(targetID_, targetID, DHT_ID_LENGTH);
  }
};

} // namespace aria2

#endif // _D_MOCK_DHT_TASK_H_