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     MOCK_METHOD2(open, bool(const std::string&, std::ios_base::openmode));
15     MOCK_METHOD0(close, void());
16     MOCK_METHOD2(write, bool(std::uint32_t, const std::vector<std::uint8_t>&));
17     MOCK_METHOD2(read, std::optional<std::vector<std::uint8_t>>(std::uint32_t,
18                                                                 std::uint32_t));
19     MOCK_METHOD0(getSize, int());
20 };
21 
22 } // namespace ipmi_flash
23