mirror of https://github.com/aria2/aria2
61 lines
1.6 KiB
C++
61 lines
1.6 KiB
C++
#ifndef D_MOCK_DHT_MESSAGE_DISPATCHER_H
|
|
#define D_MOCK_DHT_MESSAGE_DISPATCHER_H
|
|
|
|
#include "DHTMessageDispatcher.h"
|
|
#include "DHTMessageCallback.h"
|
|
#include "DHTMessage.h"
|
|
#include "DHTConstants.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MockDHTMessageDispatcher:public DHTMessageDispatcher {
|
|
public:
|
|
struct Entry {
|
|
std::unique_ptr<DHTMessage> message_;
|
|
time_t timeout_;
|
|
std::unique_ptr<DHTMessageCallback> callback_;
|
|
|
|
Entry(std::unique_ptr<DHTMessage> message, time_t timeout,
|
|
std::unique_ptr<DHTMessageCallback> callback)
|
|
: message_{std::move(message)},
|
|
timeout_{timeout},
|
|
callback_{std::move(callback)}
|
|
{}
|
|
};
|
|
|
|
std::deque<Entry> messageQueue_;
|
|
|
|
public:
|
|
MockDHTMessageDispatcher() {}
|
|
|
|
virtual void
|
|
addMessageToQueue(std::unique_ptr<DHTMessage> message,
|
|
time_t timeout,
|
|
std::unique_ptr<DHTMessageCallback> callback =
|
|
std::unique_ptr<DHTMessageCallback>{}) CXX11_OVERRIDE
|
|
{
|
|
messageQueue_.push_back(Entry(std::move(message), timeout,
|
|
std::move(callback)));
|
|
}
|
|
|
|
virtual void
|
|
addMessageToQueue(std::unique_ptr<DHTMessage> message,
|
|
std::unique_ptr<DHTMessageCallback> callback =
|
|
std::unique_ptr<DHTMessageCallback>{}) CXX11_OVERRIDE
|
|
{
|
|
messageQueue_.push_back(Entry(std::move(message), DHT_MESSAGE_TIMEOUT,
|
|
std::move(callback)));
|
|
}
|
|
|
|
virtual void sendMessages() CXX11_OVERRIDE {}
|
|
|
|
virtual size_t countMessageInQueue() const CXX11_OVERRIDE
|
|
{
|
|
return messageQueue_.size();
|
|
}
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // D_MOCK_DHT_MESSAGE_DISPATCHER_H
|