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