mirror of https://github.com/aria2/aria2
41 lines
724 B
C++
41 lines
724 B
C++
#ifndef D_MOCK_DHT_TASK_H
|
|
#define D_MOCK_DHT_TASK_H
|
|
|
|
#include "DHTTask.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include "DHTNode.h"
|
|
#include "DHTConstants.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MockDHTTask : public DHTTask {
|
|
public:
|
|
std::shared_ptr<DHTNode> remoteNode_;
|
|
|
|
unsigned char targetID_[DHT_ID_LENGTH];
|
|
|
|
bool finished_;
|
|
|
|
MockDHTTask(const std::shared_ptr<DHTNode>& remoteNode)
|
|
: remoteNode_(remoteNode), finished_(false)
|
|
{
|
|
}
|
|
|
|
virtual ~MockDHTTask() {}
|
|
|
|
virtual void startup() CXX11_OVERRIDE {}
|
|
|
|
virtual bool finished() CXX11_OVERRIDE { return finished_; }
|
|
|
|
void setTargetID(const unsigned char* targetID)
|
|
{
|
|
memcpy(targetID_, targetID, DHT_ID_LENGTH);
|
|
}
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // D_MOCK_DHT_TASK_H
|