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