xref: /openbmc/phosphor-ipmi-flash/tools/test/updater_mock.hpp (revision 328f520f44a4e986d58b204468660e7ddeedc189)
1 #pragma once
2 
3 #include "updater.hpp"
4 
5 #include <cstdint>
6 #include <string>
7 #include <vector>
8 
9 #include <gmock/gmock.h>
10 
11 namespace host_tool
12 {
13 
14 class UpdateHandlerMock : public UpdateHandlerInterface
15 {
16   public:
17     MOCK_METHOD1(checkAvailable, bool(const std::string&));
18     MOCK_METHOD1(readVersion, std::vector<uint8_t>(const std::string&));
19     MOCK_METHOD2(sendFile, void(const std::string&, const std::string&));
20     MOCK_METHOD2(verifyFile, bool(const std::string&, bool));
21     MOCK_METHOD0(cleanArtifacts, void());
22 };
23 
24 } // namespace host_tool
25