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
|
#ifndef _D_MOCK_DHT_TASK_QUEUE_H_
#define _D_MOCK_DHT_TASK_QUEUE_H_
#include "DHTTaskQueue.h"
namespace aria2 {
class MockDHTTaskQueue:public DHTTaskQueue {
public:
std::deque<SharedHandle<DHTTask> > periodicTaskQueue1_;
std::deque<SharedHandle<DHTTask> > periodicTaskQueue2_;
std::deque<SharedHandle<DHTTask> > immediateTaskQueue_;
MockDHTTaskQueue() {}
virtual ~MockDHTTaskQueue() {}
virtual void executeTask() {}
virtual void addPeriodicTask1(const SharedHandle<DHTTask>& task)
{
periodicTaskQueue1_.push_back(task);
}
virtual void addPeriodicTask2(const SharedHandle<DHTTask>& task)
{
periodicTaskQueue2_.push_back(task);
}
virtual void addImmediateTask(const SharedHandle<DHTTask>& task)
{
immediateTaskQueue_.push_back(task);
}
};
} // namespace aria2
#endif // _D_MOCK_DHT_TASK_QUEUE_H_
|