aria2/test/MockExtensionMessageFactory.h
2013-07-01 21:42:51 +09:00

24 lines
577 B
C++

#ifndef D_MOCK_EXTENSION_MESSAGE_FACTORY_H
#define D_MOCK_EXTENSION_MESSAGE_FACTORY_H
#include "ExtensionMessageFactory.h"
#include "MockExtensionMessage.h"
namespace aria2 {
class MockExtensionMessageFactory:public ExtensionMessageFactory {
public:
virtual ~MockExtensionMessageFactory() {}
virtual std::unique_ptr<ExtensionMessage>
createMessage(const unsigned char* data, size_t length)
{
return make_unique<MockExtensionMessage>
("a2_mock", *data, data+1, length-1, nullptr);
}
};
} // namespace aria2
#endif // D_MOCK_EXTENSION_MESSAGE_FACTORY_H