1 #pragma once 2 3 #include "interface.hpp" 4 5 #include <gmock/gmock.h> 6 7 namespace host_tool 8 { 9 10 class DataInterfaceMock : public DataInterface 11 { 12 public: 13 virtual ~DataInterfaceMock() = default; 14 15 MOCK_METHOD(bool, sendContents, (const std::string&, std::uint16_t), 16 (override)); 17 MOCK_METHOD(void, waitForRetry, (), (override)); 18 MOCK_METHOD(ipmi_flash::FirmwareFlags::UpdateFlags, supportedType, (), 19 (const, override)); 20 }; 21 22 } // namespace host_tool 23