mirror of
https://github.com/aria2/aria2.git
synced 2025-01-04 09:03:46 +00:00
44 lines
926 B
C++
44 lines
926 B
C++
#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<std::shared_ptr<DHTTask>> periodicTaskQueue1_;
|
|
|
|
std::deque<std::shared_ptr<DHTTask>> periodicTaskQueue2_;
|
|
|
|
std::deque<std::shared_ptr<DHTTask>> immediateTaskQueue_;
|
|
|
|
MockDHTTaskQueue() {}
|
|
|
|
virtual ~MockDHTTaskQueue() {}
|
|
|
|
virtual void executeTask() CXX11_OVERRIDE {}
|
|
|
|
virtual void
|
|
addPeriodicTask1(const std::shared_ptr<DHTTask>& task) CXX11_OVERRIDE
|
|
{
|
|
periodicTaskQueue1_.push_back(task);
|
|
}
|
|
|
|
virtual void
|
|
addPeriodicTask2(const std::shared_ptr<DHTTask>& task) CXX11_OVERRIDE
|
|
{
|
|
periodicTaskQueue2_.push_back(task);
|
|
}
|
|
|
|
virtual void
|
|
addImmediateTask(const std::shared_ptr<DHTTask>& task) CXX11_OVERRIDE
|
|
{
|
|
immediateTaskQueue_.push_back(task);
|
|
}
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // D_MOCK_DHT_TASK_QUEUE_H
|