1*ded66d0fSJason Ling #pragma once
2*ded66d0fSJason Ling 
3*ded66d0fSJason Ling #include "data_handler.hpp"
4*ded66d0fSJason Ling 
5*ded66d0fSJason Ling #include <gmock/gmock.h>
6*ded66d0fSJason Ling 
7*ded66d0fSJason Ling namespace ipmi_flash
8*ded66d0fSJason Ling {
9*ded66d0fSJason Ling 
10*ded66d0fSJason Ling class DataHandlerMock : public DataInterface
11*ded66d0fSJason Ling {
12*ded66d0fSJason Ling   public:
13*ded66d0fSJason Ling     virtual ~DataHandlerMock() = default;
14*ded66d0fSJason Ling 
15*ded66d0fSJason Ling     MOCK_METHOD0(open, bool());
16*ded66d0fSJason Ling     MOCK_METHOD0(close, bool());
17*ded66d0fSJason Ling     MOCK_METHOD1(copyFrom, std::vector<std::uint8_t>(std::uint32_t));
18*ded66d0fSJason Ling     MOCK_METHOD1(writeMeta, bool(const std::vector<std::uint8_t>&));
19*ded66d0fSJason Ling     MOCK_METHOD0(readMeta, std::vector<std::uint8_t>());
20*ded66d0fSJason Ling };
21*ded66d0fSJason Ling 
22*ded66d0fSJason Ling } // namespace ipmi_flash
23