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