aria2/test/MockDHTTask.h
Tatsuhiro Tsujikawa 5d6f1c046a 2008-02-11 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
Bootstrap through node added by port message.
	Currently bootstrap is executed if the number of buckets in routing
	table is 1.
	* src/BtPortMessage.{h, cc}
	* src/DefaultBtMessageFactory.{h, cc}
	* src/PeerInteractionCommand.cc
	* test/BtPortMessageTest.cc
	* test/MockDHTTask.h
2008-02-10 15:57:00 +00:00

37 lines
644 B
C++

#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_