#pragma once

#include "data_handler.hpp"

#include <gmock/gmock.h>

namespace ipmi_flash
{

class DataHandlerMock : public DataInterface
{
  public:
    virtual ~DataHandlerMock() = default;

    MOCK_METHOD0(open, bool());
    MOCK_METHOD0(close, bool());
    MOCK_METHOD1(copyFrom, std::vector<std::uint8_t>(std::uint32_t));
    MOCK_METHOD1(writeMeta, bool(const std::vector<std::uint8_t>&));
    MOCK_METHOD0(readMeta, std::vector<std::uint8_t>());
};

} // namespace ipmi_flash