1 #pragma once
2 
3 #include "image_handler.hpp"
4 
5 #include <gmock/gmock.h>
6 
7 namespace ipmi_flash
8 {
9 
10 class ImageHandlerMock : public ImageHandlerInterface
11 {
12   public:
13     virtual ~ImageHandlerMock() = default;
14 
15     MOCK_METHOD1(open, bool(const std::string&));
16     MOCK_METHOD0(close, void());
17     MOCK_METHOD2(write, bool(std::uint32_t, const std::vector<std::uint8_t>&));
18     MOCK_METHOD0(getSize, int());
19 };
20 
21 } // namespace ipmi_flash
22