aria2/test/MockDHTTaskQueue.h

#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