mirror of https://github.com/aria2/aria2
62 lines
1.5 KiB
C++
62 lines
1.5 KiB
C++
#ifndef D_MOCK_EXTENSION_MESSAGE_H
|
|
#define D_MOCK_EXTENSION_MESSAGE_H
|
|
|
|
#include "ExtensionMessage.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MockExtensionMessage:public ExtensionMessage {
|
|
public:
|
|
std::string extensionName_;
|
|
uint8_t extensionMessageID_;
|
|
std::string data_;
|
|
bool doReceivedActionCalled_;
|
|
public:
|
|
MockExtensionMessage(const std::string& extensionName,
|
|
uint8_t extensionMessageID,
|
|
const unsigned char* data,
|
|
size_t length):extensionName_(extensionName),
|
|
extensionMessageID_(extensionMessageID),
|
|
data_(&data[0], &data[length]),
|
|
doReceivedActionCalled_(false) {}
|
|
|
|
MockExtensionMessage(const std::string& extensionName,
|
|
uint8_t extensionMessageID,
|
|
const std::string& data):
|
|
extensionName_(extensionName),
|
|
extensionMessageID_(extensionMessageID),
|
|
data_(data),
|
|
doReceivedActionCalled_(false) {}
|
|
|
|
virtual ~MockExtensionMessage() {}
|
|
|
|
virtual std::string getPayload()
|
|
{
|
|
return data_;
|
|
}
|
|
|
|
virtual uint8_t getExtensionMessageID()
|
|
{
|
|
return extensionMessageID_;
|
|
}
|
|
|
|
virtual const char* getExtensionName() const
|
|
{
|
|
return extensionName_.c_str();
|
|
}
|
|
|
|
virtual std::string toString() const
|
|
{
|
|
return extensionName_;
|
|
}
|
|
|
|
virtual void doReceivedAction()
|
|
{
|
|
doReceivedActionCalled_ = true;
|
|
}
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // D_MOCK_EXTENSION_MESSAGE_H
|